Fix: Use DialoGPT-small model

app.py

@@ -1,49 +1,52 @@
-# app.py - HuggingFace Spaces deployment
 import gradio as gr
-from model import predict
+from transformers import pipeline
+import torch
 
-def chat(message, history):
-    history = history or []
-    response = predict(message, history)
-    history.append((message, response))
-    return history, history
+# Cek GPU
+device = 0 if torch.cuda.is_available() else -1
 
-def clear_history():
-    return [], ""
+# Load model (akan otomatis download pertama kali)
+print("Loading AI model...")
+generator = pipeline(
+    'text-generation',
+    model='microsoft/DialoGPT-small',  # Model kecil, cepat
+    device=device
+)
 
-# Custom CSS
-custom_css = """
-.gradio-container {
-    max-width: 1200px !important;
-    margin: auto !important;
-}
-"""
-
-with gr.Blocks(css=custom_css, title="Bangdim AI") as demo:
-    gr.Markdown("""
-    <div style="text-align: center; margin-bottom: 20px;">
-        <h1>🤖 Bangdim AI Customer Service</h1>
-        <p style="font-size: 16px;">AI Assistant untuk Top Up Game</p>
-    </div>
-    """)
+def chat(message, history):
+    # Build context
+    context = ""
+    for h in history[-3:]:
+        context += f"User: {h[0]}\nAI: {h[1]}\n"
     
-    chatbot_ui = gr.Chatbot(height=500, type="tuples")
+    prompt = f"{context}User: {message}\nAI:"
     
-    with gr.Row():
-        msg = gr.Textbox(scale=9, placeholder="Ketik pesan Anda di sini...", label="Pesan")
-        send_btn = gr.Button("Kirim", variant="primary", scale=1)
+    # Generate response
+    result = generator(
+        prompt,
+        max_length=150,
+        temperature=0.8,
+        do_sample=True,
+        pad_token_id=50256
+    )
     
-    clear_btn = gr.Button("Clear Chat", size="sm")
+    response = result[0]['generated_text']
+    response = response.replace(prompt, "").strip()
     
-    send_btn.click(chat, [msg, chatbot_ui], [chatbot_ui, chatbot_ui]).then(
-        lambda: "", None, msg
-    )
+    if not response:
+        response = "Maaf kak, saya kurang paham. Bisa diulang? 😊"
     
-    msg.submit(chat, [msg, chatbot_ui], [chatbot_ui, chatbot_ui]).then(
-        lambda: "", None, msg
-    )
+    history.append((message, response))
+    return history, history
+
+# UI
+with gr.Blocks() as demo:
+    gr.Markdown("# 🤖 Bangdim AI")
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox(label="Pesan")
+    clear = gr.Button("Clear")
     
-    clear_btn.click(clear_history, None, [chatbot_ui, msg])
+    msg.submit(chat, [msg, chatbot], [chatbot, msg])
+    clear.click(lambda: None, None, chatbot)
 
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()