Create app.py

app.py

@@ -0,0 +1,69 @@
+import spaces
+import gradio as gr
+import subprocess, os, sys, time
+import urllib.request, urllib.error
+from fastapi import Request
+from fastapi.responses import Response
+
+SECOND_PORT = 9001
+FLASK_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "flask-app")
+
+# --- Launch YOUR flask-app as a background process on its own port ---
+# cwd=FLASK_DIR so its templates/relative paths resolve correctly.
+subprocess.Popen(
+    [sys.executable, "app.py"],
+    cwd=FLASK_DIR,
+    stdout=sys.stdout,
+    stderr=sys.stderr,
+)
+
+# --- Required ZeroGPU function (top level) ---
+@spaces.GPU
+def gpu_function():
+    import torch
+    return "GPU OK" if torch.cuda.is_available() else "GPU function executed"
+
+# --- Gradio owns the ONLY public port 7860 ---
+with gr.Blocks() as demo:
+    gr.Markdown("Host running. Your Flask app is on 9001, bridged at /app")
+    gr.Button("GPU check").click(gpu_function, outputs=gr.Textbox())
+
+demo.queue()
+demo.launch(
+    server_name="0.0.0.0",
+    server_port=7860,
+    ssr_mode=False,            # prevents the Node SSR shutdown
+    prevent_thread_lock=True,  # returns control so we can mount the proxy after launch
+)
+
+app = demo.app  # Gradio's FastAPI app on 7860
+
+# --- A session-aware, redirect-aware proxy: /app/... -> 127.0.0.1:9001/app/... ---
+class NoRedirect(urllib.request.HTTPRedirectHandler):
+    def redirect_request(self, *args, **kwargs):
+        return None  # pass 3xx straight to the browser so cookies stick
+
+@app.api_route("/app", methods=["GET", "POST"])
+@app.api_route("/app/{path:path}", methods=["GET", "POST"])
+async def proxy_to_flask(request: Request, path: str = ""):
+    url = f"http://127.0.0.1:{SECOND_PORT}/app/{path}"
+    if request.url.query:
+        url += "?" + request.url.query
+    body = await request.body()
+    fwd_headers = {k: v for k, v in request.headers.items() if k.lower() != "host"}
+    req = urllib.request.Request(url, data=body or None, method=request.method, headers=fwd_headers)
+    try:
+        opener = urllib.request.build_opener(NoRedirect)
+        with opener.open(req) as r:
+            resp_body, status, headers = r.read(), r.status, r.headers
+    except urllib.error.HTTPError as e:
+        resp_body, status, headers = e.read(), e.code, e.headers
+    except urllib.error.URLError:
+        return Response(content=b"flask-app not up yet on 9001", status_code=502)
+    out_headers = {k: v for k, v in headers.items()
+                   if k.lower() in ("set-cookie", "location", "content-type")}
+    return Response(content=resp_body, status_code=status, headers=out_headers)
+
+# --- Keep the process alive (launch returned control) ---
+while True:
+    time.sleep(3600)