Upload web-ui.py

web-ui.py

@@ -12,6 +12,7 @@ import random
 from insightface.utils import face_align
 from pyngrok import ngrok
 import threading
+import time
 
 # Argument parser for command line options
 parser = argparse.ArgumentParser()
@@ -174,21 +175,20 @@ with gr.Blocks() as demo:
     )
 	
 def start_ngrok():
-    if args.ngrok_token:
-        ngrok.set_auth_token(args.ngrok_token)
-        url = ngrok.connect(port=7860)  # Adjust the port if necessary
-        print(f"ngrok tunnel started at {url}")
-
-def delayed_ngrok_start():
-    import time
-    time.sleep(5)  # Wait for 5 seconds before starting ngrok
-    start_ngrok()
+    # Set the correct port here. By default, Gradio uses 7860, but it might be different in your setup
+    gradio_port = 7860
+    ngrok.set_auth_token(args.ngrok_token)
+    public_url = ngrok.connect(gradio_port)
+    print(f"Public URL: {public_url}")
 
 
 if __name__ == "__main__":
-    if args.ngrok_token:
-        # Start ngrok in a separate thread
-        threading.Thread(target=delayed_ngrok_start).start()
+    # Launch the Gradio app in a separate thread
+    threading.Thread(target=lambda: demo.launch(share=args.share, inbrowser=True)).start()
 
-    # Launch Gradio app
-    demo.launch(share=args.share, inbrowser=True)
+    # Give Gradio a few seconds to start up
+    time.sleep(5)
+
+    # Start ngrok
+    if args.ngrok_token:
+        start_ngrok()