Update app.py

app.py

@@ -1,28 +1,42 @@
 import gradio as gr
 import os
 import json
+from fastapi import Request
 
 PROJECTS_DIR = "projects"
 os.makedirs(PROJECTS_DIR, exist_ok=True)
 
-def load_project(project):
+def load_project(project: str):
+    """
+    GET /load?project=FILENAME
+    """
     filename = os.path.join(PROJECTS_DIR, os.path.basename(project))
-    if not os.path.exists(filename):
+    if not os.path.isfile(filename):
         return gr.Response(status_code=404, body="Project not found.")
     with open(filename, "r") as f:
         return json.load(f)
 
-def save_project(project, request: gr.Request):
+async def save_project(request: Request):
+    """
+    POST /save?project=FILENAME
+    Body: JSON project data
+    """
+    project = request.query_params.get("project")
+    if not project:
+        return gr.Response(status_code=400, body="Missing project parameter.")
     filename = os.path.join(PROJECTS_DIR, os.path.basename(project))
-    data = request.json
+    data = await request.json()
     with open(filename, "w") as f:
-        json.dump(data, f)
-    return { "success": True }
+        json.dump(data, f, indent=2)
+    return {"success": True}
 
 with gr.Blocks() as demo:
-    gr.Markdown("### GrapesJS Storage Backend")
+    gr.Markdown("## GrapesJS Storage Backend 👷‍♂️")
 
-demo.api("/load", load_project, methods=["GET"])
-demo.api("/save", save_project, methods=["POST"])
+# Register API routes on the underlying FastAPI app:
+# demo.app is the FastAPI instance
+demo.app.add_api_route("/load", load_project, methods=["GET"])
+demo.app.add_api_route("/save", save_project, methods=["POST"])
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()