Update app.py

app.py

@@ -2,139 +2,183 @@ import os
 import gradio as gr
 from groq import Groq
 
-# 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})
+# =====================================================
+# Groq Client Initialization
+# =====================================================
+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]
+    ]
 
     try:
-        chat_completion = client.chat.completions.create(
+        completion = client.chat.completions.create(
             model="llama-3.3-70b-versatile",
             messages=messages,
         )
-
-        assistant_response = chat_completion.choices[0].message.content
-        chat_history.append((user_input, assistant_response))
-
+        history[-1] = {
+            "role": "assistant",
+            "content": completion.choices[0].message.content
+        }
     except Exception as e:
-        chat_history.append((user_input, f"❌ Error: {e}"))
+        history[-1] = {
+            "role": "assistant",
+            "content": f"⚠️ Error: {str(e)}"
+        }
+
+    return history, ""
+
+# =====================================================
+# Clear Chat
+# =====================================================
+def clear_chat():
+    return [], ""
+
+# =====================================================
+# SAFE CSS (Fixes Vertical Text Bug)
+# =====================================================
+custom_css = """
+/* RESET ONLY INSIDE CHATBOT */
+.chatbot * {
+  all: unset;
+}
 
-    return chat_history, ""
+/* RESTORE MESSAGE STRUCTURE */
+.chatbot .message,
+.chatbot .message * {
+  all: revert;
+}
 
-# UI Styling
-custom_css = """
-.gradio-container {
-    max-width: 1000px;
-    margin: auto;
-    padding: 16px;
+/* COLOR SYSTEM */
+:root {
+  --bg: #0b0f19;
+  --surface: #111827;
+  --user: #2563eb;
+  --assistant: #1f2937;
+  --text: #f9fafb;
+  --muted: #9ca3af;
+  --border: #1f2937;
 }
 
 body {
-    background: #f8f9fa;
-    color: #1f2937;
-    font-family: 'Inter', sans-serif;
+  background-color: var(--bg);
 }
 
-h1 {
-    color: #0b4f99;
-    font-weight: bold;
-    font-size: 2rem;
-    text-align: center;
-    margin-bottom: 12px;
+.gradio-container {
+  max-width: 1100px !important;
+  margin: auto;
+  padding: 24px;
+  overflow-x: hidden !important;
+  font-family: system-ui, sans-serif;
+}
+
+/* CHAT WINDOW */
+.chatbot {
+  background: var(--surface);
+  border-radius: 16px;
+  padding: 16px;
+  height: 70vh;
+  width: 100% !important;
+  border: 1px solid var(--border);
+  overflow-y: auto;
 }
 
-#chatbot {
-    background: #ffffff;
-    border: 1px solid #d1d5db;
-    border-radius: 12px;
-    padding: 12px;
-    box-shadow: 0 2px 8px rgba(0,0,0,0.1);
+/* MESSAGES */
+.chatbot .message {
+  display: block !important;
+  max-width: 72%;
+  padding: 12px 14px;
+  border-radius: 14px;
+  margin-bottom: 10px;
+  line-height: 1.6;
+  font-size: 15px;
+  white-space: pre-wrap !important;
+  word-break: normal !important;
 }
 
-.message.user {
-    background: #0b4f99;
-    color: #ffffff;
-    border-radius: 16px;
-    padding: 8px;
+.chatbot .message.user {
+  background: var(--user);
+  color: white;
+  margin-left: auto;
 }
 
-.message.bot {
-    background: #e5e7eb;
-    color: #1f2937;
-    border-radius: 16px;
-    padding: 8px;
+.chatbot .message.bot {
+  background: var(--assistant);
+  color: var(--text);
+  margin-right: auto;
 }
 
+/* INPUT */
 textarea {
-    width: 100%;
-    border: 2px solid #6b7280;
-    border-radius: 8px;
-    padding: 8px;
-    font-size: 1rem;
-    color: #1f2937;
+  background: var(--surface) !important;
+  color: var(--text) !important;
+  border: 1px solid var(--border) !important;
+  border-radius: 12px !important;
+  padding: 12px !important;
+  font-size: 15px;
 }
 
+/* BUTTONS */
 button {
-    background: #0b4f99;
-    color: #ffffff;
-    font-weight: bold;
-    border-radius: 8px;
-    padding: 12px 20px;
+  background: var(--user) !important;
+  color: white !important;
+  border-radius: 12px !important;
+  padding: 10px 16px !important;
+  font-weight: 600 !important;
 }
-button:hover, button:focus {
-    background: #093c77;
+
+/* REMOVE FLOATING ICONS */
+.chatbot-toolbar,
+.chatbot-actions,
+button[aria-label="Copy"],
+button[aria-label="Share"],
+button[aria-label="Clear"] {
+  display: none !important;
 }
 """
 
-with gr.Blocks(css=custom_css) as demo:
-    gr.Markdown("<h1>🌐 Groq ChatBot</h1>")
-    chatbot = gr.Chatbot(elem_id="chatbot", height=450)
+# =====================================================
+# 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</div>")
+
+    chatbot = gr.Chatbot(elem_classes="chatbot")
+    state = gr.State([])
 
     with gr.Row():
-        msg = gr.Textbox(
-            placeholder="💬 Type your message here...",
+        txt = gr.Textbox(
+            placeholder="Type your message…",
             show_label=False,
-            scale=4,
+            scale=5,
         )
         send = gr.Button("Send", scale=1)
+        clear = gr.Button("Clear", scale=1)
 
-    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)
+    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)
 
-demo.launch()
+# =====================================================
+# Launch
+# =====================================================
+if __name__ == "__main__":
+    demo.launch(css=custom_css)