Create app.py

app.py

@@ -0,0 +1,113 @@
+# app.py
+import os
+import tempfile
+import gradio as gr
+from pathlib import Path
+import shutil
+import subprocess
+import json
+
+HF_TOKEN = os.environ.get("HF_TOKEN") or os.environ.get("HUGGINGFACE_TOKEN")
+
+# Utility: save uploaded file and return path
+def save_temp_file(uploaded):
+    tmp = tempfile.mkdtemp()
+    if isinstance(uploaded, tuple):
+        # gradio file gives (name, fileobj)
+        filename = uploaded[0]
+        fileobj = uploaded[1]
+        path = Path(tmp) / filename
+        with open(path, "wb") as f:
+            f.write(fileobj.read())
+        return str(path)
+    else:
+        # path-like (when running locally in dev)
+        return uploaded
+
+
+def try_import_sp4d():
+    """Try importing a local SP4D pipeline from Stability-AI repo.
+    If successful, return pipeline runner function. Otherwise return None.
+    """
+    try:
+        # Users will typically install from the generative-models repo
+        # Example: pip install git+https://github.com/Stability-AI/generative-models.git@sp4d
+        from sp4d import SP4DPipeline  # type: ignore
+
+        def run_sp4d_local(input_path, guidance_scale=7.5, seed=None):
+            # Example stub. Adapt to actual pipeline signature from stabilityai repo.
+            pipe = SP4DPipeline.from_pretrained("stabilityai/sp4d")
+            # The real pipeline requires video frames tensor input; implement conversions here.
+            results = pipe(input_path, guidance_scale=guidance_scale, generator_seed=seed)
+            # save outputs to a folder and return path
+            out_dir = tempfile.mkdtemp()
+            # `results` handling depends on the real pipeline's return format
+            # Placeholder: write metadata json
+            with open(Path(out_dir)/"metadata.json", "w") as f:
+                json.dump({"note": "Replace this stub with actual results handling."}, f)
+            return out_dir
+
+        return run_sp4d_local
+    except Exception as e:
+        print("SP4D local import failed:", e)
+        return None
+
+
+# If local import not available, create fallback runner that instructs the user
+LOCAL_RUNNER = try_import_sp4d()
+
+
+def run_inference(file, guidance, seed):
+    path = save_temp_file(file)
+
+    if LOCAL_RUNNER is not None:
+        # Run local SP4D pipeline
+        try:
+            out_dir = LOCAL_RUNNER(path, guidance_scale=guidance, seed=seed)
+            # Zip outputs for download
+            zip_path = shutil.make_archive(out_dir, "zip", out_dir)
+            return (f"Ran local SP4D and produced outputs in {out_dir}", zip_path)
+        except Exception as e:
+            return (f"Local SP4D run failed: {e}", None)
+    else:
+        # Provide helpful message + instructions
+        instructions = (
+            "SP4D is not installed in this Space environment.\n\n"
+            "You can either: \n"
+            "1) Deploy SP4D by installing the Stability-AI generative-models repo in this Space:\n"
+            "   pip install git+https://github.com/Stability-AI/generative-models.git@sp4d\n"
+            "   Then ensure the Space has a GPU (Gradio 'hardware accelerator' = GPU) and restart.\n\n"
+            "2) Or run inference via a custom backend (if/when Stability deploys an inference endpoint).\n\n"
+            "See README in this Space for full setup steps."
+        )
+        return (instructions, None)
+
+
+with gr.Blocks(title="SP4D — Stable Part Diffusion 4D") as demo:
+    gr.Markdown("# SP4D — Stable Part Diffusion 4D (UI template)")
+    with gr.Row():
+        with gr.Column(scale=2):
+            inp_file = gr.File(label="Upload 4-frame input (video file or zip of 4 frames)")
+            guidance = gr.Slider(0.0, 20.0, value=7.5, label="Guidance scale")
+            seed = gr.Number(value=0, label="Seed (0 = random)")
+            run_btn = gr.Button("Run SP4D")
+            output_text = gr.Textbox(label="Status / Instructions", interactive=False)
+            download_output = gr.File(label="Outputs (zip)")
+
+        with gr.Column(scale=1):
+            gr.Markdown(
+                """
+                ## Notes
+                - This Space is a **template** UI. SP4D is large and may require custom installation.
+                - See README for installation instructions and licensing notes.
+                """
+            )
+
+    def on_run(file, guidance, seed):
+        status, out_zip = run_inference(file, guidance, seed if seed != 0 else None)
+        return status, out_zip
+
+    run_btn.click(on_run, inputs=[inp_file, guidance, seed], outputs=[output_text, download_output])
+
+if __name__ == "__main__":
+    demo.launch()