File size: 1,800 Bytes
0e1fc9f cc5532b 0e1fc9f cc5532b 0e1fc9f cc5532b 0e1fc9f cc5532b 0e1fc9f cc5532b 0e1fc9f cc5532b 0e1fc9f cc5532b 5673183 cc5532b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
import gradio as gr
import tempfile, os, subprocess, json
BLENDER = "blender" # symlinked in Dockerfile
def build_prop(spec_file):
# Load and sanity-check JSON
try:
spec_path = spec_file.name
with open(spec_path, "r") as f:
spec = json.load(f)
except Exception as e:
return None, f"Invalid JSON: {e}"
# Minimal schema checks
ptype = spec.get("type")
if not ptype:
return None, "Spec must include a 'type' field (e.g., 'book')."
if "name" not in spec:
spec["name"] = f"{ptype}_auto"
# Route to generator (extend here with more types later)
generator = "generate_book.py" if ptype == "book" else "generate_book.py"
out_dir = tempfile.mkdtemp()
cmd = [BLENDER, "-b", "-P", generator, "--", "--spec", spec_path, "--out", out_dir]
try:
run = subprocess.run(cmd, check=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True)
logs = run.stdout
except subprocess.CalledProcessError as e:
return None, f"Blender failed:\n{e.stdout}"
# Return first .glb produced
glbs = [os.path.join(out_dir, f) for f in os.listdir(out_dir) if f.lower().endswith(".glb")]
if not glbs:
return None, f"No .glb exported.\nLogs:\n{logs}"
return glbs[0], "Success."
demo = gr.Interface(
fn=build_prop,
inputs=gr.File(label="Upload PropSpec JSON"),
outputs=[gr.File(label="Generated .glb"), gr.Textbox(label="Log")],
title="Code‑Generated Props (Blender Headless)",
description="Upload a JSON PropSpec (supports 'book' for now) and download a .glb.",
allow_flagging="never" # prevent write-permission errors on HF Spaces
)
if __name__ == "__main__":
demo.launch(server_name="0.0.0.0", server_port=int(os.getenv("PORT", "7860"))) |