Update app.py

app.py

@@ -1,13 +1,14 @@
-# yellowflash_lightmode.py
-# Full light mode UI for YellowFlash.ai with chat memory.
-# TESTING ONLY: hardcoded keys included.
+# yellowflash_darkmode_with_memory.py
+# Uses native gr.ChatInterface with a top-left model dropdown and top-right title.
+# Adds session-only chat memory (from history).
+# TESTING ONLY: hardcoded keys included. Do not publish.
 
 import time
 import requests
 import gradio as gr
 
 # ---------------------------
-# HARDCODED KEYS (TESTING)
+# HARDCODED KEYS (TESTING ONLY)
 # ---------------------------
 GEMINI_KEY = "AIzaSyAPfDiu2V_aD6un00qHt5bkISm6C0Pkx7o"
 GEMINI_URL = "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:generateContent"
@@ -16,6 +17,7 @@ GROQ_KEY = "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"
 GROQ_URL = "https://api.groq.com/openai/v1/chat/completions"
 GROQ_MODEL = "meta-llama/llama-4-scout-17b-16e-instruct"
 
+
 # ---------------------------
 # Helpers
 # ---------------------------
@@ -31,35 +33,42 @@ def post_with_retries(url, headers, payload, timeout=18, max_retries=2):
             time.sleep(0.5 + i)
     raise Exception("Max retries exceeded")
 
+
 def call_gemini(api_key, message, history):
-    headers = {"Content-Type":"application/json", "x-goog-api-key": api_key}
+    headers = {"Content-Type": "application/json", "x-goog-api-key": api_key}
+    # include chat history for memory
     contents = []
     for u, m in history:
-        contents.append({"role":"user","parts":[{"text":u}]})
-        contents.append({"role":"model","parts":[{"text":m}]})
-    contents.append({"role":"user","parts":[{"text":message}]})
+        contents.append({"role": "user", "parts": [{"text": u}]})
+        contents.append({"role": "model", "parts": [{"text": m}]})
+    contents.append({"role": "user", "parts": [{"text": message}]})
     payload = {"contents": contents}
+
     r = post_with_retries(GEMINI_URL, headers, payload)
     data = r.json()
-    return data.get("candidates",[{}])[0].get("content",{}).get("parts",[{}])[0].get("text","")
+    return data.get("candidates", [{}])[0].get("content", {}).get("parts", [{}])[0].get("text", "")
+
 
 def call_llama_via_groq(api_key, model, message, history):
     headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
+    # include chat history for memory
     msgs = []
     for u, m in history:
-        msgs.append({"role":"user","content":u})
-        msgs.append({"role":"assistant","content":m})
-    msgs.append({"role":"user","content":message})
+        msgs.append({"role": "user", "content": u})
+        msgs.append({"role": "assistant", "content": m})
+    msgs.append({"role": "user", "content": message})
     payload = {"model": model, "messages": msgs}
+
     r = post_with_retries(GROQ_URL, headers, payload)
     data = r.json()
     if "choices" in data and data["choices"]:
         ch = data["choices"][0]
         if isinstance(ch.get("message"), dict):
-            return ch["message"].get("content","")
-        return ch.get("text","")
+            return ch["message"].get("content", "")
+        return ch.get("text", "")
     return str(data)
 
+
 # ---------------------------
 # Chat function
 # ---------------------------
@@ -73,68 +82,29 @@ def chat_fn(message, history, model_choice):
         ans = f"Error: {e}"
     return ans
 
+
 # ---------------------------
-# CSS: Full Light Mode
+# Dark Mode CSS
 # ---------------------------
 css = """
-/* Full Light Mode */
-body, .gradio-container {
-    background: #ffffff !important;
-    color: #000000 !important;
-}
-
-/* Topbar */
-#topbar { 
-    display:flex; 
-    justify-content:space-between; 
-    align-items:center; 
-    padding:18px 28px; 
-    background:#ffffff !important; 
-    border-bottom:1px solid #e0e0e0; 
-}
-#title { 
-    font-weight:800; 
-    color:#ffcc33; 
-    font-size:20px; 
-}
-
-/* Dropdown */
-#model_dropdown .gr-dropdown { 
-    background:#ffffff !important; 
-    border:1px solid #d0d0d0 !important; 
-    color:#000000 !important; 
-    padding:10px 12px !important; 
-    border-radius:8px !important; 
-    width:260px !important; 
-    box-shadow:none !important; 
-}
-
-/* Chat area */
-.gradio-container .chat-interface .chatbot { 
-    background:#ffffff !important; 
-    color:#000000 !important; 
-    min-height: calc(100vh - 220px); 
-    border:1px solid #e0e0e0; 
-}
-
-/* Input textbox */
-textarea, .gr-textbox { 
-    background:#ffffff !important; 
-    color:#000000 !important; 
-    border:1px solid #d0d0d0 !important; 
-    border-radius:8px !important; 
-}
-
-/* Send button */
-.gr-button { 
-    background:#ffcc33 !important; 
-    color:#000000 !important; 
-    border:none !important; 
-    border-radius:10px !important; 
-    font-weight:700; 
-}
+/* topbar layout */
+#topbar { display:flex; justify-content:space-between; align-items:center;
+          padding:18px 28px; background:#0f0f0f; border-bottom:1px solid #1f1f1f; }
+#title { font-weight:800; color:#ffcc33; font-size:20px; }
+
+/* compact, flat dropdown look */
+#model_dropdown .gr-dropdown { background:#1a1a1a !important; border:1px solid #2b2b2b !important;
+    color:#ddd !important; padding:10px 12px !important; border-radius:8px !important; 
+    width:260px !important; box-shadow:none !important; }
+
+/* make ChatInterface chat area taller */
+.gradio-container .chat-interface .chatbot { min-height: calc(100vh - 220px); background:#111; color:#eee; }
+
+/* style send button */
+.gr-button { border-radius:10px !important; background:#2c2c3f !important; color:#fff !important; }
 """
 
+
 # ---------------------------
 # Build UI
 # ---------------------------
@@ -150,8 +120,8 @@ with gr.Blocks(css=css, title="⚡ YellowFlash.ai") as app:
 
     gr.ChatInterface(
         fn=chat_fn,
-        title="⚡ YellowFlash.ai",
-        description="fast as a lightning (inspired by Lord Fourth 'Namikaze Minato')",
+        title="⚡",
+        description="FAST AS LIGHTNING",
         additional_inputs=[model_dropdown],
     )