Update app.py

app.py

@@ -1,13 +1,13 @@
 import os
 import gradio as gr
-import requests
 import whisper
+import requests
 import json, uuid, re
 from datetime import datetime
 from pydub import AudioSegment
 import emoji
 
-# Load keys and model
+# Keys and model
 GROQ_KEY = os.getenv("GROQ_API_KEY")
 MODEL = whisper.load_model("base")
 
@@ -16,18 +16,19 @@ 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 chats
+# Save system
 CHAT_DIR = "saved_chats"
 os.makedirs(CHAT_DIR, exist_ok=True)
 
-def save_chat(history):
-    title = re.sub(r"[^\w\s]", "", history[-2]["content"])[:30] or "Chat"
+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"{title} - {timestamp} - {uuid.uuid4().hex[:6]}.json"
+    fname = f"{prompt} - {timestamp} - {uuid.uuid4().hex[:6]}.json"
     with open(os.path.join(CHAT_DIR, fname), "w", encoding="utf-8") as f:
         json.dump(history, f, indent=2, ensure_ascii=False)
 
-def list_chats():
+def list_saved_chats():
     return sorted(os.listdir(CHAT_DIR))
 
 def load_chat(fname):
@@ -35,7 +36,7 @@ def load_chat(fname):
         with open(os.path.join(CHAT_DIR, fname), "r", encoding="utf-8") as f:
             hist = json.load(f)
         return hist, hist
-    except Exception:
+    except:
         return [], []
 
 # LLM chat
@@ -51,7 +52,6 @@ def chat_with_groq(user_input, history):
         reply = res.json()["choices"][0]["message"]["content"]
         reply = filter_emojis(reply)
         history += [{"role": "user", "content": user_input}, {"role": "assistant", "content": reply}]
-        save_chat(history)
         return "", history, history
     except Exception as e:
         error = f"❌ Error: {e}"
@@ -71,29 +71,46 @@ def transcribe(audio_path):
         return "❌ Transcription failed"
 
 # UI
-with gr.Blocks() as demo:
+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; }
+""") as demo:
+
     state = gr.State([])
     with gr.Row():
         with gr.Column(scale=1):
-            gr.Markdown("### Saved Chats")
-            dropdown = gr.Dropdown(choices=list_chats(), label="Load chat")
-            load_btn = gr.Button("📂 Load")
-            new_btn = gr.Button("🆕 New Chat")
+            gr.Markdown("### 📁 Saved Chats")
+            chat_list = gr.Column(elem_id="scroll-sidebar", scroll=True)
+            save_btn = gr.Button("💾 Save Chat")
+            refresh_btn = gr.Button("🔄 Refresh")
         with gr.Column(scale=3):
-            gr.Markdown("## 🤖 Neobot")
+            gr.Markdown("## 🤖 Neobot Chat")
             bot = gr.Chatbot(height=400)
-            input_box = gr.Textbox(placeholder="Type or upload voice...", scale=8, show_label=False)
-            send = gr.Button("🚀 Send")
-            audio = gr.Audio(label="🎤 Voice Input", type="filepath")
-            upload = gr.File(file_types=[".mp3", ".wav"], label="Upload Audio")
+            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")
 
-    # Events
-    send.click(chat_with_groq, inputs=[input_box, state], outputs=[input_box, bot, state])
+    # 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])
-    audio.change(transcribe, inputs=audio, outputs=[input_box])
-    upload.change(transcribe, inputs=upload, outputs=[input_box])
-    load_btn.click(lambda fname: load_chat(fname), inputs=[dropdown], outputs=[bot, state])
-    new_btn.click(lambda: ("", [], []), outputs=[input_box, bot, state])
-    demo.load(lambda: gr.update(choices=list_chats()), outputs=[dropdown])
+    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()