Update app.py

app.py

@@ -41,25 +41,30 @@ def load_chat_file(filename):
             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:
+    except Exception as e:
+        print(f"Load error: {e}")
         return [], []
 
 # 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}]
+    try:
+        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}
+        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)
+        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)
-    chat_display = [(m["content"], None) if m["role"] == "user" else (None, m["content"]) for m in history]
-    return "", chat_display, history, list_saved_chats()
+        history += [{"role": "user", "content": message}, {"role": "assistant", "content": reply}]
+        save_chat_auto(history)
+        chat_display = [(m["content"], None) if m["role"] == "user" else (None, m["content"]) for m in history]
+        return "", chat_display, history, refresh_chat_list()
+    except Exception as e:
+        print(f"Chat error: {e}")
+        return "", [(None, "❌ Error occurred while responding.")], history, refresh_chat_list()
 
 # Transcribe audio
 def transcribe_audio(audio_path):
@@ -89,7 +94,7 @@ textarea, input[type='text'] { border: 1px solid #ccc; border-radius: 20px; padd
         with gr.Column(scale=1):
             gr.Markdown("<h2 style='text-align:center;'>Chats</h2>")
             new_chat = gr.Button("🆕 New Chat")
-            saved_chats_list = gr.Column(elem_id="chat_list")
+            saved_chats_container = gr.Column(elem_id="chat_list")
 
         with gr.Column(scale=3):
             gr.Markdown("<h1 style='text-align:center;'>Neobot</h1>")
@@ -98,35 +103,35 @@ textarea, input[type='text'] { border: 1px solid #ccc; border-radius: 20px; padd
                 chat_input = gr.Textbox(placeholder="Type or speak here...", scale=8, show_label=False)
                 plus = gr.Button("➕")
                 mic = gr.Button("🎙️")
-            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_chats_list])
-    chat_input.submit(chat_with_groq, inputs=[chat_input, state], outputs=[chat_input, chatbot, state, saved_chats_list])
+    send.click(chat_with_groq, inputs=[chat_input, state], outputs=[chat_input, chatbot, state, saved_chats_container])
+    chat_input.submit(chat_with_groq, inputs=[chat_input, state], outputs=[chat_input, chatbot, state, saved_chats_container])
 
     def refresh_chat_list():
-        chats = list_saved_chats()
-        return [gr.Button(value=chat, scale=1) for chat in chats]
-
-    def load_chat_btn(filename):
-        return load_chat_file(filename)
+        files = list_saved_chats()
+        buttons = []
+        for file in files:
+            btn = gr.Button(value=file[:-5], scale=1)
+            btn.click(fn=load_chat_file, inputs=[gr.State(file)], outputs=[chatbot, state])
+            buttons.append(btn)
+        return buttons
 
     def start_stop_recording(is_recording):
         if not is_recording:
-            return "Recording...", True, gr.update(visible=True)
+            return "Recording...", True
         else:
-            return "", False, gr.update(visible=False)
+            return "", False
 
-    def handle_audio(audio):
-        return transcribe_audio(audio), gr.update(visible=False)
+    def handle_audio_transcription(audio):
+        return transcribe_audio(audio)
 
     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(start_stop_recording, inputs=recording, outputs=[chat_input, recording])
 
-    demo.load(refresh_chat_list, outputs=[saved_chats_list])
+    demo.load(refresh_chat_list, outputs=[saved_chats_container])
 
-demo.launch()
+demo.launch()