Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py
+# app.py (REVISED)
 
 from typing import Optional, Dict, List, Tuple
 import gradio as gr
@@ -32,7 +32,8 @@ def generation_code(
     _current_model_name: str,
     enable_search: bool,
     language: str,
-    provider: str
+    provider: str,
+    hf_token: str  # <-- CHANGE 1: Accept the user's HF token from Gradio
 ) -> Tuple[str, History, str, List[Dict[str, str]]]:
     # Initialize inputs
     if query is None:
@@ -60,7 +61,8 @@ def generation_code(
     messages.append({'role': 'user', 'content': final_query})
 
     # Model inference
-    client = get_inference_client(_current_model_name, provider)
+    # <-- 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,
         messages=messages,
@@ -85,6 +87,7 @@ 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
@@ -130,7 +133,8 @@ 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"):
-                chat_out = gr.Chatbot(label="History")
+                # <-- CHANGE 3: Fix the Gradio error by specifying the modern 'messages' format.
+                chat_out = gr.Chatbot(label="History", type="messages")
 
     # Events
     load_btn.click(
@@ -151,6 +155,8 @@ with gr.Blocks(theme=gr.themes.Base(), title="AnyCoder - AI Code Generator") as
         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=[
@@ -167,4 +173,5 @@ with gr.Blocks(theme=gr.themes.Base(), title="AnyCoder - AI Code Generator") as
     )
 
 if __name__ == '__main__':
-    demo.queue().launch()
+    # <-- CHANGE 4: Launch with hf_token='must' to require login and get user tokens.
+    demo.queue().launch(hf_token="must")