Update app.py

app.py

@@ -4,24 +4,16 @@ import os
 import tempfile
 from pathlib import Path
 
-def images_to_video(input_files, fps=30):
-    # Create a temporary directory to store images
+def image_to_video(input_file, fps=30):
+    # Create a temporary directory to store the image
     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}.jpg")
-            with open(file_path, 'wb') as file:
-                file.write(file_info)
-        
-        # 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.")
+        # Save the uploaded file to the temp directory
+        file_path = os.path.join(temp_dir, f"image{Path(input_file.name).suffix}")
+        with open(file_path, 'wb') as file:
+            file.write(input_file.read())
 
-        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]))
+        # Read the image to set the video size
+        frame = cv2.imread(file_path)
         height, width, layers = frame.shape
         size = (width, height)
         
@@ -31,10 +23,8 @@ def images_to_video(input_files, fps=30):
         # 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)
+        # Add the image to the video
+        out.write(frame)
         
         # Release the video writer
         out.release()
@@ -43,11 +33,11 @@ def images_to_video(input_files, fps=30):
         return video_filename
 
 iface = gr.Interface(
-    fn=images_to_video,
-    inputs=[gr.File(type="binary", label="Upload Images", multiple=True)],
+    fn=image_to_video,
+    inputs=gr.File(type="binary", label="Upload Image"),
     outputs=gr.Video(label="Output Video"),
-    title="Images to Video Converter",
-    description="Upload multiple images to create a video."
+    title="Image to Video Converter",
+    description="Upload an image to create a video."
 )
 
 iface.launch()