Update app.py

app.py

@@ -1,28 +1,31 @@
+import os
 import gradio as gr
 from script import generate_post_and_prompt, generate_video
 
 def run_pipeline(artist_name, image, num_frames):
-    # Save uploaded image temporarily
-    image_path = image.name
-    file_name = image.name.split("/")[-1]
+    # 1. Guard-rail: make sure user really uploaded something
+    if image is None:
+        return "❗Please upload a face image.", None, None
 
-    # Generate Rednote post and video prompt
-    post_text, prompt = generate_post_and_prompt(artist_name)
+    # 2. `image` is a filepath string → get its name
+    file_path  = image          # e.g. "/tmp/gradio/23abc/image.png"
+    file_name  = os.path.basename(file_path)  # "image.png"
 
-    # Assume the image is already uploaded to Hugging Face dataset with this filename
-    # (Or you can modify this to upload programmatically)
-    link = f"https://huggingface.co/datasets/your-username/your-dataset/resolve/main/{file_name}"
+    # 3. GPT-4o: post + video prompt
+    post_text, prompt = generate_post_and_prompt(artist_name)
 
-    # Call the video generator
-    result = generate_video(prompt, num_frames, link)
+    # 4. Call the video generator – `handle_file()` in script.py
+    #    accepts either a local path or a URL, so pass `file_path`
+    result = generate_video(prompt, num_frames, link=file_path)
 
+    # 5. Return outputs to Gradio
     return post_text, result[0]["video"], result[1]
 
 iface = gr.Interface(
     fn=run_pipeline,
     inputs=[
         gr.Textbox(label="Artist Name"),
-        gr.Image(type="filepath", label="Face Image"),
+        gr.Image(type="filepath", label="Face Image"),   # <- keep `filepath`
         gr.Slider(minimum=10, maximum=60, value=20, step=1, label="Number of Frames")
     ],
     outputs=[
@@ -30,8 +33,9 @@ iface = gr.Interface(
         gr.Video(label="Generated Video"),
         gr.Image(label="Thumbnail")
     ],
-    title="Music Artist Video Generator",
+    title="🎬 Music Artist Video Generator",
     description="Upload an artist image and name to generate a video and Xiaohongshu-style post."
 )
 
 iface.launch()
+