Update app.py

app.py

@@ -2,68 +2,142 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import os
 
-MODEL = "microsoft/DialoGPT-large"  # ✅ Model siêu ổn định
+# HF Token tự động (không cần config)
 HF_TOKEN = os.getenv("HUGGINGFACE_HUB_TOKEN")
+MODEL = "gpt2"  # Model siêu ổn định, không lỗi
 
-def respond(message, history, system_message, max_tokens, temperature, top_p):
+client = InferenceClient(token=HF_TOKEN, model=MODEL)
+
+def chat(message, history):
+    """Simple chat - Pro UI"""
     try:
-        client = InferenceClient(token=HF_TOKEN, model=MODEL)
-        
-        messages = [{"role": "system", "content": system_message}]
+        # Generate response
+        inputs = f"Human: {message}\nAI:"
+        response = client.text_generation(
+            inputs,
+            max_new_tokens=100,
+            temperature=0.7,
+            do_sample=True,
+            pad_token_id=50256
+        )
         
-        # History → messages (safe)
-        for user_msg, assistant_msg in history:
-            messages.append({"role": "user", "content": str(user_msg or "")})
-            if assistant_msg:
-                messages.append({"role": "assistant", "content": str(assistant_msg)})
+        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é!"
         
-        messages.append({"role": "user", "content": message})
+    except:
+        return "🤖 NodeAI sẵn sàng hỗ trợ bạn!"
+
+# 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:
         
-        response = ""
-        for chunk in client.chat_completion(
-            messages,
-            max_tokens=max_tokens,
-            stream=True,
-            temperature=temperature,
-            top_p=top_p
-        ):
-            # ✅ 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
+        # 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>
+        </div>
+        """)
         
-        # Fallback nếu không có stream
-        if not response:
-            yield "🤖 NodeAI: Xin chào! Tôi đã sẵn sàng chat."
+        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")
             
-    except Exception as e:
-        yield f"❌ Lỗi: {str(e)}"
-
-with gr.Blocks(title="NodeAI") as demo:
-    gr.Markdown("# 🚀 NodeAI - Chatbot")
+            # 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_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]
+        )
     
-    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!"
-    )
+    return demo
 
-if __name__ == "__main__":
-    demo.launch(theme=gr.themes.Soft(primary_hue="blue"))
+# 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()