Update app.py

app.py

@@ -7,7 +7,7 @@ This file defines the user interface, handles user interactions, and orchestrate
 the calls to the backend logic in other modules.
 """
 import gradio as gr
-import os  # <--- FIX: Import the os module
+import os
 
 from config import AVAILABLE_MODELS, MULTIMODAL_MODELS
 from core import generate_code
@@ -54,7 +54,6 @@ def on_model_change(model_name: str):
 def on_language_change(language: str):
     return gr.update(language=get_gradio_language(language))
 
-# <--- FIX: Use gr.Request to get the auth token
 def on_deploy_click(code: str, space_name: str, sdk_name: str, request: gr.Request):
     """Handler for the deploy button, now correctly using gr.Request."""
     hf_token = request.token
@@ -67,6 +66,7 @@ def on_deploy_click(code: str, space_name: str, sdk_name: str, request: gr.Reque
 
 # --- Gradio UI Layout ---
 with gr.Blocks(theme=gr.themes.Default(primary_hue="blue"), title="AnyCoder") as demo:
+    # 1. DEFINE ALL COMPONENTS FIRST
     history_state = gr.State([])
 
     gr.Markdown("# 🤖 AnyCoder - AI Code Generator")
@@ -88,21 +88,15 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue"), title="AnyCoder") as
                 clear_btn = gr.Button("Clear")
                 generate_btn = gr.Button("Generate", variant="primary", scale=2)
 
-            # <--- FIX: Conditionally render the deployment UI only when on a HF Space
+            # Conditionally define the deployment components
+            # We must define them here so they exist for the event wiring later.
             if os.getenv("SPACE_ID"):
                 with gr.Accordion("🚀 Deploy to Hugging Face", open=False):
-                    login_button = gr.LoginButton()
+                    login_button = gr.LoginButton() # This is just a placeholder in this context
                     space_name_input = gr.Textbox(label="New App Name", placeholder="my-cool-app")
                     sdk_dropdown = gr.Dropdown(choices=["Static (HTML)", "Gradio (Python)"], value="Static (HTML)", label="App Type")
                     deploy_btn = gr.Button("Deploy")
                     deploy_status = gr.Markdown(visible=False)
-                    
-                    # <--- FIX: Correct event handler wiring
-                    deploy_btn.click(
-                        on_deploy_click,
-                        inputs=[code_output, space_name_input, sdk_dropdown],
-                        outputs=deploy_status
-                    )
             
         with gr.Column(scale=2):
             with gr.Tabs():
@@ -116,19 +110,33 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue"), title="AnyCoder") as
                         type="messages",
                         height=500
                     )
+
+    # 2. WIRE UP ALL EVENTS AT THE END
+    # This ensures all component variables (like 'code_output') are defined before being used.
     
-    # --- Event Wiring ---
+    # Main generation event
     generate_btn.click(
         fn=on_generate_click,
         inputs=[input_prompt, image_input, file_input, website_url_input, history_state, model_dropdown, search_toggle, language_dropdown],
         outputs=[code_output, sandbox, history_state, history_output]
     )
+
+    # Clear button event
     clear_btn.click(lambda: ([], [], None, None, None, "", ""), 
                     outputs=[history_state, history_output, input_prompt, image_input, file_input, website_url_input, code_output])
     
+    # UI change events
     model_dropdown.change(on_model_change, inputs=model_dropdown, outputs=image_input)
     language_dropdown.change(on_language_change, inputs=language_dropdown, outputs=code_output)
     language_dropdown.change(lambda code, lang: send_to_sandbox(code, lang), inputs=[code_output, language_dropdown], outputs=sandbox)
 
+    # Conditionally wire up the deployment event
+    if os.getenv("SPACE_ID"):
+        deploy_btn.click(
+            on_deploy_click,
+            inputs=[code_output, space_name_input, sdk_dropdown],
+            outputs=deploy_status
+        )
+
 if __name__ == "__main__":
     demo.queue().launch()