Update app.py

app.py

@@ -1,45 +1,88 @@
 import os
 import subprocess
 import threading
+import requests
 import gradio as gr
+from fastapi import FastAPI, Request
+from fastapi.responses import Response
+import uvicorn
 
-def install_pufferpanel():
-    if not os.path.exists("/usr/sbin/pufferpanel"):
-        # Download amd64 deb package
+# ==========================
+# CONFIG
+# ==========================
+PUFFER_URL = "http://127.0.0.1:8080"  # Internal PufferPanel address
+PUFFER_USER = "admin"
+PUFFER_PASS = "admin123"
+
+
+# ==========================
+# Start PufferPanel
+# ==========================
+def start_pufferpanel():
+    # Make sure pufferpanel binary exists
+    if not os.path.exists("pufferpanel"):
+        print("Downloading and building PufferPanel...")
         subprocess.run(
-            "wget https://github.com/pufferpanel/pufferpanel/releases/download/v3.0.0-rc.14/pufferpanel_3.0.0-rc.14_amd64.deb -O pufferpanel.deb",
-            shell=True, check=True
+            [
+                "wget",
+                "https://github.com/pufferpanel/pufferpanel/releases/download/v3.0.0-rc.14/pufferpanel_3.0.0-rc.14_linux_amd64.tar.gz",
+                "-O",
+                "pufferpanel.tar.gz",
+            ],
+            check=True,
         )
-        # Install dependencies and package
-        subprocess.run("apt-get update && apt-get install -y ./pufferpanel.deb", shell=True, check=True)
-        # Enable PufferPanel
-        subprocess.run("pufferpanel user add admin --password admin123", shell=True, check=True)
+        subprocess.run(["tar", "xvzf", "pufferpanel.tar.gz"], check=True)
+
+    # Run pufferpanel
+    subprocess.Popen(["./pufferpanel"], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+
+
+threading.Thread(target=start_pufferpanel, daemon=True).start()
 
-def run_pufferpanel():
-    subprocess.run("pufferpanel run", shell=True)
 
-install_pufferpanel()
-threading.Thread(target=run_pufferpanel, daemon=True).start()
+# ==========================
+# FastAPI Reverse Proxy
+# ==========================
+app = FastAPI()
 
-def launch_info():
-    return """
-## 🚀 PufferPanel v3.0.0-rc.14
 
-Running inside Hugging Face Space.
+@app.api_route("/proxy/{path:path}", methods=["GET", "POST", "PUT", "DELETE", "PATCH"])
+async def proxy(request: Request, path: str):
+    """Reverse proxy to PufferPanel backend"""
+    url = f"{PUFFER_URL}/{path}"
+    headers = dict(request.headers)
+    body = await request.body()
+
+    try:
+        resp = requests.request(
+            method=request.method,
+            url=url,
+            headers=headers,
+            data=body,
+            cookies=request.cookies,
+            allow_redirects=False,
+        )
+        return Response(
+            content=resp.content,
+            status_code=resp.status_code,
+            headers=dict(resp.headers),
+        )
+    except Exception as e:
+        return Response(content=f"Proxy error: {e}", status_code=500)
+
 
-**Login Credentials:**  
-- **User:** `admin`  
-- **Pass:** `admin123`
+# ==========================
+# Gradio UI
+# ==========================
+def launch_panel():
+    return f"""
+    <iframe src="/proxy/" style="width: 100%; height: 700px; border: none;"></iframe>
+    """
 
-"""
 
-with gr.Blocks(title="PufferPanel on Hugging Face") as demo:
-    gr.Markdown(launch_info)
-    gr.HTML("""
-        <iframe src="http://localhost:8080" 
-                style="width:100%; height:800px; border:none;">
-        </iframe>
-    """)
+with gr.Blocks() as demo:
+    gr.Markdown("# 🚀 PufferPanel on Hugging Face\nLogin with your credentials below.")
+    gr.HTML(value=launch_panel())
 
-if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+# Mount FastAPI inside Gradio
+demo.launch(server_name="0.0.0.0", server_port=7860, root_path="/")