Skip to content

Commit 824dbee

Browse files
committed
Fix Jinja2 TemplateResponse for Starlette 1.0 API
Starlette 1.0 changed TemplateResponse signature to require request as first positional arg and context as keyword arg. Updated all four TemplateResponse calls to match new API.
1 parent 8402510 commit 824dbee

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

server.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -831,7 +831,7 @@ async def get_web_ui(request: Request):
831831
"success": None,
832832
}
833833

834-
return templates.TemplateResponse("index.html", template_context)
834+
return templates.TemplateResponse(request, "index.html", context=template_context)
835835

836836
except Exception as e:
837837
logger.error(f"Error rendering Web UI: {e}", exc_info=True)
@@ -909,7 +909,7 @@ async def handle_web_ui_generate(
909909
"output_file_url": None,
910910
"generation_time": None,
911911
}
912-
return templates.TemplateResponse("index.html", error_context, status_code=503)
912+
return templates.TemplateResponse(request, "index.html", context=error_context, status_code=503)
913913

914914
# --- Start processing the valid request ---
915915
logger.info(
@@ -1053,7 +1053,7 @@ async def handle_web_ui_generate(
10531053
"generation_time": None,
10541054
}
10551055
return templates.TemplateResponse(
1056-
"index.html", error_context, status_code=400
1056+
request, "index.html", context=error_context, status_code=400
10571057
) # Bad Request
10581058

10591059
# --- Generation ---
@@ -1219,9 +1219,10 @@ async def handle_web_ui_generate(
12191219

12201220
# Render and return the HTML response
12211221
return templates.TemplateResponse(
1222+
request,
12221223
"index.html",
1223-
template_context, # Pass the complete context dictionary
1224-
status_code=status_code, # Set appropriate HTTP status
1224+
context=template_context,
1225+
status_code=status_code,
12251226
)
12261227

12271228

0 commit comments

Comments
 (0)