Update app.py

app.py

@@ -2,213 +2,139 @@ import os
 import gradio as gr
 from groq import Groq
 
-# =====================================================
-# Groq Client Initialization (GLOBAL)
-# =====================================================
-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()
-
-# =====================================================
-# Chat Logic (Gradio 6 compatible)
-# =====================================================
-def chat_with_groq(user_message, history):
-    if not user_message or not user_message.strip():
-        return history, ""
-
-    history.append({"role": "user", "content": user_message})
-    history.append({"role": "assistant", "content": "⚡ Groq is thinking..."})
-
-    messages = [
-        {"role": msg["role"], "content": msg["content"]}
-        for msg in history[:-1]
-    ]
+# Initialize Groq client
+client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
+
+def normalize_chat_history(chat_history):
+    """
+    Converts chat_history into list of tuples (user, assistant)
+    Works with both tuple and dict formats returned by Gradio.
+    """
+    normalized = []
+    for item in chat_history:
+        if isinstance(item, dict):
+            # If it's a dict, convert accordingly
+            if item.get("role") == "user":
+                normalized.append((item.get("content"), ""))
+            elif item.get("role") == "assistant":
+                # merge assistant into last user entry
+                if normalized:
+                    last_user, _ = normalized[-1]
+                    normalized[-1] = (last_user, item.get("content"))
+        elif isinstance(item, (list, tuple)) and len(item) == 2:
+            normalized.append((item[0], item[1]))
+    return normalized
+
+def chat_with_groq(user_input, chat_history):
+    if not user_input.strip():
+        return chat_history, ""
+
+    chat_history = normalize_chat_history(chat_history)
+
+    messages = []
+    for u, a in chat_history:
+        if u:
+            messages.append({"role": "user", "content": u})
+        if a:
+            messages.append({"role": "assistant", "content": a})
+
+    messages.append({"role": "user", "content": user_input})
 
     try:
-        completion = client.chat.completions.create(
+        chat_completion = client.chat.completions.create(
             model="llama-3.3-70b-versatile",
             messages=messages,
         )
-        history[-1] = {
-            "role": "assistant",
-            "content": completion.choices[0].message.content
-        }
-    except Exception as e:
-        history[-1] = {
-            "role": "assistant",
-            "content": f"⚠️ Error: {str(e)}"
-        }
-
-    return history, ""
-
-# =====================================================
-# Clear Chat
-# =====================================================
-def clear_chat():
-    return [], ""
-
-# =====================================================
-# CSS (Light/Dark + Gradio 6 UI FIX)
-# =====================================================
-custom_css = """
-/* ---------- Color System ---------- */
-: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;
-  }
-}
+        assistant_response = chat_completion.choices[0].message.content
+        chat_history.append((user_input, assistant_response))
 
-/* ---------- Base Layout ---------- */
-body {
-  background-color: var(--bg);
-}
+    except Exception as e:
+        chat_history.append((user_input, f"❌ Error: {e}"))
 
-.gradio-container {
-  max-width: 1100px !important;
-  margin: auto;
-  padding: 24px;
-  overflow-x: hidden !important;
-  font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, sans-serif;
-}
+    return chat_history, ""
 
-/* ---------- Typography ---------- */
-h1 {
-  color: var(--text);
-  text-align: center;
-  font-size: 2rem;
-  margin-bottom: 6px;
+# UI Styling
+custom_css = """
+.gradio-container {
+    max-width: 1000px;
+    margin: auto;
+    padding: 16px;
 }
 
-.subtitle {
-  color: var(--muted);
-  text-align: center;
-  margin-bottom: 20px;
+body {
+    background: #f8f9fa;
+    color: #1f2937;
+    font-family: 'Inter', sans-serif;
 }
 
-/* ---------- Chatbot ---------- */
-.chatbot {
-  background: var(--surface);
-  border-radius: 16px;
-  padding: 16px;
-  height: 70vh;
-  width: 100% !important;
-  border: 1px solid var(--border);
-  overflow-y: auto;
+h1 {
+    color: #0b4f99;
+    font-weight: bold;
+    font-size: 2rem;
+    text-align: center;
+    margin-bottom: 12px;
 }
 
-/* ---------- Messages ---------- */
-.message {
-  max-width: 72%;
-  padding: 12px 14px;
-  border-radius: 14px;
-  margin-bottom: 10px;
-  line-height: 1.6;
-  font-size: 15px;
-  box-sizing: border-box;
+#chatbot {
+    background: #ffffff;
+    border: 1px solid #d1d5db;
+    border-radius: 12px;
+    padding: 12px;
+    box-shadow: 0 2px 8px rgba(0,0,0,0.1);
 }
 
 .message.user {
-  background: var(--user);
-  color: white;
-  margin-left: auto;
-  border-bottom-right-radius: 4px;
+    background: #0b4f99;
+    color: #ffffff;
+    border-radius: 16px;
+    padding: 8px;
 }
 
 .message.bot {
-  background: var(--assistant);
-  color: var(--text);
-  margin-right: auto;
-  border-bottom-left-radius: 4px;
+    background: #e5e7eb;
+    color: #1f2937;
+    border-radius: 16px;
+    padding: 8px;
 }
 
-/* ---------- Input ---------- */
 textarea {
-  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::placeholder {
-  color: var(--muted);
+    width: 100%;
+    border: 2px solid #6b7280;
+    border-radius: 8px;
+    padding: 8px;
+    font-size: 1rem;
+    color: #1f2937;
 }
 
-/* ---------- Buttons ---------- */
 button {
-  background: var(--user) !important;
-  color: white !important;
-  border-radius: 12px !important;
-  padding: 10px 16px !important;
-  font-weight: 600 !important;
+    background: #0b4f99;
+    color: #ffffff;
+    font-weight: bold;
+    border-radius: 8px;
+    padding: 12px 20px;
 }
-
-button:hover {
-  opacity: 0.9;
-}
-
-/* ---------- GRADIO 6 FIXES ---------- */
-.chatbot-toolbar,
-.chatbot-actions,
-button[aria-label="Copy"],
-button[aria-label="Share"],
-button[aria-label="Clear"] {
-  display: none !important;
-}
-
-.chatbot .scroll-to-bottom {
-  right: 16px !important;
-  bottom: 80px !important;
+button:hover, button:focus {
+    background: #093c77;
 }
 """
 
-# =====================================================
-# Gradio UI
-# =====================================================
-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(elem_classes="chatbot")
-    state = gr.State([])
+with gr.Blocks(css=custom_css) as demo:
+    gr.Markdown("<h1>🌐 Groq ChatBot</h1>")
+    chatbot = gr.Chatbot(elem_id="chatbot", height=450)
 
     with gr.Row():
-        txt = gr.Textbox(
-            placeholder="Type your message…",
+        msg = gr.Textbox(
+            placeholder="💬 Type your message here...",
             show_label=False,
-            scale=5,
+            scale=4,
         )
         send = gr.Button("Send", scale=1)
-        clear = gr.Button("Clear", scale=1)
 
-    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)
+    clear = gr.Button("Clear Chat")
+
+    send.click(chat_with_groq, [msg, chatbot], [chatbot, msg])
+    msg.submit(chat_with_groq, [msg, chatbot], [chatbot, msg])
+    clear.click(lambda: [], None, chatbot)
 
-# =====================================================
-# Launch (Gradio 6)
-# =====================================================
-if __name__ == "__main__":
-    demo.launch(css=custom_css)
+demo.launch()