Update app.py

app.py

@@ -5,15 +5,25 @@ Compatible with Gradio 6.0+
 """
 import os
 import gradio as gr
+import spaces
 from client import BackendClient
 
 # ── Environment Configurations ────────────────────────────────────────────
 BACKEND_URL = os.getenv("BACKEND_URL", "http://localhost:7860")
 HF_TOKEN = os.getenv("HF_TOKEN") or os.getenv("API_KEY", "")
-USER_KEY = os.getenv("USER_KEY", "samuel_super_secret_key_1234567890_key")
+# No hardcoded fallback — if USER_KEY isn't set as a Space secret, start blank.
+USER_KEY = os.getenv("USER_KEY", "")
 
 client = BackendClient(backend_url=BACKEND_URL, token=HF_TOKEN)
 
+
+@spaces.GPU
+def _warmup():
+    """Dummy function so HF's ZeroGPU startup check finds a @spaces.GPU function.
+    Not actually used for compute — all real work happens on the remote backend."""
+    return "ok"
+
+
 # ── Custom CSS for ChatGPT-style Layout ───────────────────────────────────
 custom_css = """
 .sidebar-container {
@@ -41,7 +51,11 @@ custom_css = """
 # ── Helper to Load & Parse Backend Schema ─────────────────────────────────
 def load_backend_modules(api_key: str):
     """Fetch live OpenAPI spec and parse available backend routes into modules."""
-    ok, schema = client.fetch_openapi_schema(user_api_key=api_key)
+    try:
+        ok, schema = client.fetch_openapi_schema(user_api_key=api_key)
+    except Exception as e:
+        ok, schema = False, {"error": str(e)}
+
     modules_map = {}
 
     if ok and isinstance(schema, dict) and "paths" in schema:
@@ -68,7 +82,7 @@ def load_backend_modules(api_key: str):
     return modules_map, module_names
 
 
-# Load initial schema at startup
+# Load initial schema at startup (safe even if backend is unreachable)
 INITIAL_MAP, INITIAL_CHOICES = load_backend_modules(USER_KEY)
 
 # ── Gradio Blocks Interface (Clean Gradio 6.0 Constructor) ────────────────
@@ -84,9 +98,9 @@ with gr.Blocks(title="SOG-X Modular Workspace") as demo:
             with gr.Accordion("⚙️ Backend & Credentials", open=False):
                 user_key_input = gr.Textbox(
                     label="Active API Key",
-                    value=USER_KEY,
+                    value="",  # never pre-fill a real secret into a client-visible field
                     type="password",
-                    placeholder="Enter your 32+ char key...",
+                    placeholder="Enter your API key...",
                     interactive=True,
                 )
                 sync_btn = gr.Button("🔄 Sync Backend Modules", size="sm")
@@ -170,31 +184,37 @@ with gr.Blocks(title="SOG-X Modular Workspace") as demo:
         params = {}
         payload = {}
         files = {}
-
-        # Parse uploaded files
-        if file_obj is not None:
-            file_path = file_obj.name if hasattr(file_obj, "name") else str(file_obj)
-            files["file"] = open(file_path, "rb")
-
-        # Parse text input
-        if text_val and text_val.strip():
-            cleaned_text = text_val.strip()
-            if method == "GET":
-                params["query"] = cleaned_text
-                params["url"] = cleaned_text
-            else:
-                payload["input"] = cleaned_text
-                payload["url"] = cleaned_text
-                payload["prompt"] = cleaned_text
-
-        success, response = client.execute_dynamic_request(
-            path=path,
-            method=method,
-            params=params,
-            payload=payload,
-            files=files,
-            user_api_key=active_key,
-        )
+        file_handle = None
+
+        try:
+            # Parse uploaded files
+            if file_obj is not None:
+                file_path = file_obj.name if hasattr(file_obj, "name") else str(file_obj)
+                file_handle = open(file_path, "rb")
+                files["file"] = file_handle
+
+            # Parse text input
+            if text_val and text_val.strip():
+                cleaned_text = text_val.strip()
+                if method == "GET":
+                    params["query"] = cleaned_text
+                    params["url"] = cleaned_text
+                else:
+                    payload["input"] = cleaned_text
+                    payload["url"] = cleaned_text
+                    payload["prompt"] = cleaned_text
+
+            success, response = client.execute_dynamic_request(
+                path=path,
+                method=method,
+                params=params,
+                payload=payload,
+                files=files,
+                user_api_key=active_key,
+            )
+        finally:
+            if file_handle:
+                file_handle.close()
 
         # Check for downloadable output path in JSON response
         if success and isinstance(response, dict) and "file_path" in response:
@@ -210,9 +230,11 @@ with gr.Blocks(title="SOG-X Modular Workspace") as demo:
 
 # ── Launch Application (Gradio 6.0 Compliant) ─────────────────────────────
 if __name__ == "__main__":
+    _warmup()  # satisfies HF ZeroGPU startup check; app itself does no local GPU work
+    port = int(os.getenv("PORT", 7860))
     demo.launch(
         server_name="0.0.0.0",
-        server_port=7860,
+        server_port=port,
         theme=gr.themes.Soft(),
         css=custom_css,
         share=False,