Update app.py

app.py

@@ -1,54 +1,56 @@
-# GOAL: Building an ai powered chatbot
-# GOAL: Building an AI-powered chatbot using GROQ API
+# app.py
 import gradio as gr
 import requests
 import os
-from groq import Groq
 from dotenv import load_dotenv
 
-load_dotenv()  # Load GROQ_API_KEY from .env
+load_dotenv()  # Load .env if running locally
 
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
+if not GROQ_API_KEY:
+    raise ValueError("Missing GROQ_API_KEY. Set it in Hugging Face Secrets or a .env file.")
+
 GROQ_API_URL = "https://api.groq.com/openai/v1/chat/completions"
-MODEL = "llama3-8b-8192"  # or "mixtral-8x7b-32768", etc.
+MODEL = "llama3-8b-8192"
 
 headers = {
     "Authorization": f"Bearer {GROQ_API_KEY}",
     "Content-Type": "application/json"
 }
 
-# backend: Python
 def echo(message, history):
-    messages = [
-        {"role": "system", "content": "You are a helpful LLM teacher."},
-        {"role": "user", "content": message}
-    ]
-
-    response = requests.post(
-        GROQ_API_URL,
-        headers=headers,
-        json={
-            "model": MODEL,
-            "messages": messages
-        }
-    )
-
-    if response.status_code == 200:
-        completion = response.json()
-        return completion["choices"][0]["message"]["content"]
-    else:
-        return f"Error: {response.status_code}\n{response.text}"
-
-# frontend: Gradio
+    messages = [{"role": "system", "content": "You are a helpful LLM teacher."}]
+    
+    # Add chat history for better context
+    for user, bot in history:
+        messages.append({"role": "user", "content": user})
+        messages.append({"role": "assistant", "content": bot})
+    
+    messages.append({"role": "user", "content": message})
+
+    try:
+        response = requests.post(
+            GROQ_API_URL,
+            headers=headers,
+            json={
+                "model": MODEL,
+                "messages": messages
+            }
+        )
+
+        if response.status_code == 200:
+            return response.json()["choices"][0]["message"]["content"]
+        else:
+            return f"Error: {response.status_code}\n{response.text}"
+    except Exception as e:
+        return f"Exception occurred: {str(e)}"
+
 demo = gr.ChatInterface(
     fn=echo,
     type="messages",
-    examples=[
-        "I want to learn about LLMs",
-        "What is NLP",
-        "What is RAG"
-    ],
+    examples=["I want to learn about LLMs", "What is NLP?", "What is RAG?"],
     title="LLM Mentor"
 )
 
-demo.launch()
+# Disable SSR to avoid Hugging Face crash
+demo.launch(ssr_mode=False)