Update app.py

app.py

@@ -1,40 +1,48 @@
 # app.py
-import gradio as gr
-import subprocess
-import shlex
 import os
+import gradio as gr
+from huggingface_hub import InferenceClient
+
+# ── CONFIG ─────────────────────────────────────────────────────────────────────
+
+# (1) Your HF token must be set in Secrets as HF_HUB_TOKEN
+HF_TOKEN = os.getenv("HF_HUB_TOKEN", None)
+if not HF_TOKEN:
+    raise RuntimeError("Please set a HF_HUB_TOKEN in your Space secrets!")
+
+# (2) Model to call
+MODEL_ID = "damo-vilab/text-to-video-ms-1.7b"
+
+# Initialize the HF Inference client
+client = InferenceClient(token=HF_TOKEN)
+
+# ── GENERATION FUNCTION ────────────────────────────────────────────────────────
 
 def generate_video(prompt: str) -> str:
     """
-    Replace the inside of this function with your real video-gen call.
-    It must return a path to an .mp4 file.
+    Calls HF's text-to-video API, writes out an MP4, and returns its path.
     """
     out_path = "/tmp/out.mp4"
 
-    # === placeholder: ffmpeg text-on-black demo ===
-    # (so you can verify the UI/video pipeline is wired up)
-    # Draw your prompt centered in white on black for 3 seconds.
-    cmd = f"""
-        ffmpeg -y \
-          -f lavfi -i color=c=black:s=640x360:d=3 \
-          -vf "drawtext=fontfile=/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf:
-                text='{prompt}':
-                fontcolor=white:fontsize=48:
-                x=(w-text_w)/2:y=(h-text_h)/2" \
-          -codec:a copy {out_path}
-    """
-    # run it
-    subprocess.run(shlex.split(cmd), check=True)
+    # 1) Fire off the inference request
+    # The `text_to_video` method returns raw bytes for the first generated video.
+    result = client.text_to_video(model=MODEL_ID, inputs=prompt)
+
+    # 2) Save to disk
+    video_bytes = result["generated_video"]  # bytes
+    with open(out_path, "wb") as f:
+        f.write(video_bytes)
+
     return out_path
 
-# === Gradio UI ===
+# ── GRADIO UI ─────────────────────────────────────────────────────────────────
+
 with gr.Blocks() as demo:
-    gr.Markdown("# 🎬 Video Generator")
-    prompt = gr.Textbox(label="Enter your prompt here")
+    gr.Markdown("## 🎬 Text-to-Video Generator")
+    prompt = gr.Textbox(label="Enter your prompt here", placeholder="a chicken crosses the road")
     video_out = gr.Video(label="Generated Video")
     btn = gr.Button("Generate Video")
 
     btn.click(fn=generate_video, inputs=prompt, outputs=video_out)
 
-# Launch with a public link on Spaces
 demo.launch(share=True)