Update app.py

app.py

@@ -1,116 +1,90 @@
-import os
 import gradio as gr
-import whisper
-import requests
-import json, uuid, re
+import openai
+import os
+import json
+import re
 from datetime import datetime
-from pydub import AudioSegment
-import emoji
-
-# Keys and model
-GROQ_KEY = os.getenv("GROQ_API_KEY")
-MODEL = whisper.load_model("base")
-
-# Emoji filter
-ALLOWED = {"😊", "😄", "👍", "🤖", "✨", "🎉", "💬", "🙌", "😎", "📢", "🧠", "✅"}
-def filter_emojis(text):
-    return "".join(c if c not in emoji.EMOJI_DATA or c in ALLOWED else "" for c in text)
-
-# Save system
-CHAT_DIR = "saved_chats"
-os.makedirs(CHAT_DIR, exist_ok=True)
-
-def save_chat_manual(history):
-    if not history: return
-    prompt = re.sub(r"[^\w\s]", "", history[-2]["content"])[:30] or "Chat"
-    timestamp = datetime.now().strftime("%b %d %Y %H-%M-%S")
-    fname = f"{prompt} - {timestamp} - {uuid.uuid4().hex[:6]}.json"
-    with open(os.path.join(CHAT_DIR, fname), "w", encoding="utf-8") as f:
+
+# 🔐 Groq API setup
+os.environ["OPENAI_API_KEY"] = "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"  # Replace with your actual key
+openai.api_key = os.environ["OPENAI_API_KEY"]
+openai.api_base = "https://api.groq.com/openai/v1"
+
+# 💬 Chat function
+def chat_with_groq(message, history):
+    messages = [{"role": "system", "content": "You are JAWERIA'SBOT 🤖 – cheerful, emoji-savvy, and sleek."}]
+    messages += history
+    messages.append({"role": "user", "content": message})
+    response = openai.ChatCompletion.create(model="llama3-70b-8192", messages=messages)
+    reply = response["choices"][0]["message"]["content"]
+    history.append({"role": "user", "content": message})
+    history.append({"role": "assistant", "content": reply})
+    return "", history, history
+
+# 💾 Save chat session with readable filename
+def save_session(history):
+    prompt = next((m["content"] for m in history if m["role"] == "user"), "chat")
+    title = re.sub(r"[^\w\s]", "", prompt).strip()
+    title = " ".join(title.split()[:6])
+    timestamp = datetime.now().strftime("%b %d %Y %H-%M")
+    filename = f"{title} - {timestamp}.json"
+    with open(filename, "w", encoding="utf-8") as f:
         json.dump(history, f, indent=2, ensure_ascii=False)
+    return f"✅ Chat saved as: {filename[:-5]}"
 
-def list_saved_chats():
-    return sorted(os.listdir(CHAT_DIR))
+# 📜 List saved chats for dropdown
+def list_saved_files():
+    files = [f[:-5] for f in os.listdir() if f.endswith(".json")]
+    return sorted(files)
 
-def load_chat(fname):
+# 📥 Load selected chat
+def load_chat(name):
+    filename = f"{name}.json"
     try:
-        with open(os.path.join(CHAT_DIR, fname), "r", encoding="utf-8") as f:
-            hist = json.load(f)
-        return hist, hist
-    except:
-        return [], []
-
-# LLM chat
-def chat_with_groq(user_input, history):
-    try:
-        messages = [{"role": "system", "content": "You are Neobot, a helpful assistant."}]
-        messages += history + [{"role": "user", "content": user_input}]
-        res = requests.post(
-            "https://api.groq.com/openai/v1/chat/completions",
-            headers={"Authorization": f"Bearer {GROQ_KEY}", "Content-Type": "application/json"},
-            json={"model": "llama3-70b-8192", "messages": messages}
-        )
-        reply = res.json()["choices"][0]["message"]["content"]
-        reply = filter_emojis(reply)
-        history += [{"role": "user", "content": user_input}, {"role": "assistant", "content": reply}]
-        return "", history, history
+        with open(filename, "r", encoding="utf-8") as f:
+            history = json.load(f)
+        return history, history, f"✅ Loaded {name}"
     except Exception as e:
-        error = f"❌ Error: {e}"
-        history += [{"role": "assistant", "content": error}]
-        return "", history, history
+        return [], [], f"❌ Could not load {name}: {e}"
 
-# Transcribe audio
-def transcribe(audio_path):
-    if not audio_path: return ""
-    try:
-        temp_wav = f"{uuid.uuid4()}.wav"
-        AudioSegment.from_file(audio_path).export(temp_wav, format="wav")
-        result = MODEL.transcribe(temp_wav)
-        os.remove(temp_wav)
-        return result["text"]
-    except:
-        return "❌ Transcription failed"
-
-# UI
+# 🌒 Dark-themed UI
 with gr.Blocks(css="""
-body { background-color: white; font-family: 'Segoe UI', sans-serif; }
-.gr-button { background-color: #f7f7f7; border-radius: 8px; }
-.gr-chatbot { border: 1px solid #ccc; border-radius: 12px; }
-.scroll-sidebar button { margin: 2px 0; width: 100%; text-align: left; border-radius: 6px; background: white; transition: background 0.2s; }
-.scroll-sidebar button:hover { background: #eee; }
+body {background-color: #111 !important; color: white;}
+.gr-chatbot-message {background-color: #222 !important; color: white !important; border-radius: 8px; padding: 6px;}
+input[type='text'] {background-color: #222 !important; color: white !important; border: 1px solid #555;}
+.gr-button {background-color: #007acc !important; color: white !important; border-radius: 6px;}
 """) as demo:
 
     state = gr.State([])
+
+    gr.Markdown("<h1 style='text-align:center; color:#00ccff;'>✨ JAWERIA'SBOT 🤖</h1>")
+    gr.Markdown("<div style='text-align:center;'>Ask boldly, explore endlessly. JAWERIA'SBOT is always listening 🌌</div>")
+
+    chatbot = gr.Chatbot(type="messages")  # 🔕 No label shown
+
+    with gr.Row():
+        chat_input = gr.Textbox(placeholder="Type your message... 😊", show_label=False, scale=8)
+        send_btn = gr.Button("Send 🚀", scale=2)
+
     with gr.Row():
-        with gr.Column(scale=1):
-            gr.Markdown("### 📁 Saved Chats")
-            chat_list = gr.Column(elem_id="scroll-sidebar")  # Without 'scroll'
-            save_btn = gr.Button("💾 Save Chat")
-            refresh_btn = gr.Button("🔄 Refresh")
-        with gr.Column(scale=3):
-            gr.Markdown("## 🤖 Neobot Chat")
-            bot = gr.Chatbot(height=400)
-            with gr.Row():
-                input_box = gr.Textbox(placeholder="Type a message or record one...", scale=6, show_label=False)
-                send_btn = gr.Button("🚀 Send")
-            voice_input = gr.Audio(label="🎙️ Record Voice", type="filepath")
-
-    # Chat actions
-    send_btn.click(chat_with_groq, inputs=[input_box, state], outputs=[input_box, bot, state])
-    input_box.submit(chat_with_groq, inputs=[input_box, state], outputs=[input_box, bot, state])
-    voice_input.change(transcribe, inputs=voice_input, outputs=[input_box])
-    save_btn.click(lambda hist: save_chat_manual(hist), inputs=[state], outputs=[])
-
-    # Sidebar buttons
-    def render_chat_buttons():
-        btns = []
-        for f in list_saved_chats():
-            def load_fn(fname=f): return load_chat(fname)
-            btn = gr.Button(f[:-5])
-            btn.click(load_fn, outputs=[bot, state])
-            btns.append(btn)
-        return btns
-
-    refresh_btn.click(render_chat_buttons, outputs=[chat_list])
-    demo.load(render_chat_buttons, outputs=[chat_list])
-
-demo.launch()
+        new_chat_btn = gr.Button("🆕 Start New Chat")
+        save_btn = gr.Button("💾 Save Chat")
+
+    save_msg = gr.Markdown()
+    load_msg = gr.Markdown()
+
+    saved_files_dropdown = gr.Dropdown(label="📂 Saved Chats", choices=list_saved_files(), interactive=True, allow_custom_value=False)
+    load_btn = gr.Button("📥 Load Selected Chat")
+
+    # 🚀 Message interactions
+    send_btn.click(chat_with_groq, inputs=[chat_input, state], outputs=[chat_input, chatbot, state])
+    chat_input.submit(chat_with_groq, inputs=[chat_input, state], outputs=[chat_input, chatbot, state])
+
+    new_chat_btn.click(fn=lambda: ("", [], []), outputs=[chat_input, chatbot, state])
+    
+    save_btn.click(fn=save_session, inputs=[state], outputs=[save_msg])
+    save_btn.click(fn=list_saved_files, outputs=[saved_files_dropdown])  # 🔁 Refresh dropdown after saving
+
+    load_btn.click(fn=load_chat, inputs=[saved_files_dropdown], outputs=[chatbot, state, load_msg])
+
+demo.launch()