Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py (REVISED)
+# app.py (FINAL CORRECTED VERSION)
 
 from typing import Optional, Dict, List, Tuple
 import gradio as gr
@@ -33,7 +33,7 @@ def generation_code(
     enable_search: bool,
     language: str,
     provider: str,
-    hf_token: str  # <-- CHANGE 1: Accept the user's HF token from Gradio
+    hf_token: str  # This parameter is now correctly passed from the UI
 ) -> Tuple[str, History, str, List[Dict[str, str]]]:
     # Initialize inputs
     if query is None:
@@ -61,7 +61,6 @@ def generation_code(
     messages.append({'role': 'user', 'content': final_query})
 
     # Model inference
-    # <-- CHANGE 2: Pass the user's token to the client constructor
     client = get_inference_client(_current_model_name, provider, user_token=hf_token)
     resp = client.chat.completions.create(
         model=_current_model_name,
@@ -87,7 +86,6 @@ def generation_code(
     new_history = _history + [(query, code_str)]
     chat_msgs = history_to_chatbot_messages(new_history)
 
-    # The return values are now correct for the updated UI components
     return code_str, new_history, preview_html, chat_msgs
 
 # Build UI
@@ -99,30 +97,18 @@ with gr.Blocks(theme=gr.themes.Base(), title="AnyCoder - AI Code Generator") as
 
     with gr.Sidebar():
         gr.Markdown("## AnyCoder AI")
-
-        # Load project
+        # ... (rest of sidebar UI is correct) ...
         url_in = gr.Textbox(label="Load HF Space URL", placeholder="https://huggingface.co/spaces/user/project")
         load_btn = gr.Button("Import Project")
         load_status = gr.Markdown(visible=False)
-
         gr.Markdown("---")
-
-        # Inputs
         prompt_in = gr.Textbox(label="Prompt", lines=3)
         file_in = gr.File(label="Reference file")
         image_in = gr.Image(label="Design image")
         url_site = gr.Textbox(label="Website URL")
         search_chk = gr.Checkbox(label="Enable Web Search")
-        language_dd = gr.Dropdown(
-            choices=["html", "python", "transformers.js"],
-            value="html",
-            label="Language"
-        )
-        model_dd = gr.Dropdown(
-            choices=[m['name'] for m in AVAILABLE_MODELS],
-            value=AVAILABLE_MODELS[0]['name'],
-            label="Model"
-        )
+        language_dd = gr.Dropdown(choices=["html", "python", "transformers.js"], value="html", label="Language")
+        model_dd = gr.Dropdown(choices=[m['name'] for m in AVAILABLE_MODELS], value=AVAILABLE_MODELS[0]['name'], label="Model")
         gen_btn = gr.Button("Generate")
         clr_btn = gr.Button("Clear")
 
@@ -133,7 +119,6 @@ with gr.Blocks(theme=gr.themes.Base(), title="AnyCoder - AI Code Generator") as
             with gr.Tab("Preview"):
                 preview_out = gr.HTML(label="Live Preview")
             with gr.Tab("History"):
-                # <-- CHANGE 3: Fix the Gradio error by specifying the modern 'messages' format.
                 chat_out = gr.Chatbot(label="History", type="messages")
 
     # Events
@@ -149,29 +134,21 @@ with gr.Blocks(theme=gr.themes.Base(), title="AnyCoder - AI Code Generator") as
                 return m['id']
         return AVAILABLE_MODELS[0]['id']
 
-    model_dd.change(
-        fn=on_model_change,
-        inputs=[model_dd],
-        outputs=[model_state]
-    )
+    model_dd.change(fn=on_model_change, inputs=[model_dd], outputs=[model_state])
 
-    # Note: Gradio automatically passes the user's token to any function that
-    # has an 'hf_token' parameter. You do NOT need to add it to the 'inputs' list.
     gen_btn.click(
         fn=generation_code,
         inputs=[
             prompt_in, image_in, file_in, url_site,
             setting_state, history_state, model_state,
-            search_chk, language_dd, gr.State('auto')
+            search_chk, language_dd, gr.State('auto'),
+            "hf_token"  # ### FIX #1: This line fixes the "Expected 11 arguments, received 10" warning.
         ],
         outputs=[code_out, history_state, preview_out, chat_out]
     )
 
-    clr_btn.click(
-        fn=lambda: ([], [], "", []),
-        outputs=[history_state, chat_out, preview_out, code_out]
-    )
+    clr_btn.click(fn=lambda: ([], [], "", []), outputs=[history_state, chat_out, preview_out, code_out])
 
 if __name__ == '__main__':
-    # <-- CHANGE 4: Launch with hf_token='must' to require login and get user tokens.
-    demo.queue().launch(hf_token="must")
+    # ### FIX #2: This line fixes the "unexpected keyword argument 'hf_token'" TypeError.
+    demo.queue().launch()