Update app.py

app.py

@@ -4,101 +4,73 @@ import requests
 import json
 
 groq_key = os.getenv("GROQ_API_KEY")
-MODEL = "llama3-8b-8192"
+MODEL = "llama-3.1-8b-instant"  # ✅ Model mới thay thế llama3-8b-8192
 
 status_msg = "❌ Chưa set GROQ_API_KEY"
 if groq_key:
-    status_msg = f"✅ Ready - Groq {MODEL} (key: {len(groq_key)} chars)"
-
-def test_groq():
-    """Test API key"""
-    if not groq_key:
-        return "❌ NO GROQ_API_KEY"
-    
-    try:
-        url = "https://api.groq.com/openai/v1/models"
-        headers = {"Authorization": f"Bearer {groq_key}"}
-        r = requests.get(url, headers=headers, timeout=10)
-        if r.status_code == 200:
-            models = r.json().get('data', [])
-            model_names = [m['id'] for m in models[:3]]
-            return f"✅ OK - Models: {model_names}"
-        return f"❌ {r.status_code}: {r.text[:100]}"
-    except Exception as e:
-        return f"❌ Test error: {str(e)}"
+    status_msg = f"✅ Ready - Groq {MODEL}"
 
 def call_groq(messages):
-    """Groq API với full error debug"""
     url = "https://api.groq.com/openai/v1/chat/completions"
     headers = {
         "Authorization": f"Bearer {groq_key}",
         "Content-Type": "application/json"
     }
     
-    payload = {
-        "messages": messages,
+    data = {
         "model": MODEL,
-        "stream": False,
-        "temperature": 0.7,
-        "max_tokens": 1000
+        "messages": messages,
+        "max_tokens": 1000,
+        "temperature": 0.7
     }
     
     try:
-        print(f"Sending to Groq: {json.dumps(payload, indent=2)}")  # DEBUG
-        response = requests.post(url, headers=headers, json=payload, timeout=30)
-        
-        print(f"Response status: {response.status_code}")  # DEBUG
-        print(f"Response: {response.text[:500]}")  # DEBUG
-        
-        if response.status_code == 400:
-            return f"❌ 400 Bad Request\n{response.text}"
-        if response.status_code == 401:
-            return "❌ API Key sai"
-        if response.status_code == 429:
-            return "⏳ Rate limit"
-            
+        response = requests.post(url, headers=headers, json=data, timeout=20)
         response.raise_for_status()
-        data = response.json()
-        return data["choices"][0]["message"]["content"]
-        
+        return response.json()["choices"][0]["message"]["content"]
     except Exception as e:
-        return f"❌ Error: {str(e)}\nStatus: {getattr(response, 'status_code', 'N/A')}"
+        if "model" in str(e).lower():
+            return "❌ Model không tồn tại. Dùng: llama-3.1-8b-instant"
+        return f"❌ Lỗi: {str(e)}"
 
 def respond(message, history):
     if not groq_key:
         return "", history + [{"role": "assistant", "content": status_msg}]
     
-    # Clean messages format
-    messages = [{"role": "system", "content": "Bạn là NodeAI - trợ lý AI tiếng Việt."}]
+    messages = [{"role": "system", "content": "Bạn là NodeAI - trợ lý AI tiếng Việt thông minh."}]
     
     for msg in history:
-        if isinstance(msg, dict) and "role" in msg and msg.get("content"):
-            messages.append({"role": msg["role"], "content": str(msg["content"])})
+        if isinstance(msg, dict) and "role" in msg:
+            messages.append(msg)
     
-    messages.append({"role": "user", "content": str(message)})
+    messages.append({"role": "user", "content": message})
     
     reply = call_groq(messages)
     return "", history + [{"role": "assistant", "content": reply}]
 
 with gr.Blocks(title="NodeAI") as demo:
-    gr.Markdown("# 🚀 NodeAI - Groq AI")
+    gr.Markdown("# 🚀 NodeAI - Groq AI Chat")
+    gr.Markdown("**⚡ Siêu nhanh Llama 3.1 - Miễn phí!**")
     gr.Markdown(f"**Status:** {status_msg}")
     
-    # Test button
-    test_btn = gr.Button("🧪 Test Groq API")
-    test_output = gr.Textbox(label="API Test", lines=3)
-    
-    chatbot = gr.Chatbot(height=400, avatar_images=("👤", "🤖"))
+    chatbot = gr.Chatbot(
+        height=500,
+        show_label=False,
+        avatar_images=("👤", "🤖")
+    )
     
     with gr.Row():
-        msg = gr.Textbox(placeholder="Hỏi NodeAI...", scale=7)
-        send_btn = gr.Button("📤", variant="primary", scale=1)
-        clear_btn = gr.Button("🗑️", scale=1)
+        msg = gr.Textbox(placeholder="Nhập câu hỏi...", scale=7)
+        send_btn = gr.Button("📤 Gửi", variant="primary", scale=1)
+        clear_btn = gr.Button("🗑️ Clear", scale=1)
     
-    test_btn.click(test_groq, outputs=test_output)
     msg.submit(respond, [msg, chatbot], [msg, chatbot])
     send_btn.click(respond, [msg, chatbot], [msg, chatbot])
     clear_btn.click(lambda: [], None, chatbot)
 
 if __name__ == "__main__":
-    demo.launch(theme=gr.themes.Soft(primary_hue="blue"))
+    demo.launch(
+        theme=gr.themes.Soft(primary_hue="blue"),
+        server_name="0.0.0.0",
+        server_port=7860
+    )