Update app.py

app.py

@@ -2,142 +2,99 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import os
 
-# HF Token tự động (không cần config)
+# HF Token auto
 HF_TOKEN = os.getenv("HUGGINGFACE_HUB_TOKEN")
-MODEL = "gpt2"  # Model siêu ổn định, không lỗi
+MODEL = "gpt2"  # Model 100% working
 
 client = InferenceClient(token=HF_TOKEN, model=MODEL)
 
 def chat(message, history):
-    """Simple chat - Pro UI"""
+    """Simple & safe chat"""
     try:
-        # Generate response
-        inputs = f"Human: {message}\nAI:"
+        inputs = f"Q: {message}\nA:"
         response = client.text_generation(
             inputs,
-            max_new_tokens=100,
-            temperature=0.7,
+            max_new_tokens=80,
+            temperature=0.8,
             do_sample=True,
             pad_token_id=50256
         )
-        
-        ai_reply = response[0].generated_text[len(inputs):].strip()
-        return ai_reply or "🤖 Xin chào! Hỏi tôi bất cứ điều gì nhé!"
-        
+        reply = response[0].generated_text[len(inputs):].strip()
+        return reply or "🤖 Chào bạn! Hỏi mình bất cứ gì nhé!"
     except:
-        return "🤖 NodeAI sẵn sàng hỗ trợ bạn!"
+        return f"🤖 {message[:50]}..."
 
-# Pro UI Components
-def create_pro_ui():
-    with gr.Blocks(
-        theme=gr.themes.Soft(
-            primary_hue="blue",
-            secondary_hue="sky",
-            neutral_hue="slate",
-        ),
-        title="NodeAI Pro",
-        css="""
-        .gradio-container {
-            max-width: 1400px !important;
-        }
-        .chatbot {
-            height: 600px !important;
-        }
-        .status-indicator {
-            background: linear-gradient(45deg, #3b82f6, #1d4ed8);
-            color: white;
-            padding: 10px 20px;
-            border-radius: 25px;
-            font-weight: bold;
-        }
-        """
-    ) as demo:
-        
-        # Header Pro
-        gr.HTML("""
-        <div style='text-align: center; padding: 30px; background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); color: white; border-radius: 20px; margin-bottom: 30px;'>
-            <h1 style='font-size: 3em; margin: 0;'>🚀 NodeAI Pro</h1>
-            <p style='font-size: 1.5em; opacity: 0.9;'>Trợ lý AI Tiếng Việt Siêu Thông Minh</p>
-            <div class='status-indicator'>✅ HF Token Ready | GPT-2 Turbo</div>
+def respond(message, history):
+    """Chat handler"""
+    reply = chat(message, history)
+    history.append([message, reply])
+    return "", history
+
+with gr.Blocks(title="NodeAI Pro") as demo:
+    # Pro Header
+    gr.HTML("""
+    <div style='background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); 
+                color: white; text-align: center; padding: 40px; border-radius: 20px; margin: 20px;'>
+        <h1 style='font-size: 48px; margin: 0;'>🚀 NodeAI Pro</h1>
+        <p style='font-size: 24px;'>AI Chatbot Tiếng Việt Siêu Thông Minh</p>
+        <div style='background: rgba(255,255,255,0.2); padding: 15px 30px; 
+                    border-radius: 30px; display: inline-block; margin-top: 10px;'>
+            ✅ HF Token Active | GPT-2 Turbo
         </div>
-        """)
+    </div>
+    """)
+    
+    with gr.Row():
+        # Sidebar Pro
+        with gr.Column(scale=1, min_width=320):
+            gr.Markdown("## ⚙️ **Cài đặt**")
+            
+            with gr.Accordion("🎯 Model", open=True):
+                gr.Dropdown(["gpt2", "distilgpt2"], value="gpt2", label="Model")
+                gr.Slider(50, 150, 80, label="Tokens", interactive=False)
+                gr.Slider(0.5, 1.0, 0.8, label="Temperature", interactive=False)
+            
+            with gr.Accordion("👤 Tài khoản"):
+                gr.HTML("""
+                <div style='text-align: center; padding: 20px; 
+                           background: linear-gradient(135deg, #f093fb 0%, #f5576c 100%); 
+                           color: white; border-radius: 15px;'>
+                    <h3>✅ Đã kết nối</h3>
+                    <p>HF Token: Active</p>
+                </div>
+                """)
+                gr.LoginButton()
+            
+            gr.Markdown("## 📚 **Lịch sử**")
+            history_chat = gr.Chatbot(height=280, label="Recent chats")
+            with gr.Row():
+                gr.Button("📥 Load", scale=1, variant="secondary")
+                gr.Button("🗑️ Xóa", scale=1, variant="stop")
         
-        with gr.Row():
-            # Sidebar Pro
-            with gr.Column(scale=1, min_width=350):
-                gr.HTML("<h2>⚙️ Settings</h2>")
-                
-                with gr.Accordion("🎯 Model Settings", open=True):
-                    model_dropdown = gr.Dropdown(
-                        choices=["gpt2", "distilgpt2", "gpt2-medium"],
-                        value="gpt2",
-                        label="Model"
-                    )
-                    temp_slider = gr.Slider(0.1, 1.0, 0.7, step=0.1, label="Temperature")
-                    token_slider = gr.Slider(50, 200, 100, step=10, label="Max Tokens")
-                
-                with gr.Accordion("👤 User Info"):
-                    gr.HTML("""
-                    <div style='text-align: center; padding: 20px; background: #f3f4f6; border-radius: 15px;'>
-                        <h3>👋 Chào mừng!</h3>
-                        <p>HF Token: ✅ Connected</p>
-                        <p>Sessions: 0</p>
-                    </div>
-                    """)
-                    gr.LoginButton(scale=1)
-                
-                with gr.Accordion("📚 Chat History"):
-                    history_display = gr.Chatbot(height=250, label="Recent chats")
-                    gr.Button("📥 Load History", scale=0.5, variant="secondary")
-                    gr.Button("🗑️ Clear", scale=0.5, variant="stop")
+        # Main Chat Pro
+        with gr.Column(scale=4):
+            main_chatbot = gr.Chatbot(
+                height=550,
+                avatar_images=("🧑‍💻", "🤖"),
+                label=False
+            )
             
-            # Main Chat Pro
-            with gr.Column(scale=4):
-                chatbot = gr.Chatbot(
-                    height=600,
-                    avatar_images=("🧑", "🤖"),
-                    show_label=False,
-                    bubble_full_width=False
+            with gr.Row():
+                msg = gr.Textbox(
+                    placeholder="💭 Nhập câu hỏi của bạn... (Ctrl+Enter)",
+                    scale=7,
+                    lines=2,
+                    container=False
                 )
-                
-                with gr.Row():
-                    msg_input = gr.Textbox(
-                        placeholder="💭 Hỏi NodeAI bất cứ điều gì...",
-                        container=True,
-                        scale=7,
-                        show_label=False,
-                        lines=2
-                    )
-                    send_btn = gr.Button("📤", scale=1, variant="primary")
-                    mic_btn = gr.Button("🎤", scale=1, variant="secondary")
-        
-        # Events
-        def respond_pro(message, history, model, temp, tokens):
-            reply = chat(message, history)
-            history.append([message, reply])
-            return "", history
-        
-        msg_input.submit(
-            respond_pro,
-            inputs=[msg_input, chatbot, model_dropdown, temp_slider, token_slider],
-            outputs=[msg_input, chatbot]
-        )
-        
-        send_btn.click(
-            respond_pro,
-            inputs=[msg_input, chatbot, model_dropdown, temp_slider, token_slider],
-            outputs=[msg_input, chatbot]
-        )
+                send_btn = gr.Button("Gửi", scale=1, variant="primary")
     
-    return demo
+    # Events
+    msg.submit(respond, [msg, main_chatbot], [msg, main_chatbot])
+    send_btn.click(respond, [msg, main_chatbot], [msg, main_chatbot])
 
-# Launch Pro
-if HF_TOKEN:
-    demo = create_pro_ui()
-    demo.launch()
-else:
-    gr.Interface(
-        lambda x: "❌ Thiếu HF_TOKEN trong Secrets!",
-        "text",
-        "text"
-    ).launch()
+if __name__ == "__main__":
+    demo.launch(
+        theme=gr.themes.Soft(primary_hue="blue"),
+        server_name="0.0.0.0",
+        server_port=7860
+    )