Update app.py

app.py

@@ -23,9 +23,9 @@ def generate_video(image):
     response.raise_for_status()
     response_body = response.json()
 
-    # The response from the API might include a video URL or base64-encoded video data
-    # Adjust the return value based on the actual response
-    return response_body  # Modify this line based on the actual API response format
+    # Assuming the response contains a URL to the generated video
+    video_url = response_body.get('video_url')
+    return video_url
 
 # Create a Gradio interface with a heading
 def create_interface():
@@ -34,8 +34,8 @@ def create_interface():
     # Gradio Interface
     iface = gr.Interface(
         fn=generate_video, 
-        inputs=gr.inputs.Image(type="file", label="Upload Your Product Image"), 
-        outputs=gr.outputs.Video(label="Generated Product Video"),
+        inputs=gr.Image(type="file", label="Upload Your Product Image"), 
+        outputs=gr.Video(label="Generated Product Video"),  # Use gr.Video to display the video output
         title=title,
         description="Upload a product shot to generate a moving video version of it."
     )