Update app.py

app.py

@@ -2,27 +2,56 @@ import os
 from groq import Groq
 import gradio as gr
 
-# Get API key securely from Hugging Face secrets
+# Initialize Groq client using HF secret
 client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
+# System prompt (bot personality)
+SYSTEM_PROMPT = {
+    "role": "system",
+    "content": "You are a helpful, friendly, and intelligent AI assistant. Answer clearly and concisely."
+}
+
+# Chatbot function
 def chatbot(message, history):
-    messages = [{"role": "system", "content": "You are a helpful AI assistant."}]
+    try:
+        messages = [SYSTEM_PROMPT]
+
+        # Limit history to avoid token overflow
+        history = history[-5:] if history else []
+
+        # Convert Gradio history → Groq format
+        for user_msg, bot_msg in history:
+            if user_msg:
+                messages.append({"role": "user", "content": user_msg})
+            if bot_msg:
+                messages.append({"role": "assistant", "content": bot_msg})
 
-    for user, bot in history:
-        messages.append({"role": "user", "content": user})
-        messages.append({"role": "assistant", "content": bot})
+        # Add current message
+        messages.append({"role": "user", "content": message})
 
-    messages.append({"role": "user", "content": message})
+        # API call
+        response = client.chat.completions.create(
+            model="llama-3.3-70b-versatile",
+            messages=messages,
+            temperature=0.7,   # creativity
+            max_tokens=1024,   # response length
+        )
 
-    response = client.chat.completions.create(
-        messages=messages,
-        model="llama-3.3-70b-versatile",
-    )
+        reply = response.choices[0].message.content.strip()
+        return reply
 
-    return response.choices[0].message.content
+    except Exception as e:
+        return f"⚠️ Error: {str(e)}"
 
-gr.ChatInterface(
+
+# UI
+demo = gr.ChatInterface(
     fn=chatbot,
     title="🚀 Groq AI Chatbot",
-    description="Fast chatbot using Groq + Llama",
-).launch()
+    description="Fast, smart chatbot powered by Groq",
+    theme="soft",
+)
+
+# Launch (important for Hugging Face)
+if __name__ == "__main__":
+    demo.launch()