Adjust Gradio launch for Spaces

app.py

@@ -13,7 +13,9 @@ FEATURE_NAMES = [
 
 CACHE_ROOT = Path(__file__).with_name(".cache")
 CACHE_ROOT.mkdir(exist_ok=True)
-os.environ.setdefault("MPLCONFIGDIR", str(CACHE_ROOT / "matplotlib"))
+matplotlib_cache = CACHE_ROOT / "matplotlib"
+matplotlib_cache.mkdir(parents=True, exist_ok=True)
+os.environ.setdefault("MPLCONFIGDIR", str(matplotlib_cache))
 os.environ.setdefault("XDG_CACHE_HOME", str(CACHE_ROOT))
 
 import gradio as gr
@@ -274,23 +276,33 @@ with gr.Blocks(title="UNLV Study Location Predictor") as demo:
 
 
 if __name__ == "__main__":
-    default_port = int(os.environ.get("GRADIO_SERVER_PORT", "7860"))
+    queued_app = demo.queue()
+    is_hf_space = bool(os.environ.get("SPACE_ID"))
+    default_port = int(
+        os.environ.get(
+            "PORT",
+            os.environ.get("GRADIO_SERVER_PORT", "7860"),
+        )
+    )
+    launch_kwargs = {
+        "server_name": os.environ.get("GRADIO_SERVER_NAME", "0.0.0.0"),
+        "server_port": default_port,
+        "show_error": True,
+        "share": os.environ.get("GRADIO_SHARE", "").lower() == "true",
+    }
+    if not is_hf_space:
+        launch_kwargs["prevent_thread_lock"] = True
+
     try:
-        demo.queue().launch(
-            server_name=os.environ.get("GRADIO_SERVER_NAME", "0.0.0.0"),
-            server_port=default_port,
-            prevent_thread_lock=True,
-            show_error=True,
-            share=False,
+        queued_app.launch(**launch_kwargs)
+    except OSError:
+        fallback_port = int(
+            os.environ.get(
+                "GRADIO_FALLBACK_PORT",
+                str(default_port + 1111),
+            )
         )
-    except OSError as err:
-        fallback_port = int(os.environ.get("GRADIO_FALLBACK_PORT", default_port + 1111))
-        if fallback_port == default_port:
+        if fallback_port == launch_kwargs["server_port"]:
             raise
-        demo.queue().launch(
-            server_name=os.environ.get("GRADIO_SERVER_NAME", "0.0.0.0"),
-            server_port=fallback_port,
-            prevent_thread_lock=True,
-            show_error=True,
-            share=False,
-        )
+        launch_kwargs["server_port"] = fallback_port
+        queued_app.launch(**launch_kwargs)