Update app.py

app.py

@@ -226,8 +226,12 @@ async def async_edge_tts_generate(text, voice, username, rate=0, pitch=0, file_f
     try:
         communicate = edge_tts.Communicate(text, voice, rate=f"{rate:+d}%", pitch=f"{pitch:+d}Hz")
         await communicate.save(filename)
-        st.session_state['audio_cache'][cache_key] = filename
-        return filename, time.time() - start_time
+        if os.path.exists(filename) and os.path.getsize(filename) > 0:
+            st.session_state['audio_cache'][cache_key] = filename
+            return filename, time.time() - start_time
+        else:
+            print(f"Audio file {filename} was not created or is empty.")
+            return None, 0
     except edge_tts.exceptions.NoAudioReceived as e:
         print(f"No audio received for text: '{text}' with voice: {voice}. Error: {e}")
         return None, 0
@@ -239,6 +243,8 @@ def play_and_download_audio(file_path):
     if file_path and os.path.exists(file_path):
         st.audio(file_path)
         st.markdown(get_download_link(file_path), unsafe_allow_html=True)
+    else:
+        st.warning(f"Audio file not found: {file_path}")
 
 def load_mp3_viewer():
     mp3_files = sorted(glob.glob("*.mp3"), key=os.path.getmtime)
@@ -264,6 +270,8 @@ async def save_chat_entry(username, message, voice, is_markdown=False):
         if st.session_state.get('speech_processed', False) and st.session_state.get('message_input', '') == message:
             st.session_state['message_input'] = ""
             st.session_state['speech_processed'] = False
+    else:
+        st.warning(f"Failed to generate audio for: {message}")
     await broadcast_message(f"{username}|{message}", "chat")
     st.session_state.last_chat_update = time.time()
     st.session_state.chat_history.append(entry)
@@ -412,13 +420,15 @@ async def broadcast_message(message, room_id):
                 del st.session_state.active_connections[room_id][client_id]
 
 async def run_websocket_server():
-    if not st.session_state.server_running:
+    if not st.session_state.get('server_running', False):  # Safe access with default
         server = await websockets.serve(websocket_handler, '0.0.0.0', 8765)
-        st.session_state.server_running = True
+        st.session_state['server_running'] = True
         await server.wait_closed()
 
 def start_websocket_server():
-    asyncio.run(run_websocket_server())
+    loop = asyncio.new_event_loop()
+    asyncio.set_event_loop(loop)
+    loop.run_until_complete(run_websocket_server())
 
 # 📚 PDF to Audio
 class AudioProcessor:
@@ -605,7 +615,7 @@ def paste_image_component():
                     base64_str = paste_input.split(',')[1]
                     img_bytes = base64.b64decode(base64_str)
                     img = Image.open(io.BytesIO(img_bytes))
-                    st.image(img, caption="Image Pasted Successfully", use_column_width=True)  # Immediate feedback
+                    st.image(img, caption="Image Pasted Successfully", use_column_width=True)
                     return img
                 except Exception as e:
                     st.error(f"Error decoding pasted image: {e}")
@@ -617,7 +627,7 @@ def paste_image_component():
 
 # 🎮 Main Interface
 def main():
-    init_session_state()
+    init_session_state()  # Ensure session state is initialized before any threads
     load_mp3_viewer()
     saved_username = load_username()
     if saved_username and saved_username in FUN_USERNAMES:
@@ -670,7 +680,6 @@ def main():
                 with st.spinner("Saving image..."):
                     filename = asyncio.run(save_pasted_image(pasted_image, st.session_state.username, image_prompt))
                 if filename:
-                    st.session_state.pasted_image_data = filename
                     st.success(f"Image saved as: {filename}")
                     if image_prompt:
                         with st.spinner("Processing with Claude..."):
@@ -781,7 +790,7 @@ def main():
 
     st.subheader("🎵 Audio (MP3)")
     for filename, (num, mp3) in sorted(st.session_state['mp3_files'].items(), key=lambda x: x[1][0]):
-        with st.expander(f"{num}. {os.path.basename(mp3)}"):
+        with st.expander(f"{num}. ${filename}"):
             st.audio(mp3)
             st.markdown(get_download_link(mp3, "mp3"), unsafe_allow_html=True)
 
@@ -902,7 +911,8 @@ def main():
     else:
         timer_placeholder.markdown(f"<p class='timer'>⏳ Next refresh in: {remaining_time} seconds</p>", unsafe_allow_html=True)
 
-    if not st.session_state.server_running and not st.session_state.server_task:
+    # Start WebSocket server only after session state is initialized
+    if not st.session_state.get('server_running', False) and not st.session_state.get('server_task', None):
         st.session_state.server_task = threading.Thread(target=start_websocket_server, daemon=True)
         st.session_state.server_task.start()