Update app.py

app.py

@@ -561,7 +561,12 @@ theme = gr.themes.Soft(
 
 # ─── Gradio UI ────────────────────────────────────────────────────────────────
 
-with gr.Blocks(title="Codle AI") as demo:
+with gr.Blocks(
+    title="Codle AI",
+    theme=theme,
+    css=CSS,
+    head=f"<script>{JS_SRC}</script>",
+) as demo:
 
     gr.HTML(_header_html())
 
@@ -785,7 +790,4 @@ if __name__ == "__main__":
     demo.launch(
         server_name="0.0.0.0",
         server_port=int(os.environ.get("PORT", 7860)),
-        theme=theme,
-        css=CSS,
-        head=f"<script>{JS_SRC}</script>",
     )