Update app.py

app.py

@@ -2,26 +2,41 @@ import os
 import gradio as gr
 from groq import Groq
 
-# -----------------------------
-# Groq Client Initialization
-# -----------------------------
+# =====================================================
+# Groq Client Initialization (GLOBAL – IMPORTANT)
+# =====================================================
+def get_groq_client():
+    api_key = os.getenv("GROQ_API_KEY")
+    if not api_key:
+        raise RuntimeError("GROQ_API_KEY not set")
+    return Groq(api_key=api_key)
+
+client = get_groq_client()  # ✅ must be global
+
+# =====================================================
+# Model Inference Logic (HF-safe, Gradio v4+)
+# =====================================================
 def chat_with_groq(user_message, history):
     if not user_message or not user_message.strip():
         return history, ""
 
-    # Add temporary thinking message
-    history.append({"role": "assistant", "content": "⚡ Groq is thinking..."})
-
-    messages = []
+    # Add user message
+    history.append({
+        "role": "user",
+        "content": user_message
+    })
 
-    for msg in history[:-1]:
-        if isinstance(msg, dict):
-            messages.append({
-                "role": msg["role"],
-                "content": msg["content"]
-            })
+    # Add temporary thinking message
+    history.append({
+        "role": "assistant",
+        "content": "⚡ Groq is thinking..."
+    })
 
-    messages.append({"role": "user", "content": user_message})
+    # Prepare messages for Groq
+    messages = [
+        {"role": msg["role"], "content": msg["content"]}
+        for msg in history[:-1]
+    ]
 
     try:
         chat_completion = client.chat.completions.create(
@@ -31,6 +46,7 @@ def chat_with_groq(user_message, history):
 
         response = chat_completion.choices[0].message.content
 
+        # Replace thinking message
         history[-1] = {
             "role": "assistant",
             "content": response
@@ -44,105 +60,143 @@ def chat_with_groq(user_message, history):
 
     return history, ""
 
-# -----------------------------
+# =====================================================
 # Clear Chat Logic
-# -----------------------------
+# =====================================================
 def clear_chat():
     return [], ""
 
-# -----------------------------
-# Neon CSS
-# -----------------------------
+# =====================================================
+# CSS (Visible in Light + Dark Mode)
+# =====================================================
 custom_css = """
+:root {
+  --bg: #f9fafb;
+  --surface: #ffffff;
+  --user: #2563eb;
+  --assistant: #e5e7eb;
+  --text: #111827;
+  --muted: #6b7280;
+  --border: #d1d5db;
+}
+
+@media (prefers-color-scheme: dark) {
+  :root {
+    --bg: #0b0f19;
+    --surface: #111827;
+    --assistant: #1f2937;
+    --text: #f9fafb;
+    --muted: #9ca3af;
+    --border: #1f2937;
+  }
+}
+
 body {
-    background: radial-gradient(circle at top, #020617, #020617);
+  background-color: var(--bg);
 }
 
 .gradio-container {
-    max-width: 900px !important;
-    margin: auto;
+  max-width: 1100px !important;
+  margin: auto;
+  padding: 24px;
+  font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, sans-serif;
 }
 
 h1 {
-    text-align: center;
-    font-size: 2.6rem;
-    background: linear-gradient(90deg, #38bdf8, #22d3ee, #818cf8);
-    -webkit-background-clip: text;
-    -webkit-text-fill-color: transparent;
+  color: var(--text);
+  text-align: center;
+  font-size: 2rem;
+  margin-bottom: 6px;
+}
+
+.subtitle {
+  color: var(--muted);
+  text-align: center;
+  margin-bottom: 20px;
 }
 
 .chatbot {
-    background: rgba(15, 23, 42, 0.75);
-    backdrop-filter: blur(16px);
-    border-radius: 18px;
-    box-shadow: 0 0 35px rgba(56, 189, 248, 0.2);
+  background: var(--surface);
+  border-radius: 16px;
+  padding: 16px;
+  height: 70vh;
+  overflow-y: auto;
+  border: 1px solid var(--border);
 }
 
 .message {
-    animation: fadeSlide 0.35s ease-in-out;
+  max-width: 72%;
+  padding: 12px 14px;
+  border-radius: 14px;
+  margin-bottom: 10px;
+  line-height: 1.6;
+  font-size: 15px;
+}
+
+.message.user {
+  background: var(--user);
+  color: white;
+  margin-left: auto;
+  border-bottom-right-radius: 4px;
+}
+
+.message.bot {
+  background: var(--assistant);
+  color: var(--text);
+  margin-right: auto;
+  border-bottom-left-radius: 4px;
 }
 
 textarea {
-    background: rgba(2, 6, 23, 0.85) !important;
-    color: #e5e7eb !important;
-    border: 1px solid #38bdf8 !important;
-    border-radius: 14px !important;
+  background: var(--surface) !important;
+  color: var(--text) !important;
+  border: 1px solid var(--border) !important;
+  border-radius: 12px !important;
+  padding: 12px !important;
+  font-size: 15px;
 }
 
-textarea:focus {
-    box-shadow: 0 0 18px #38bdf8 !important;
+textarea::placeholder {
+  color: var(--muted);
 }
 
 button {
-    background: linear-gradient(135deg, #22d3ee, #3b82f6) !important;
-    color: #020617 !important;
-    font-weight: 800 !important;
-    border-radius: 14px !important;
-    box-shadow: 0 0 22px rgba(56, 189, 248, 0.7);
+  background: var(--user) !important;
+  color: white !important;
+  border-radius: 12px !important;
+  padding: 10px 16px !important;
+  font-weight: 600 !important;
 }
 
 button:hover {
-    transform: scale(1.05);
-    box-shadow: 0 0 35px rgba(56, 189, 248, 1);
+  opacity: 0.9;
 }
 """
 
-# -----------------------------
-# Gradio UI
-# -----------------------------
-with gr.Blocks(css=custom_css, title="Groq ChatBot") as demo:
-    gr.Markdown("# ⚡ GROQ CHATBOT")
-    gr.Markdown("### Ultra-fast AI • Cyber UI • LLaMA 3.3")
+# =====================================================
+# Gradio UI (HF Compatible)
+# =====================================================
+with gr.Blocks(css=custom_css, 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(elem_classes="chatbot")
+    chatbot = gr.Chatbot(type="messages", elem_classes="chatbot")
     state = gr.State([])
 
     with gr.Row():
         txt = gr.Textbox(
-            placeholder="Ask anything…",
+            placeholder="Type your message…",
             show_label=False,
-            scale=4,
+            scale=5,
         )
-        send = gr.Button("🚀 Send", scale=1)
-        clear = gr.Button("🧹 Clear", scale=1)
+        send = gr.Button("Send", scale=1)
+        clear = gr.Button("Clear", scale=1)
 
-    send.click(
-        chat_with_groq,
-        inputs=[txt, state],
-        outputs=[chatbot, txt],
-    )
-
-    txt.submit(
-        chat_with_groq,
-        inputs=[txt, state],
-        outputs=[chatbot, txt],
-    )
-
-    clear.click(
-        clear_chat,
-        outputs=[chatbot, txt],
-        show_progress=False,
-    )
+    send.click(chat_with_groq, [txt, state], [chatbot, txt])
+    txt.submit(chat_with_groq, [txt, state], [chatbot, txt])
+    clear.click(clear_chat, outputs=[chatbot, txt], show_progress=False)
 
 if __name__ == "__main__":
     demo.launch()