Update app.py

app.py

@@ -1,4 +1,4 @@
-# /app.py
+# /app.py (Corrected)
 
 """
 Main Gradio application for AnyCoder.
@@ -16,11 +16,16 @@ from utils import get_gradio_language
 # --- UI Helper Functions ---
 
 def send_to_sandbox(code: str, language: str) -> str:
-    """Generates an iframe for the preview tab."""
+    """Generates an iframe for the preview tab, safely handling quotes."""
     if language == "html" and code:
-        return f'<iframe srcdoc="{code.replace("\"", """)}" width="100%" height="920px" style="border:0;"></iframe>'
+        # First, replace all double quotes in the HTML code with the " entity.
+        # This prevents the srcdoc attribute from breaking.
+        safe_code = code.replace('"', '"')
+        # Now, inject the safe code into the iframe.
+        return f'<iframe srcdoc="{safe_code}" width="100%" height="920px" style="border:0;"></iframe>'
     return "<div style='padding:1em;color:#888;text-align:center;'>Preview is only for HTML.</div>"
 
+
 # --- Event Handlers ---
 
 def on_generate_click(
@@ -39,9 +44,12 @@ def on_generate_click(
 
     # Stream the response
     for response in generate_code(query, image, file.name if file else None, website_url, history, model_config, enable_search, language):
+        # Determine the code to be used for the sandbox preview
+        preview_code = response.get("code_output", "")
+        
         ui_update = {
-            code_output: gr.update(value=response.get("code_output"), language=get_gradio_language(language)),
-            sandbox: send_to_sandbox(response.get("code_output", ""), language)
+            code_output: gr.update(value=preview_code, language=get_gradio_language(language)),
+            sandbox: send_to_sandbox(preview_code, language)
         }
         if "history" in response:
             ui_update[history_state] = response["history"]
@@ -119,7 +127,7 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue"), title="AnyCoder") as
     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)
     
-    login_button.login(lambda token: gr.update(value=token), outputs=hf_token_state)
+    login_button.login(lambda token: gr.update(value=token.token if token else None), outputs=hf_token_state)
     deploy_btn.click(
         on_deploy_click,
         inputs=[code_output, space_name_input, sdk_dropdown, hf_token_state],