Upload web-ui.py

web-ui.py

@@ -11,6 +11,7 @@ import argparse
 import random
 from insightface.utils import face_align
 from pyngrok import ngrok
+import threading
 
 # Argument parser for command line options
 parser = argparse.ArgumentParser()
@@ -175,12 +176,19 @@ with gr.Blocks() as demo:
 def start_ngrok():
     if args.ngrok_token:
         ngrok.set_auth_token(args.ngrok_token)
-        url = ngrok.connect(port=7860)  # You can choose your desired port
-        print(f"ngrok tunnel started at {url}")	
+        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()
 
-if __name__ == "__main__":
-    demo.launch(share=args.share, inbrowser=True)
 
-    # Start ngrok after launching Gradio
+if __name__ == "__main__":
     if args.ngrok_token:
-        start_ngrok()
+        # Start ngrok in a separate thread
+        threading.Thread(target=delayed_ngrok_start).start()
+
+    # Launch Gradio app
+    demo.launch(share=args.share, inbrowser=True)