Upload server.py with huggingface_hub

server.py

@@ -51,7 +51,8 @@ def generate():
         from diffusers.utils import export_to_video
 
         data = request.get_json(force=True)
-        prompt = data.get("inputs", "")
+        # Support both HF format (inputs) and backend format (prompt)
+        prompt = data.get("inputs", "") or data.get("prompt", "")
         params = data.get("parameters", {})
 
         num_frames = int(params.get("num_frames", 25))
@@ -94,12 +95,21 @@ def generate():
             "format": "mp4",
             "frames": len(frames),
             "duration": duration,
+            "success": True,
+            "video_base64": video_b64,
+            "duration_seconds": duration,
+            "model": "wan2.2-ti2v-5b",
         })
     except Exception as e:
         tb = traceback.format_exc()
-        print(f"[video] ERROR: {e}", flush=True)
-        print(f"[video] TRACEBACK:\n{tb}", flush=True)
-        return jsonify({"error": str(e), "traceback": tb}), 500
+        print(f"[video] ERROR: {e}\n{tb}", flush=True)
+        return jsonify({"error": str(e), "traceback": tb, "success": False}), 500
+
+# Also add /api/generate-video route for backend compatibility
+@app.route("/api/generate-video", methods=["POST"])
+def api_generate_video():
+    """Handle the backend's /api/generate-video format by forwarding to generate()"""
+    return generate()
 
 if __name__ == "__main__":
     load_model()