Update app.py

app.py

@@ -1,10 +1,11 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-def respond(message, history: list[dict[str,str]], system_message, max_tokens, temperature, top_p, hf_token: gr.OAuthToken):
+def respond(message, history: list[dict[str,str]], system_message, max_tokens, temperature, top_p, hf_token):
     if not hf_token:
         yield "⚠️ You must login first! (Bạn phải đăng nhập trước!)"
         return
+
     client = InferenceClient(token=hf_token.token, model="openai/gpt-oss-20b")
     
     messages = [{"role": "system", "content": system_message}]
@@ -13,14 +14,14 @@ def respond(message, history: list[dict[str,str]], system_message, max_tokens, t
     
     response = ""
     try:
-        for message in client.chat_completion(
+        for message_chunk in client.chat_completion(
             messages,
             max_tokens=max_tokens,
             stream=True,
             temperature=temperature,
             top_p=top_p,
         ):
-            token = message.choices[0].delta.get("content","") if message.choices else ""
+            token = message_chunk.choices[0].delta.get("content","") if message_chunk.choices else ""
             response += token
             yield response
     except Exception as e:
@@ -28,19 +29,24 @@ def respond(message, history: list[dict[str,str]], system_message, max_tokens, t
 
 with gr.Blocks(css=".gradio-container {background-color: #1a1a1a; color: #eee;}") as demo:
     with gr.Row():
+        # Sidebar
         with gr.Column(scale=3, min_width=300):
             gr.Markdown("## ⚡ Login & Settings (Đăng nhập & Cài đặt)")
-            hf_token = gr.LoginButton(label="Login to HF (Đăng nhập HF)")
+            hf_token = gr.LoginButton()
+            gr.Markdown("Login to HF (Đăng nhập HF)")
             system_message = gr.Textbox(value="You are a friendly Chatbot. (Bạn là chatbot thân thiện.)", label="System message", lines=2)
             max_tokens = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max tokens")
             temperature = gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature")
             top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)")
+        
+        # Chat area
         with gr.Column(scale=7):
             chatbot = gr.Chatbot(label="AI Chat (Trò chuyện với AI)").style(height=600)
             with gr.Row():
                 msg = gr.Textbox(placeholder="Type your message here... (Gõ tin nhắn vào đây)", lines=1)
                 submit = gr.Button("Send (Gửi)")
     
+    # Bind events
     submit.click(
         respond,
         inputs=[msg, chatbot, system_message, max_tokens, temperature, top_p, hf_token],