Update app.py

app.py

@@ -1,32 +1,54 @@
 import cv2
+import gradio as gr
 import os
-
-def images_to_video(image_folder, output_video_file, fps=30):
-    # Get all the image files from the provided folder
-    images = [img for img in os.listdir(image_folder) if img.endswith(".png") or img.endswith(".jpg") or img.endswith(".jpeg")]
-
-    # Ensure the images are in the correct order if needed
-    images.sort()  # You might want to customize this logic
-
-    # Get the dimensions of the first image
-    frame = cv2.imread(os.path.join(image_folder, images[0]))
-    height, width, layers = frame.shape
-    size = (width, height)
-
-    # Initialize the video writer
-    out = cv2.VideoWriter(output_video_file, cv2.VideoWriter_fourcc(*'mp4v'), fps, size)
-
-    # Loop through all images and add to the video
-    for image in images:
-        frame = cv2.imread(os.path.join(image_folder, image))
-        out.write(frame)
-
-    # Release the video writer
-    out.release()
-
-# Define the image folder and the output video file
-image_folder = 'path_to_image_folder'  # Replace with the path to your image folder
-output_video_file = 'output_video.mp4'  # Replace with your desired output file name
-
-# Create the video
-images_to_video(image_folder, output_video_file)
+import tempfile
+from pathlib import Path
+
+def images_to_video(input_files, fps=30):
+    # Create a temporary directory to store images
+    with tempfile.TemporaryDirectory() as temp_dir:
+        # Save uploaded files to temp directory
+        for i, file_info in enumerate(input_files):
+            file_path = os.path.join(temp_dir, f"image_{i}{Path(file_info.name).suffix}")
+            with open(file_path, 'wb') as file:
+                file.write(file_info.read())
+        # Get all the image files from temp directory
+        images = [img for img in os.listdir(temp_dir) if img.lower().endswith(('.png', '.jpg', '.jpeg'))]
+        if not images:
+            raise ValueError("No images found to create a video.")
+
+        images.sort()  # Sort the images by name
+        
+        # Read the first image to set the video size
+        frame = cv2.imread(os.path.join(temp_dir, images[0]))
+        height, width, layers = frame.shape
+        size = (width, height)
+        
+        # Create a unique filename for the output video
+        video_filename = os.path.join('/mnt/data', 'output_video.mp4')
+        
+        # Initialize the video writer
+        out = cv2.VideoWriter(video_filename, cv2.VideoWriter_fourcc(*'mp4v'), fps, size)
+        
+        # Loop through all images and add to the video
+        for image in images:
+            frame = cv2.imread(os.path.join(temp_dir, image))
+            out.write(frame)
+        
+        # Release the video writer
+        out.release()
+        
+        # Return the path of the created video file
+        return video_filename
+
+# Create the Gradio interface
+iface = gr.Interface(
+    fn=images_to_video,
+    inputs=[gr.inputs.Image(type="file", label="Upload Images", multiple=True)],
+    outputs=gr.outputs.Video(label="Output Video"),
+    title="Images to Video Converter",
+    description="Upload multiple images to create a video."
+)
+
+# Launch the interface
+iface.launch()