Update app.py

app.py

@@ -1,60 +1,77 @@
 import streamlit as st
-import os
 import cv2
 import numpy as np
 from PIL import Image
-from io import BytesIO
-
-# Application title
-st.title("Image to Video Converter")
-st.write("Upload your images, and we'll convert them into a simple video!")
-
-# Directory to save uploaded images
-upload_dir = "uploaded_images"
-os.makedirs(upload_dir, exist_ok=True)
-
-# Step 1: Upload multiple images
-uploaded_files = st.file_uploader("Upload images (at least 2)", type=["jpg", "jpeg", "png"], accept_multiple_files=True)
-
-if uploaded_files:
-    # Display uploaded images
-    st.write("Preview of uploaded images:")
-    images = []
-    for file in uploaded_files:
-        image = Image.open(file)
-        images.append(np.array(image))
-        st.image(image, caption=file.name, width=150)
-
-    # Check if there are at least two images
-    if len(images) >= 2:
-        # Step 2: Choose video settings
-        fps = st.slider("Select frames per second (FPS)", min_value=1, max_value=60, value=10)
-        video_name = st.text_input("Enter the video file name", "output.mp4")
-
-        # Step 3: Generate video
-        if st.button("Generate Video"):
-            # Define video parameters
-            height, width, _ = images[0].shape
-            video_path = os.path.join(upload_dir, video_name)
-            video_writer = cv2.VideoWriter(video_path, cv2.VideoWriter_fourcc(*"mp4v"), fps, (width, height))
+import tempfile
+import os
 
-            # Write images to video
-            for img in images:
-                video_writer.write(cv2.cvtColor(img, cv2.COLOR_RGB2BGR))
+def create_simple_video(images, output_path, fps=1):
+    """
+    Create a simple video from a list of images
+    
+    Args:
+        images (list): List of PIL Image objects
+        output_path (str): Path to save the output video
+        fps (int): Frames per second
+    """
+    # Convert PIL images to OpenCV format
+    cv_images = [cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR) for img in images]
+    
+    # Get dimensions from first image
+    height, width, _ = cv_images[0].shape
+    
+    # Define video writer
+    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+    out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
+    
+    # Write frames
+    for img in cv_images:
+        out.write(img)
+    
+    out.release()
 
-            video_writer.release()
-            
-            # Read video into memory for playback
-            with open(video_path, "rb") as video_file:
-                video_bytes = video_file.read()
-            
-            st.success("Video created successfully!")
-            st.video(video_bytes)
-    else:
-        st.warning("Please upload at least two images to create a video.")
+def main():
+    st.title("🖼️ Image to Simple Video Converter")
+    
+    # Sidebar for configuration
+    st.sidebar.header("Video Settings")
+    fps = st.sidebar.slider("Frames per Second", min_value=1, max_value=5, value=1)
+    
+    # File uploader
+    uploaded_files = st.file_uploader(
+        "Upload Images", 
+        type=['png', 'jpg', 'jpeg', 'webp'], 
+        accept_multiple_files=True
+    )
+    
+    if uploaded_files:
+        # Sort images to maintain order
+        uploaded_files.sort(key=lambda x: x.name)
+        
+        # Convert uploaded files to PIL Images
+        images = [Image.open(file) for file in uploaded_files]
+        
+        # Create temporary file for video
+        with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as temp_video:
+            video_path = temp_video.name
+        
+        # Create video
+        create_simple_video(images, video_path, fps)
+        
+        # Display video
+        st.video(video_path)
+        
+        # Provide download button
+        with open(video_path, 'rb') as video_file:
+            st.download_button(
+                label="Download Video",
+                data=video_file.read(),
+                file_name="converted_video.mp4",
+                mime="video/mp4"
+            )
+        
+        # Clean up temporary file
+        os.unlink(video_path)
 
-# Step 4: Clean up uploaded files
-if st.button("Clear uploaded files"):
-    for file in os.listdir(upload_dir):
-        file_path = os.path.join(upload_dir, file)
-        os.remove
+if __name__ == "__main__":
+    main()