Upload web-ui.py

web-ui.py

@@ -174,23 +174,17 @@ with gr.Blocks() as demo:
         outputs=[output_gallery, output_text, display_seed],
     )
 	
-def start_gradio():
-    demo.launch(share=args.share, inbrowser=True)
-
 def start_ngrok():
-    # Set the correct port here. By default, Gradio uses 7860, but it might be different in your setup
-    gradio_port = 7860
+    time.sleep(10)  # Delay for 10 seconds to ensure Gradio starts first
     ngrok.set_auth_token(args.ngrok_token)
-    public_url = ngrok.connect(gradio_port)
-    print(f"Public URL: {public_url}")
-	
-if __name__ == "__main__":
-    # Launch the Gradio app in a separate thread
-    threading.Thread(target=start_gradio).start()
+    public_url = ngrok.connect(port=7860)  # Adjust to your Gradio app's port
+    print(f"ngrok tunnel started at {public_url}")
 
-    # Give Gradio a few seconds to start up
-    time.sleep(15)  # Adjust the sleep time if necessary
-
-    # Start ngrok
+if __name__ == "__main__":
     if args.ngrok_token:
-        start_ngrok()	
+        # Start ngrok in a daemon thread with a delay
+        ngrok_thread = threading.Thread(target=start_ngrok, daemon=True)
+        ngrok_thread.start()
+
+    # Launch the Gradio app
+    demo.launch(share=args.share, inbrowser=True)