Update app.py

app.py

@@ -350,6 +350,9 @@ async def run_websocket_server():
         st.session_state.server_running = True
         await server.wait_closed()
 
+def start_websocket_server():
+    asyncio.run(run_websocket_server())
+
 # 📚 PDF to Audio
 class AudioProcessor:
     def __init__(self):
@@ -552,7 +555,7 @@ def main():
     useArxivAudio = st.checkbox("ArXiv Audio", False, key="use_arxiv_audio")
     # Use st.checkbox without reassigning to st.session_state
     st.checkbox("Autosend Chat", value=True, key="autosend")
-    st.checkbox("Autosearch ArXiv", value=True, key="autosearch")  # Fixed line: no reassignment
+    st.checkbox("Autosearch ArXiv", value=True, key="autosearch")
 
     # 🎤 Chat & Voice
     if tab_main == "🎤 Chat & Voice":
@@ -726,8 +729,10 @@ def main():
         if zip_name:
             st.sidebar.markdown(get_download_link(zip_name, "zip"), unsafe_allow_html=True)
 
-    if not st.session_state.server_task:
-        st.session_state.server_task = asyncio.create_task(run_websocket_server())
+    # Start WebSocket server in a separate thread
+    if not st.session_state.server_running and not st.session_state.server_task:
+        st.session_state.server_task = threading.Thread(target=start_websocket_server, daemon=True)
+        st.session_state.server_task.start()
 
 if __name__ == "__main__":
     main()