Update app.py

app.py

@@ -1,26 +1,24 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-def respond(
-    message,
-    history,
-    system_message,
-    max_tokens,
-    temperature,
-    top_p,
-    hf_token
-):
+# --- Response function ---
+def respond(message, history, system_message, max_tokens, temperature, top_p, hf_token_info):
     """
-    Handles streaming responses from Hugging Face Inference API
+    Handles streaming responses from Hugging Face Inference API.
+    `hf_token_info` comes from LoginButton: hf_token_info['access_token']
     """
-    client = InferenceClient(token=hf_token.token, model="openai/gpt-oss-20b")
+    if not hf_token_info or "access_token" not in hf_token_info:
+        yield "Error: You must log in with your Hugging Face token."
+        return
+
+    hf_token = hf_token_info["access_token"]
+    client = InferenceClient(token=hf_token, model="openai/gpt-oss-20b")
 
     messages = [{"role": "system", "content": system_message}]
     messages.extend(history)
     messages.append({"role": "user", "content": message})
 
     response = ""
-
     for message_chunk in client.chat_completion(
         messages,
         max_tokens=max_tokens,
@@ -32,15 +30,14 @@ def respond(
         token = ""
         if len(choices) and choices[0].delta.content:
             token = choices[0].delta.content
-        
         response += token
         yield response
 
-# Gradio app
+# --- Gradio UI ---
 with gr.Blocks() as demo:
     with gr.Sidebar():
-        hf_token = gr.OAuthToken()  # Correct usage: no label
-        gr.Markdown("## Configure Chatbot Settings")
+        gr.Markdown("## Hugging Face Login & Chatbot Settings")
+        login_btn = gr.LoginButton("Log in with Hugging Face", provider="huggingface")
         system_message = gr.Textbox(
             value="You are a friendly Chatbot.",
             label="System message"
@@ -55,14 +52,14 @@ with gr.Blocks() as demo:
             minimum=0.1, maximum=1.0, value=0.95, step=0.05,
             label="Top-p (nucleus sampling)"
         )
-    
+
     chatbot = gr.Chatbot()
-    state = gr.State([])  # to hold conversation history
+    state = gr.State([])  # conversation history
     msg = gr.Textbox(label="Your message")
 
     msg.submit(
         respond,
-        inputs=[msg, state, system_message, max_tokens, temperature, top_p, hf_token],
+        inputs=[msg, state, system_message, max_tokens, temperature, top_p, login_btn],
         outputs=[chatbot],
         show_progress=True
     )