Update app.py

app.py

@@ -2,8 +2,7 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import os
 
-# ✅ MODEL HOẠT ĐỘNG 100%
-MODEL = "meta-llama/Llama-3.2-1B-Instruct"  
+MODEL = "microsoft/DialoGPT-large"  # ✅ Model siêu ổn định
 HF_TOKEN = os.getenv("HUGGINGFACE_HUB_TOKEN")
 
 def respond(message, history, system_message, max_tokens, temperature, top_p):
@@ -12,9 +11,9 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
         
         messages = [{"role": "system", "content": system_message}]
         
-        # History → messages
+        # History → messages (safe)
         for user_msg, assistant_msg in history:
-            messages.append({"role": "user", "content": str(user_msg)})
+            messages.append({"role": "user", "content": str(user_msg or "")})
             if assistant_msg:
                 messages.append({"role": "assistant", "content": str(assistant_msg)})
         
@@ -28,35 +27,43 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
             temperature=temperature,
             top_p=top_p
         ):
-            if chunk.choices[0].delta.content:
-                response += chunk.choices[0].delta.content
-                yield response
-                
+            # ✅ SAFE ACCESS - Không lỗi index
+            try:
+                if (hasattr(chunk, 'choices') and 
+                    len(chunk.choices) > 0 and 
+                    hasattr(chunk.choices[0], 'delta') and 
+                    hasattr(chunk.choices[0].delta, 'content') and 
+                    chunk.choices[0].delta.content):
+                    
+                    content = chunk.choices[0].delta.content
+                    response += content
+                    yield response
+            except (IndexError, AttributeError):
+                continue  # Skip chunk lỗi
+        
+        # Fallback nếu không có stream
+        if not response:
+            yield "🤖 NodeAI: Xin chào! Tôi đã sẵn sàng chat."
+            
     except Exception as e:
         yield f"❌ Lỗi: {str(e)}"
 
 with gr.Blocks(title="NodeAI") as demo:
-    gr.Markdown("# 🚀 NodeAI - Llama 3.2 Chat")
+    gr.Markdown("# 🚀 NodeAI - Chatbot")
     
-    with gr.Row():
-        with gr.Column(scale=1, min_width=300):
-            gr.Markdown("### 🔐 Login")
-            gr.LoginButton()
-            
-            gr.Markdown("### 📚 History")
-            gr.Chatbot(height=250)
-            
-        with gr.Column(scale=4):
-            chatbot = gr.ChatInterface(
-                respond,
-                additional_inputs=[
-                    gr.Textbox(value="Bạn là NodeAI - trợ lý AI tiếng Việt.", label="System"),
-                    gr.Slider(1, 512, value=256, step=1, label="Max tokens"),
-                    gr.Slider(0.1, 1.0, value=0.7, step=0.1, label="Temperature"),
-                    gr.Slider(0.1, 1.0, value=0.9, step=0.05, label="Top-p"),
-                ],
-                title="💬 Chat Llama 3.2"
-            )
+    chatbot = gr.ChatInterface(
+        respond,
+        additional_inputs=[
+            gr.Textbox(
+                value="Bạn là NodeAI - trợ lý AI tiếng Việt thân thiện.",
+                label="System prompt"
+            ),
+            gr.Slider(1, 512, value=256, step=1, label="Max tokens"),
+            gr.Slider(0.1, 1.0, value=0.7, step=0.1, label="Temperature"),
+            gr.Slider(0.1, 1.0, value=0.9, step=0.05, label="Top-p"),
+        ],
+        title="💬 Chat ngay!"
+    )
 
 if __name__ == "__main__":
     demo.launch(theme=gr.themes.Soft(primary_hue="blue"))