Update app.py

app.py

@@ -13,12 +13,12 @@ import emoji
 groq_key = os.getenv("GROQ_API_KEY")
 whisper_model = whisper.load_model("base")
 
-# Filter emojis
+# Emoji filtering
 allowed_emojis = {"😊", "😄", "👍", "🤖", "✨", "🎉", "💬", "🙌", "😎", "📢", "🧠", "✅"}
 def filter_emojis(text):
     return "".join(char if char not in emoji.EMOJI_DATA or char in allowed_emojis else "" for char in text)
 
-# Auto-save chat
+# Save/load
 CHAT_DIR = "saved_chats"
 os.makedirs(CHAT_DIR, exist_ok=True)
 
@@ -28,38 +28,38 @@ def save_chat_auto(history):
     title = " ".join(title.split()[:5]) or "Chat"
     timestamp = datetime.now().strftime("%b %d %Y %H-%M-%S")
     filename = f"{title} - {timestamp}.json"
-    path = os.path.join(CHAT_DIR, filename)
-    with open(path, "w", encoding="utf-8") as f:
+    with open(os.path.join(CHAT_DIR, filename), "w", encoding="utf-8") as f:
         json.dump(history, f, indent=2, ensure_ascii=False)
     return filename
 
 def list_saved_chats():
-    return sorted([f[:-5] for f in os.listdir(CHAT_DIR) if f.endswith(".json")])
+    return sorted([f for f in os.listdir(CHAT_DIR) if f.endswith(".json")])
 
-def load_chat(name):
+def load_chat_file(filename):
     try:
-        with open(os.path.join(CHAT_DIR, f"{name}.json"), "r", encoding="utf-8") as f:
+        with open(os.path.join(CHAT_DIR, filename), "r", encoding="utf-8") as f:
             history = json.load(f)
         chat_display = [(m["content"], None) if m["role"] == "user" else (None, m["content"]) for m in history]
         return chat_display, history
     except:
         return [], []
 
-# Chat with Groq
+# Chat handler
 def chat_with_groq(message, history):
     messages = [{"role": "system", "content": "You are Neobot – a helpful, professional assistant."}]
     messages += history + [{"role": "user", "content": message}]
-    
+
     headers = {"Authorization": f"Bearer {groq_key}", "Content-Type": "application/json"}
     payload = {"model": "llama3-70b-8192", "messages": messages}
-    
+
     res = requests.post("https://api.groq.com/openai/v1/chat/completions", headers=headers, json=payload)
     reply = res.json()["choices"][0]["message"]["content"]
     reply = filter_emojis(reply)
-    
+
     history += [{"role": "user", "content": message}, {"role": "assistant", "content": reply}]
     save_chat_auto(history)
-    return "", [(m["content"], None) if m["role"] == "user" else (None, m["content"]) for m in history], history, list_saved_chats()
+    chat_display = [(m["content"], None) if m["role"] == "user" else (None, m["content"]) for m in history]
+    return "", chat_display, history, list_saved_chats()
 
 # Transcribe audio
 def transcribe_audio(audio_path):
@@ -71,14 +71,15 @@ def transcribe_audio(audio_path):
         os.remove(temp_wav)
         return result["text"]
     except Exception:
-        return "❌ Error during transcription"
+        return "❌ Transcription failed"
 
 # UI
 with gr.Blocks(css="""
-body { background-color: #fff; font-family: 'Segoe UI', sans-serif; }
-.gr-button, .gr-textbox, .gr-dropdown { color: black !important; background: #fff !important; }
-textarea, input[type='text'] { border-radius: 20px; padding: 10px 15px; border: 1px solid #ccc; height: 48px; }
-.gr-button:hover { background: #eee !important; }
+body { background: white; font-family: 'Segoe UI', sans-serif; }
+.gr-button, .gr-textbox { background: #fff; color: #000; }
+textarea, input[type='text'] { border: 1px solid #ccc; border-radius: 20px; padding: 10px; height: 48px; }
+.sidebar { background: #f5f5f5; height: 100%; overflow-y: auto; padding: 10px; }
+.sidebar button { width: 100%; margin: 5px 0; text-align: left; border: 1px solid #ccc; border-radius: 5px; background: white; }
 """) as demo:
 
     state = gr.State([])
@@ -88,7 +89,7 @@ textarea, input[type='text'] { border-radius: 20px; padding: 10px 15px; border:
         with gr.Column(scale=1):
             gr.Markdown("<h2 style='text-align:center;'>Chats</h2>")
             new_chat = gr.Button("🆕 New Chat")
-            saved_list = gr.Dropdown(choices=list_saved_chats(), label="Click to load", interactive=True)
+            saved_chats_list = gr.Column(elem_id="chat_list")
 
         with gr.Column(scale=3):
             gr.Markdown("<h1 style='text-align:center;'>Neobot</h1>")
@@ -97,31 +98,35 @@ textarea, input[type='text'] { border-radius: 20px; padding: 10px 15px; border:
                 chat_input = gr.Textbox(placeholder="Type or speak here...", scale=8, show_label=False)
                 plus = gr.Button("➕")
                 mic = gr.Button("🎙️")
-            audio = gr.Audio(type="filepath", visible=False)
-            upload = gr.File(file_types=[".mp3", ".wav"], visible=False)
+            hidden_audio = gr.Audio(type="filepath", visible=False)
+            file_input = gr.File(file_types=[".mp3", ".wav"], visible=False)
 
     send = gr.Button("Send 🚀")
-    send.click(chat_with_groq, inputs=[chat_input, state], outputs=[chat_input, chatbot, state, saved_list])
-    chat_input.submit(chat_with_groq, inputs=[chat_input, state], outputs=[chat_input, chatbot, state, saved_list])
+    send.click(chat_with_groq, inputs=[chat_input, state], outputs=[chat_input, chatbot, state, saved_chats_list])
+    chat_input.submit(chat_with_groq, inputs=[chat_input, state], outputs=[chat_input, chatbot, state, saved_chats_list])
 
-    new_chat.click(lambda: ("", [], []), outputs=[chat_input, chatbot, state])
-    saved_list.change(load_chat, inputs=[saved_list], outputs=[chatbot, state])
+    def refresh_chat_list():
+        chats = list_saved_chats()
+        return [gr.Button(value=chat, scale=1) for chat in chats]
 
-    demo.load(fn=list_saved_chats, outputs=[saved_list])
+    def load_chat_btn(filename):
+        return load_chat_file(filename)
 
-    plus.click(lambda: gr.update(visible=True), outputs=[upload])
-    upload.change(fn=transcribe_audio, inputs=upload, outputs=chat_input)
-
-    def toggle_recording(state):
-        if not state:
-            return gr.update(value="Recording..."), True, gr.update(visible=True)
+    def start_stop_recording(is_recording):
+        if not is_recording:
+            return "Recording...", True, gr.update(visible=True)
         else:
             return "", False, gr.update(visible=False)
 
-    def handle_transcription(audio_path):
-        return transcribe_audio(audio_path), gr.update(visible=False)
+    def handle_audio(audio):
+        return transcribe_audio(audio), gr.update(visible=False)
+
+    new_chat.click(lambda: ("", [], []), outputs=[chat_input, chatbot, state])
+    plus.click(lambda: gr.update(visible=True), outputs=[file_input])
+    file_input.change(transcribe_audio, inputs=file_input, outputs=chat_input)
+    mic.click(start_stop_recording, inputs=recording, outputs=[chat_input, recording, hidden_audio])
+    hidden_audio.change(handle_audio, inputs=hidden_audio, outputs=[chat_input, hidden_audio])
 
-    mic.click(toggle_recording, inputs=recording, outputs=[chat_input, recording, audio])
-    audio.change(handle_transcription, inputs=[audio], outputs=[chat_input, audio])
+    demo.load(refresh_chat_list, outputs=[saved_chats_list])
 
 demo.launch()