Update app.py

app.py

@@ -2,38 +2,50 @@ import http.server
 import socketserver
 import os
 import subprocess
-import threading
 import time
+import threading
 
-# Define the port for the public-facing HTTP server
+# Define the public-facing port for the HTTP server
 PORT = 7860
 
-# The handler for the web server
+# Command to run EchoAudio.py
+ECHO_AUDIO_CMD = ["python", "EchoAudio.py"]
+
+# Function to run the EchoAudio.py script
+def run_echo_audio():
+    print("Starting EchoAudio.py...")
+    # This runs the script in the background and waits for it to complete.
+    # It's a blocking call, but since it's in a separate thread, it's fine.
+    subprocess.run(ECHO_AUDIO_CMD)
+    print("EchoAudio.py process has finished.")
+
+# Custom handler to serve client.html
 class MyHandler(http.server.SimpleHTTPRequestHandler):
     def do_GET(self):
-        # Serve client.html when the root URL is requested
         if self.path == '/':
             self.path = '/client.html'
         return http.server.SimpleHTTPRequestHandler.do_GET(self)
 
-# Function to run EchoAudio.py in a separate thread
-def run_echo_audio():
-    print("Starting EchoAudio.py...")
-    # This runs the script as a subprocess
-    subprocess.run(["python", "EchoAudio.py"])
-
 if __name__ == "__main__":
-    # Start the EchoAudio.py script in a new thread
-    thread = threading.Thread(target=run_echo_audio)
-    thread.start()
+    # Start the EchoAudio.py script in a separate thread
+    echo_audio_thread = threading.Thread(target=run_echo_audio)
+    echo_audio_thread.start()
     
-    # Wait a moment for EchoAudio.py to start up
-    time.sleep(5)
+    # Give the WebSocket servers a moment to initialize
+    print("Waiting for WebSocket servers to start...")
+    time.sleep(10) # ⏳ You might need to adjust this value
+    print("Continuing with HTTP server startup.")
     
-    # Set the directory to serve files from (e.g., where client.html is)
+    # Change to the directory where your files are located
     os.chdir(os.path.dirname(os.path.abspath(__file__)))
 
-    # Start the web server on the public port
+    # Start the HTTP server on the public port
     with socketserver.TCPServer(("", PORT), MyHandler) as httpd:
-        print("Serving client.html on port", PORT)
-        httpd.serve_forever()
+        print(f"Serving client.html on port {PORT}")
+        try:
+            httpd.serve_forever()
+        except KeyboardInterrupt:
+            pass
+    
+    # Wait for the background thread to finish
+    echo_audio_thread.join()