Only run web app on localhost in dev mode

Author: sd109

Tiltfile

@@ -53,6 +53,6 @@ if run_ui_locally:
         serve_cmd=" && ".join([
             "source {}/bin/activate".format(venv_name),
             "cd chart/web-app",
-            "python3 app.py {}".format(hf_model),
+            "python3 app.py {} localhost".format(hf_model),
         ])
     )

chart/web-app/app.py

@@ -174,5 +174,9 @@ def inference_wrapper(*args):
     css=css_overrides,
 ) as app:
     logger.debug("Gradio chat interface config: %s", app.config)
-    # app.launch(server_name="0.0.0.0") # Do we need this for k8s service?
-    app.launch()
+    # For running locally in tilt dev setup
+    if len(sys.argv) > 2 and sys.argv[2] == "localhost":
+        app.launch()
+    # For running on cluster
+    else:
+        app.launch(server_name="0.0.0.0")