Update app.py

app.py

@@ -1,9 +1,11 @@
 import streamlit as st
 import cv2
 import numpy as np
-from moviepy.editor import VideoFileClip
 from PIL import Image
-import io
+import tempfile
+import os
+from moviepy.editor import VideoFileClip
+from moviepy.editor import vfx
 
 # Function to apply filters
 def apply_filter(frame, filter_name):
@@ -15,7 +17,10 @@ def apply_filter(frame, filter_name):
         return cv2.convertScaleAbs(frame, alpha=1.5, beta=0)
     elif filter_name == 'Contrast':
         return cv2.convertScaleAbs(frame, alpha=2, beta=0)
-    # More filters can be added here
+    elif filter_name == 'Saturation':
+        hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
+        hsv[...,1] = hsv[...,1] * 1.5  # increase saturation
+        return cv2.cvtColor(hsv, cv2.COLOR_HSV2BGR)
     else:
         return frame
 
@@ -27,23 +32,32 @@ st.write("Upload a video and apply filters")
 video_file = st.file_uploader("Upload your video", type=["mp4", "avi", "mov"])
 
 if video_file:
-    # Load the video
+    # Display the uploaded video
     st.video(video_file)
 
     # Select filter
     filter_choice = st.selectbox("Select a filter", ["None", "Grayscale", "Blur", "Sharpness", "Contrast", "Saturation"])
 
-    # Process the video
+    # Process the video and apply filter
     if st.button("Apply Filter and Download"):
-        video_clip = VideoFileClip(video_file)
-        temp_output = "output_video.mp4"
+        # Temporary file to store the video
+        with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as tmp_file:
+            tmp_file.write(video_file.read())
+            video_path = tmp_file.name
+
+        video_clip = VideoFileClip(video_path)
         
+        # Apply filter to video
         def process_frame(frame):
             return apply_filter(frame, filter_choice)
-        
+
         processed_video = video_clip.fl_image(process_frame)
-        processed_video.write_videofile(temp_output)
-        
+
+        # Saving the output video
+        output_path = os.path.join("output", "edited_video.mp4")
+        os.makedirs(os.path.dirname(output_path), exist_ok=True)
+        processed_video.write_videofile(output_path)
+
         # Provide download link
-        with open(temp_output, "rb") as f:
+        with open(output_path, "rb") as f:
             st.download_button("Download edited video", f, file_name="edited_video.mp4")