Update app.py

app.py

@@ -946,26 +946,46 @@ atexit.register(cleanup_session)
 
 if __name__ == "__main__":
     import sys
-    
+    import threading
+    import time
+
     # Set logging level based on environment
     if "--debug" in sys.argv:
         logging.getLogger().setLevel(logging.DEBUG)
         logger.info("🐛 Debug logging enabled")
-    
+
+    # --- Thread-safe function to run the API server ---
+    def run_api():
+        print("🚀 Starting optimized API server on http://0.0.0.0:8000")
+        run_api_server(host="0.0.0.0", port=8000)
+
+    # --- Main function to start Gradio ---
+    def run_gradio():
+        print("🚀 Starting optimized Gradio interface...")
+        # The demo.launch() will block the main thread
+        run_gradio_interface()
+
+    # --- Logic to run API, Gradio, or both ---
     if len(sys.argv) > 1 and sys.argv[1] == "api":
-        # Run only API server
-        print("🚀 Starting optimized API server...")
-        run_api_server()
-    elif len(sys.argv) > 1 and sys.argv[1] == "both":
-        # Run both API and Gradio in separate threads
+        # Run only the API server
+        run_api()
+
+    elif len(sys.argv) > 1 and sys.argv[1] == "gradio":
+        # Run only the Gradio interface
+        run_gradio()
+
+    else:
+        # Default behavior: Run both API and Gradio
         print("🚀 Starting both API server and Gradio interface...")
-        api_thread = threading.Thread(target=run_api_server, daemon=True)
+
+        # Run the API server in a daemon thread
+        # A daemon thread will exit when the main program exits.
+        api_thread = threading.Thread(target=run_api, daemon=True)
         api_thread.start()
+
+        # Give the API server a moment to initialize
         print("⏳ Waiting 2 seconds for API server to start...")
         time.sleep(2)
-        run_gradio_interface()
-    else:
-        # Default: Run only Gradio
-        print("🚀 Starting optimized Gradio interface...")
-        run_gradio_interface()
-        run_api_server()
+
+        # Run the Gradio interface in the main thread
+        run_gradio()