Merge pull request #1 from faisal-fida/copilot/fix-ef9c9fc8-3c2b-4ce1-a1c1-fa92b73cd4a1

Add --public argument to webui.py for creating public URLs

Author: faisal-fida

README.md

@@ -87,6 +87,16 @@ cp .env.example .env
     ```bash
     python webui.py --ip 127.0.0.1 --port 7788
     ```
+    
+    **Optional Arguments:**
+    - `--public`: Create a public URL that can be accessed from anywhere (useful for sharing or remote access)
+    - `--theme`: Choose a UI theme (Default, Soft, Monochrome, Glass, Origin, Citrus, Ocean, Base)
+    
+    **Example with public access:**
+    ```bash
+    python webui.py --ip 127.0.0.1 --port 7788 --public
+    ```
+    
 2. **Access the WebUI:** Open your web browser and navigate to `http://127.0.0.1:7788`.
 3. **Using Your Own Browser(Optional):**
     - Set `BROWSER_PATH` to the executable path of your browser and `BROWSER_USER_DATA` to the user data directory of your browser. Leave `BROWSER_USER_DATA` empty if you want to use local user data.

webui.py

@@ -9,10 +9,11 @@ def main():
     parser.add_argument("--ip", type=str, default="127.0.0.1", help="IP address to bind to")
     parser.add_argument("--port", type=int, default=7788, help="Port to listen on")
     parser.add_argument("--theme", type=str, default="Ocean", choices=theme_map.keys(), help="Theme to use for the UI")
+    parser.add_argument("--public", action="store_true", help="Create a public URL that can be accessed from anywhere")
     args = parser.parse_args()
 
     demo = create_ui(theme_name=args.theme)
-    demo.queue().launch(server_name=args.ip, server_port=args.port)
+    demo.queue().launch(server_name=args.ip, server_port=args.port, share=args.public)
 
 
 if __name__ == '__main__':