Update app.py

app.py

@@ -1,31 +1,40 @@
-# app.py
 import gradio as gr
 import requests
 import os
 from dotenv import load_dotenv
 
-load_dotenv()  # Load .env if running locally
-
-GROQ_API_KEY = os.getenv("GROQ_API_KEY", "").strip()
+# Load environment variables
+load_dotenv()
+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.")
+    raise ValueError("Missing GROQ_API_KEY. Set it in Hugging Face Secrets or .env.")
 
+# GROQ API setup
 GROQ_API_URL = "https://api.groq.com/openai/v1/chat/completions"
 MODEL = "llama3-8b-8192"
 
 headers = {
-    "Authorization": f"Bearer {GROQ_API_KEY}",
+    "Authorization": f"Bearer {GROQ_API_KEY.strip()}",
     "Content-Type": "application/json"
 }
 
+# Core logic for handling chatbot responses
 def echo(message, history):
-    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 = [
+        {
+            "role": "system",
+            "content": (
+                "You are an expert AI tutor. Answer student questions clearly and in short, "
+                "bite-sized explanations. Avoid jargon. Be patient and helpful like a good teacher."
+            )
+        }
+    ]
+
+    # Add chat history for context
+    for user_msg, bot_reply in history:
+        messages.append({"role": "user", "content": user_msg})
+        messages.append({"role": "assistant", "content": bot_reply})
+
     messages.append({"role": "user", "content": message})
 
     try:
@@ -41,16 +50,28 @@ def echo(message, history):
         if response.status_code == 200:
             return response.json()["choices"][0]["message"]["content"]
         else:
-            return f"Error: {response.status_code}\n{response.text}"
+            return f"❌ Error: {response.status_code}\n{response.text}"
     except Exception as e:
-        return f"Exception occurred: {str(e)}"
+        return f"⚠️ Exception: {str(e)}"
 
+# Gradio UI setup
 demo = gr.ChatInterface(
     fn=echo,
     type="messages",
-    examples=["I want to learn about LLMs", "What is NLP?", "What is RAG?"],
-    title="LLM Mentor"
+    return_messages=True,
+    title="🤖 LLM Mentor - Your AI Tutor",
+    examples=[
+        "What is a language model?",
+        "How does machine learning work?",
+        "Explain RAG in simple terms.",
+        "What is fine-tuning?"
+    ],
+    description=(
+        "Ask me anything about LLMs, NLP, or AI! I'll explain in short, simple steps. "
+        "Great for beginners trying to learn how language models work."
+    )
 )
 
-# Disable SSR to avoid Hugging Face crash
+# Hugging Face Spaces fix
 demo.launch(ssr_mode=False)
+