Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 from groq import Groq
 
 # =====================================================
-# Groq Client Initialization (GLOBAL – IMPORTANT)
+# Groq Client Initialization (GLOBAL)
 # =====================================================
 def get_groq_client():
     api_key = os.getenv("GROQ_API_KEY")
@@ -11,10 +11,10 @@ def get_groq_client():
         raise RuntimeError("GROQ_API_KEY not set")
     return Groq(api_key=api_key)
 
-client = get_groq_client()  # ✅ must be global
+client = get_groq_client()
 
 # =====================================================
-# Model Inference Logic (HF-safe, Gradio v4+)
+# Chat Logic (Gradio 6 compatible)
 # =====================================================
 def chat_with_groq(user_message, history):
     if not user_message or not user_message.strip():
@@ -26,7 +26,7 @@ def chat_with_groq(user_message, history):
         "content": user_message
     })
 
-    # Add temporary thinking message
+    # Add thinking placeholder
     history.append({
         "role": "assistant",
         "content": "⚡ Groq is thinking..."
@@ -39,17 +39,14 @@ def chat_with_groq(user_message, history):
     ]
 
     try:
-        chat_completion = client.chat.completions.create(
+        completion = client.chat.completions.create(
             model="llama-3.3-70b-versatile",
             messages=messages,
         )
 
-        response = chat_completion.choices[0].message.content
-
-        # Replace thinking message
         history[-1] = {
             "role": "assistant",
-            "content": response
+            "content": completion.choices[0].message.content
         }
 
     except Exception as e:
@@ -61,13 +58,13 @@ def chat_with_groq(user_message, history):
     return history, ""
 
 # =====================================================
-# Clear Chat Logic
+# Clear Chat
 # =====================================================
 def clear_chat():
     return [], ""
 
 # =====================================================
-# CSS (Visible in Light + Dark Mode)
+# CSS (Light + Dark mode safe)
 # =====================================================
 custom_css = """
 :root {
@@ -174,15 +171,15 @@ button:hover {
 """
 
 # =====================================================
-# Gradio UI (HF Compatible)
+# Gradio UI
 # =====================================================
-with gr.Blocks(css=custom_css, title="GROQ AI CHATBOT") as demo:
+with gr.Blocks(title="GROQ AI CHATBOT") as demo:
     gr.Markdown("# 🤖 GROQ AI CHATBOT")
     gr.Markdown(
         "<div class='subtitle'>Fast, reliable AI powered by Groq & LLaMA-3.3</div>"
     )
 
-    chatbot = gr.Chatbot(type="messages", elem_classes="chatbot")
+    chatbot = gr.Chatbot(elem_classes="chatbot")
     state = gr.State([])
 
     with gr.Row():
@@ -198,5 +195,8 @@ with gr.Blocks(css=custom_css, title="GROQ AI CHATBOT") as demo:
     txt.submit(chat_with_groq, [txt, state], [chatbot, txt])
     clear.click(clear_chat, outputs=[chatbot, txt], show_progress=False)
 
+# =====================================================
+# Launch (CSS moved here in Gradio 6)
+# =====================================================
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(css=custom_css)