Update app.py

app.py

@@ -1,4 +1,4 @@
-# /app.py (Corrected)
+# /app.py (Final Corrected Version)
 
 """
 Main Gradio application for AnyCoder.
@@ -7,11 +7,12 @@ 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
 
 from config import AVAILABLE_MODELS, MULTIMODAL_MODELS
 from core import generate_code
 from deployment import deploy_to_hf_space
-from utils import get_gradio_language, history_to_chatbot_messages # <--- FIX: Import new helper
+from utils import get_gradio_language, history_to_chatbot_messages
 
 # --- UI Helper Functions ---
 def send_to_sandbox(code: str, language: str) -> str:
@@ -29,7 +30,7 @@ def on_generate_click(
     if not query and not image and not file and not website_url:
         yield {
             code_output: gr.update(value="Please provide a prompt, image, file, or URL.", language="text"),
-            history_output: history_to_chatbot_messages(history) # <--- FIX: Use new helper
+            history_output: history_to_chatbot_messages(history)
         }
         return
 
@@ -43,7 +44,6 @@ def on_generate_click(
         }
         if "history" in response:
             ui_update[history_state] = response["history"]
-            # <--- FIX: Use the helper function to format history for the chatbot
             ui_update[history_output] = history_to_chatbot_messages(response["history"])
         yield ui_update
 
@@ -54,10 +54,10 @@ def on_model_change(model_name: str):
 def on_language_change(language: str):
     return gr.update(language=get_gradio_language(language))
 
-# <--- FIX: Modified function to accept the auth token from the button
-def on_deploy_click(code: str, space_name: str, sdk_name: str, auth_token: gr.OAuthToken | None):
-    """Handler for the deploy button."""
-    hf_token = auth_token.token if auth_token else None
+# <--- 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
     if not hf_token:
         return gr.update(value="⚠️ Please log in with your Hugging Face account to deploy.", visible=True)
     
@@ -68,7 +68,6 @@ def on_deploy_click(code: str, space_name: str, sdk_name: str, auth_token: gr.OA
 # --- Gradio UI Layout ---
 with gr.Blocks(theme=gr.themes.Default(primary_hue="blue"), title="AnyCoder") as demo:
     history_state = gr.State([])
-    # hf_token_state = gr.State() # <--- FIX: No longer needed
 
     gr.Markdown("# 🤖 AnyCoder - AI Code Generator")
     gr.Markdown("Describe what you want to build, upload a design, or provide a URL to redesign. The AI will generate the code for you.")
@@ -89,13 +88,22 @@ 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)
 
-            with gr.Accordion("🚀 Deploy to Hugging Face", open=False):
-                login_button = gr.LoginButton()
-                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: Conditionally render the deployment UI only when on a HF Space
+            if os.getenv("SPACE_ID"):
+                with gr.Accordion("🚀 Deploy to Hugging Face", open=False):
+                    login_button = gr.LoginButton()
+                    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():
                 with gr.Tab("Preview"):
@@ -103,7 +111,6 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue"), title="AnyCoder") as
                 with gr.Tab("Code"):
                     code_output = gr.Code(label="Generated Code", language="html", interactive=True)
                 with gr.Tab("History"):
-                    # <--- FIX: Update Chatbot component to modern standards
                     history_output = gr.Chatbot(
                         label="Conversation History", 
                         type="messages",
@@ -116,22 +123,12 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue"), title="AnyCoder") as
         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]
     )
-    # <--- FIX: Update clear button to output correctly formatted empty list for chatbot
-    clear_btn.click(lambda: ([], [], None, None, None, "", []), 
+    clear_btn.click(lambda: ([], [], None, None, None, "", ""), 
                     outputs=[history_state, history_output, input_prompt, image_input, file_input, website_url_input, code_output])
     
     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)
-    
-    # <--- FIX: REMOVED the incorrect .login() event handler
-    
-    # <--- FIX: Correct way to handle deployment click
-    deploy_btn.click(
-        on_deploy_click,
-        inputs=[code_output, space_name_input, sdk_dropdown, login_button], # Pass the button itself as input
-        outputs=deploy_status
-    )
 
 if __name__ == "__main__":
     demo.queue().launch()