Update app.py

app.py

@@ -1,115 +1,73 @@
 import os
-import requests
 import gradio as gr
+from groq import Groq
 
-# -----------------------------
-# ENVIRONMENT
-# -----------------------------
+# ===== Setup =====
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 if not GROQ_API_KEY:
-    raise ValueError("GROQ_API_KEY is not set.")
+    raise RuntimeError("⚠️ GROQ_API_KEY is missing! Add it in Hugging Face → Settings → Secrets.")
 
-GROQ_URL = "https://api.groq.com/openai/v1/chat/completions"
-TEXT_MODEL = "llama-3.1-8b-instant"
+client = Groq(api_key=GROQ_API_KEY)
 
-# -----------------------------
-# MODE PROMPTS
-# -----------------------------
+# Mode-specific system prompts
 MODE_PROMPTS = {
-    "Normal": "You are NeoHelper, a helpful, concise assistant.",
-    "School": "You are NeoHelper, explaining clearly for a Grade 5 student.",
-    "Shopping": "You are NeoHelper, a shopping assistant in Saudi Arabia.",
-    "Gaming": "You are NeoHelper, a practical gaming helper.",
+    "Normal Chat": "You are NeoHelper, Eyad’s branded assistant. Be concise, friendly, and always respond in English.",
+    "School Helper": "You are NeoHelper, a school helper. Explain clearly, step-by-step, in simple English. Keep it encouraging.",
+    "Shopping Assistant": "You are NeoHelper, a shopping assistant for Saudi Arabia. Compare specs and value, be concise, and respond in English.",
+    "Gaming Help": "You are NeoHelper, an energetic gaming helper. Provide practical tips, fixes, and short steps. Respond in English."
 }
 
-# -----------------------------
-# FIXED GROQ CHAT FUNCTION
-# -----------------------------
-def groq_chat(user_msg, history, mode):
-    system_prompt = MODE_PROMPTS.get(mode, MODE_PROMPTS["Normal"])
-
-    # Start with system message
-    messages = [{"role": "system", "content": system_prompt}]
-
-    # Sanitize history to avoid Groq format errors
-    if history:
-        for pair in history:
-            if isinstance(pair, list) and len(pair) == 2:
-                user, bot = pair
-                if isinstance(user, str) and user.strip():
-                    messages.append({"role": "user", "content": user.strip()})
-                if isinstance(bot, str) and bot.strip():
-                    messages.append({"role": "assistant", "content": bot.strip()})
-
-    # Add current user message
-    if user_msg and user_msg.strip():
-        messages.append({"role": "user", "content": user_msg.strip()})
-    else:
-        return "Please type something."
-
-    payload = {
-        "model": TEXT_MODEL,
-        "messages": messages,
-        "max_tokens": 400,
-        "temperature": 0.4,
-    }
-
-    headers = {
-        "Authorization": f"Bearer {GROQ_API_KEY}",
-        "Content-Type": "application/json",
-    }
-
-    resp = requests.post(GROQ_URL, json=payload, headers=headers, timeout=30)
-    resp.raise_for_status()
-    data = resp.json()
-    return data["choices"][0]["message"]["content"]
-
-# -----------------------------
-# GRADIO UI
-# -----------------------------
-def build_ui():
-    with gr.Blocks(title="NeoHelper (Text Only)") as demo:
-        gr.Markdown("# **NeoHelper — Text‑Only (Groq)**")
-
-        with gr.Row():
-            with gr.Column(scale=3):
-                chatbot = gr.Chatbot(height=520, label="NeoHelper")
-
-                with gr.Row():
-                    user_input = gr.Textbox(
-                        placeholder="Type your message…",
-                        lines=3,
-                        show_label=False,
-                    )
-                    send_btn = gr.Button("Send")
-
-                clear_btn = gr.Button("Clear chat")
-
-            with gr.Column(scale=1):
-                mode = gr.Radio(
-                    ["Normal", "School", "Shopping", "Gaming"],
-                    value="Normal",
-                    label="Mode",
-                )
-
-        def on_send(msg, history, mode):
-            history = history or []
-            reply = groq_chat(msg, history, mode)
-            history.append([msg, reply])
-            return history, ""
-
-        send_btn.click(
-            on_send,
-            inputs=[user_input, chatbot, mode],
-            outputs=[chatbot, user_input],
+DEFAULT_MODEL = "llama-3.1-8b-instant"
+
+# ===== Core chat function =====
+def chat_fn(message, history, mode="Normal Chat", model=DEFAULT_MODEL):
+    try:
+        # Build messages list from history
+        messages = [{"role": "system", "content": MODE_PROMPTS.get(mode, MODE_PROMPTS["Normal Chat"])}]
+        for user_msg, assistant_msg in history:
+            if user_msg:
+                messages.append({"role": "user", "content": user_msg})
+            if assistant_msg:
+                messages.append({"role": "assistant", "content": assistant_msg})
+
+        # Add current user message
+        messages.append({"role": "user", "content": message})
+
+        # Call Groq API
+        response = client.chat.completions.create(
+            model=model,
+            messages=messages
+        )
+        assistant_response = response.choices[0].message.content
+        return assistant_response
+    except Exception as e:
+        return f"⚠️ Error: {str(e)}"
+
+# ===== UI =====
+with gr.Blocks(title="🦙 NeoHelper — Eyad’s Groq-powered Assistant") as demo:
+    gr.Markdown("## 🦙 NeoHelper — Eyad’s Groq-powered Assistant")
+    gr.Markdown("Modes: School Helper, Shopping Assistant, Gaming Help, Normal Chat (English only).")
+
+    with gr.Row():
+        mode_dd = gr.Dropdown(
+            choices=list(MODE_PROMPTS.keys()),
+            value="Normal Chat",
+            label="Mode"
+        )
+        model_dd = gr.Dropdown(
+            choices=[
+                "llama-3.1-8b-instant",
+                "llama-3.1-70b-versatile",
+                "mixtral-8x7b-instruct",
+                "gemma2-9b-it"
+            ],
+            value=DEFAULT_MODEL,
+            label="Model"
         )
 
-        clear_btn.click(lambda: ([], ""), None, [chatbot, user_input])
-
-    return demo
-
-
-app = build_ui()
+    chat = gr.ChatInterface(
+        fn=lambda msg, hist: chat_fn(msg, hist, mode_dd.value, model_dd.value),
+        title=None
+    )
 
-if __name__ == "__main__":
-    app.launch(server_name="0.0.0.0", server_port=7860, theme="gradio/soft")
+demo.launch()