Update localGPTUI/localGPTUI.py

localGPTUI/localGPTUI.py

@@ -56,6 +56,20 @@ def home_page():
         response_dict={"Prompt": "None", "Answer": "None", "Sources": [("ewf", "wef")]},
     )
 
+def run():
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--port", type=int, default=5111, help="Port to run the UI on. Defaults to 5111.")
+    parser.add_argument(
+        "--host",
+        type=str,
+        default="0.0.0.0",
+        help="Host to run the UI on. Defaults to 127.0.0.1. "
+        "Set to 0.0.0.0 to make the UI externally "
+        "accessible from other devices.",
+    )
+    args = parser.parse_args()
+    app.run(debug=False, host=args.host, port=args.port)
+
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()