Update app.py

app.py

@@ -1,4 +1,3 @@
-# app.py
 import os
 import gradio as gr
 from groq import Groq
@@ -7,49 +6,23 @@ from groq import Groq
 # 1. Setup
 # =========================
 
-# Get API key from environment variables (set in Hugging Face Spaces settings)
 groq_api_key = os.environ.get("GROQ_API_KEY")
-
 if not groq_api_key:
-    raise ValueError(
-        "GROQ_API_KEY not found. Please set it in Hugging Face Spaces → Settings → Repository Secrets."
-    )
+    raise ValueError("GROQ_API_KEY not found. Please set it in Hugging Face Spaces → Settings → Repository Secrets.")
 
 client = Groq(api_key=groq_api_key)
-
 MODEL_NAME = "llama-3.3-70b-versatile"
 
-
 # =========================
 # 2. Chat Function
 # =========================
 
 def chat_with_groq(message, history):
-    """
-    Gradio 6 passes history as OpenAI-style messages:
-    [
-        {"role": "user", "content": "..."},
-        {"role": "assistant", "content": "..."}
-    ]
-    """
-
-    messages = [
-        {
-            "role": "system",
-            "content": "You are a helpful, clear, and practical AI assistant."
-        }
-    ]
-
-    # Add previous conversation history
+    messages = [{"role": "system", "content": "You are a helpful, clear, and practical AI assistant."}]
     if history:
         for item in history:
-            role = item.get("role")
-            content = item.get("content")
-
-            if role in ["user", "assistant"] and isinstance(content, str):
-                messages.append({"role": role, "content": content})
-
-    # Add latest user message
+            if item.get("role") in ["user", "assistant"] and isinstance(item.get("content"), str):
+                messages.append({"role": item["role"], "content": item["content"]})
     messages.append({"role": "user", "content": message})
 
     response = client.chat.completions.create(
@@ -58,10 +31,8 @@ def chat_with_groq(message, history):
         temperature=0.7,
         max_completion_tokens=1024,
     )
-
     return response.choices[0].message.content
 
-
 # =========================
 # 3. Gradio App
 # =========================
@@ -80,15 +51,15 @@ demo = gr.ChatInterface(
     fill_width=False,
 )
 
-
 # =========================
 # 4. Launch
 # =========================
 
 if __name__ == "__main__":
     demo.launch(
-        server_name="0.0.0.0",  # Required for Hugging Face Spaces
+        server_name="0.0.0.0",
         server_port=7860,
-        theme=gr.themes.Soft(),
         debug=True
+        # Remove theme if Gradio < 6
+        # theme=gr.themes.Soft()
     )