Update app.py

app.py

@@ -1,5 +1,3 @@
-# app.py (FINAL CORRECTED VERSION)
-
 from typing import Optional, Dict, List, Tuple
 import gradio as gr
 
@@ -22,6 +20,8 @@ from deploy import send_to_sandbox, load_project_from_url
 History = List[Tuple[str, str]]
 
 # Core generation function
+# This function signature correctly includes 'hf_token'. Gradio will automatically
+# provide this value if the user is logged in.
 def generation_code(
     query: Optional[str],
     image: Optional[gr.Image],
@@ -33,7 +33,7 @@ def generation_code(
     enable_search: bool,
     language: str,
     provider: str,
-    hf_token: str  # This parameter is now correctly passed from the UI
+    hf_token: str
 ) -> Tuple[str, History, str, List[Dict[str, str]]]:
     # Initialize inputs
     if query is None:
@@ -60,7 +60,7 @@ def generation_code(
     final_query = enhance_query_with_search(query, enable_search)
     messages.append({'role': 'user', 'content': final_query})
 
-    # Model inference
+    # Model inference - using the provided user token for billing
     client = get_inference_client(_current_model_name, provider, user_token=hf_token)
     resp = client.chat.completions.create(
         model=_current_model_name,
@@ -97,7 +97,6 @@ with gr.Blocks(theme=gr.themes.Base(), title="AnyCoder - AI Code Generator") as
 
     with gr.Sidebar():
         gr.Markdown("## AnyCoder AI")
-        # ... (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)
@@ -119,6 +118,7 @@ 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"):
+                # FIX: The chatbot now correctly uses the modern 'messages' format.
                 chat_out = gr.Chatbot(label="History", type="messages")
 
     # Events
@@ -138,11 +138,13 @@ with gr.Blocks(theme=gr.themes.Base(), title="AnyCoder - AI Code Generator") as
 
     gen_btn.click(
         fn=generation_code,
+        # FIX: The 'inputs' list now only contains Gradio components.
+        # The 'hf_token' is passed automatically by Gradio because it's
+        # in the function's signature.
         inputs=[
             prompt_in, image_in, file_in, url_site,
             setting_state, history_state, model_state,
-            search_chk, language_dd, gr.State('auto'),
-            "hf_token"  # ### FIX #1: This line fixes the "Expected 11 arguments, received 10" warning.
+            search_chk, language_dd, gr.State('auto')
         ],
         outputs=[code_out, history_state, preview_out, chat_out]
     )
@@ -150,5 +152,5 @@ with gr.Blocks(theme=gr.themes.Base(), title="AnyCoder - AI Code Generator") as
     clr_btn.click(fn=lambda: ([], [], "", []), outputs=[history_state, chat_out, preview_out, code_out])
 
 if __name__ == '__main__':
-    # ### FIX #2: This line fixes the "unexpected keyword argument 'hf_token'" TypeError.
+    # FIX: The launch() method is now correct, without any extra arguments.
     demo.queue().launch()