Update app.py

app.py

@@ -1,4 +1,6 @@
 import gradio as gr
+import shutil
+from pathlib import Path
 from huggingface_hub import list_models
 
 # list_spaces is not guaranteed to exist in all HF runtimes
@@ -8,6 +10,50 @@ except ImportError:
     list_spaces = None
 
 
+# ===============================
+# Persistent storage
+# ===============================
+WORKFLOW_DIR = Path("/data/workflows")
+WORKFLOW_DIR.mkdir(parents=True, exist_ok=True)
+
+ALLOWED_EXTS = {".json", ".yaml", ".yml", ".png"}
+
+
+# ===============================
+# Workflow upload handlers
+# ===============================
+def upload_workflow(file):
+    if file is None:
+        return "No file uploaded."
+
+    src = Path(file.name)
+    ext = src.suffix.lower()
+
+    if ext not in ALLOWED_EXTS:
+        return f"❌ Unsupported file type: `{ext}`"
+
+    dest = WORKFLOW_DIR / src.name
+    shutil.copy(src, dest)
+
+    return f"✅ Saved workflow: `{dest.name}`"
+
+
+def list_uploaded_workflows():
+    files = sorted(WORKFLOW_DIR.glob("*"))
+
+    if not files:
+        return "No uploaded workflows yet."
+
+    return "\n---\n".join(
+        f"### {f.name}\n"
+        f"`{f.suffix}` | {f.stat().st_size // 1024} KB"
+        for f in files
+    )
+
+
+# ===============================
+# Hub browsing functions
+# ===============================
 def browse_loras(limit=20):
     models = list(list_models(
         tags=["lora"],
@@ -28,34 +74,48 @@ def browse_loras(limit=20):
 
 
 def browse_workflows(limit=20):
-    if list_spaces is None:
-        return "Spaces API unavailable in this runtime."
+    uploaded_files = sorted(WORKFLOW_DIR.glob("*"))
 
-    spaces = list(list_spaces(
-        sort="likes",
-        direction=-1,
-        limit=limit,
-    ))
-
-    workflows = [
-        s for s in spaces
-        if any(
-            tag in (s.tags or [])
-            for tag in ["comfyui", "workflow", "pipeline", "diffusers"]
+    uploaded_md = ""
+    if uploaded_files:
+        uploaded_md = "## 🧑‍💻 Uploaded Workflows\n\n" + "\n---\n".join(
+            f"### {f.name}\nLocal upload"
+            for f in uploaded_files
         )
-    ]
 
-    if not workflows:
-        return "No workflows found."
+    if list_spaces is None:
+        remote_md = "Spaces API unavailable."
+    else:
+        spaces = list(list_spaces(
+            sort="likes",
+            direction=-1,
+            limit=limit,
+        ))
+
+        workflows = [
+            s for s in spaces
+            if any(
+                tag in (s.tags or [])
+                for tag in ["comfyui", "workflow", "pipeline", "diffusers"]
+            )
+        ]
+
+        if workflows:
+            remote_md = "## 🌍 Community Workflows\n\n" + "\n---\n".join(
+                f"### {s.id}\n"
+                f"⭐ {s.likes or 0}\n"
+                f"https://huggingface.co/spaces/{s.id}"
+                for s in workflows[:limit]
+            )
+        else:
+            remote_md = "No community workflows found."
 
-    return "\n---\n".join(
-        f"### {s.id}\n"
-        f"⭐ {s.likes or 0}\n"
-        f"https://huggingface.co/spaces/{s.id}"
-        for s in workflows[:limit]
-    )
+    return uploaded_md + "\n\n" + remote_md
 
 
+# ===============================
+# Gradio UI
+# ===============================
 with gr.Blocks() as demo:
     gr.Markdown("# 🤗 Workflow & LoRA Browser")
 
@@ -71,7 +131,3 @@ with gr.Blocks() as demo:
             lora_out = gr.Markdown()
             gr.Button("Browse LoRAs").click(
                 browse_loras,
-                outputs=lora_out,
-            )
-
-demo.launch()