Update app.py

app.py

@@ -25,7 +25,7 @@ def process_image(image_path):
             width = int(height * aspect_ratio)
         image = image.resize((width, height), Image.ANTIALIAS)
 
-    # 画像を5秒の動画にするためのフレーム数を決定（例えば30fpsで5秒分なので150フレーム）
+    # 動画のフレーム数（30fpsで5秒分）
     fps = 30
     duration = 5
     total_frames = fps * duration
@@ -37,8 +37,8 @@ def process_image(image_path):
         frame = np.array(image)
         frames.append(frame)
 
-    # 動画保存先のパス
-    output_video_path = os.path.splitext(image_path)[0] + '.mp4'
+    # 動画保存先の一時ファイルパス
+    output_video_path = "/tmp/output_video.mp4"
 
     # 動画ファイルにフレームを保存
     fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # MP4形式
@@ -51,6 +51,7 @@ def process_image(image_path):
 
     video_writer.release()
     
+    # 動画ファイルをGradioに返す
     return output_video_path
 
 class WebUI:
@@ -70,7 +71,7 @@ class WebUI:
                 with gr.Row():
                     with gr.Column():
                         with gr.Tab("Output"):
-                            output_mp4 = gr.Gallery()
+                            output_mp4 = gr.Video()
             submit.click(self.main, inputs=[input_image], outputs=[output_mp4])
 
         self.demo.queue()