Update app.py

app.py

@@ -4,12 +4,11 @@ import numpy as np
 import tempfile
 import os
 
-# Function to apply filters
+# Function to apply selected filter
 def apply_filter(frame, filter_name, filter_param):
     if filter_name == 'Grayscale':
         return cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
     elif filter_name == 'Blur':
-        # Adjustable blur radius using the filter_param
         return cv2.GaussianBlur(frame, (filter_param, filter_param), 0)
     elif filter_name == 'Sharpness':
         return cv2.convertScaleAbs(frame, alpha=filter_param, beta=0)
@@ -19,68 +18,65 @@ def apply_filter(frame, filter_name, filter_param):
         hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
         hsv[..., 1] = hsv[..., 1] * filter_param  # Adjust saturation
         return cv2.cvtColor(hsv, cv2.COLOR_HSV2BGR)
-    elif filter_name == 'Pen':
-        # Adjustable intensity of "pen" effect
-        return cv2.addWeighted(frame, filter_param, np.zeros_like(frame), 0, 50)
-    elif filter_name == 'Magic Eraser':
-        # Adjustable threshold for the "eraser" effect
-        lower_white = np.array([200, 200, 200]) * filter_param
-        upper_white = np.array([255, 255, 255])
-        mask = cv2.inRange(frame, lower_white, upper_white)
-        frame[mask > 0] = [0, 0, 0]  # Turn white areas into black
-        return frame
+    elif filter_name == 'Brightness':
+        return cv2.convertScaleAbs(frame, alpha=1, beta=filter_param)
+    elif filter_name == 'Tint':
+        tint_color = np.array([filter_param, 0, 0])  # Red tint for example
+        return cv2.addWeighted(frame, 1, tint_color, 0.2, 0)
     else:
         return frame
 
 # Streamlit UI
-st.title("Video Editor")
-st.write("Upload a video and apply adjustable filters")
+st.title("Real-Time Video Editor")
+st.write("Upload a video, apply filters, and adjust in real-time!")
 
 # Video upload
 video_file = st.file_uploader("Upload your video", type=["mp4", "avi", "mov"])
 
 if video_file:
-    # Display the uploaded video
+    # Show the uploaded video
     st.video(video_file)
 
-    # Select filter
+    # Select filter from dropdown menu
     filter_choice = st.selectbox("Select a filter", [
-        "None", "Grayscale", "Blur", "Sharpness", "Contrast", "Saturation", "Pen", "Magic Eraser"])
+        "None", "Grayscale", "Blur", "Sharpness", "Contrast", "Saturation", "Brightness", "Tint"])
 
-    # Filter parameter adjustments based on the selected filter
+    # Adjustable parameters for each filter
     if filter_choice == "Blur":
         filter_param = st.slider("Blur Radius", min_value=3, max_value=21, step=2, value=5)
-    elif filter_choice in ["Sharpness", "Contrast"]:
-        filter_param = st.slider("Intensity", min_value=1.0, max_value=5.0, step=0.1, value=1.5)
+    elif filter_choice == "Sharpness":
+        filter_param = st.slider("Sharpness Factor", min_value=1.0, max_value=5.0, step=0.1, value=1.5)
+    elif filter_choice == "Contrast":
+        filter_param = st.slider("Contrast Intensity", min_value=1.0, max_value=5.0, step=0.1, value=2.0)
     elif filter_choice == "Saturation":
         filter_param = st.slider("Saturation Multiplier", min_value=1.0, max_value=2.5, step=0.1, value=1.5)
-    elif filter_choice == "Pen":
-        filter_param = st.slider("Pen Effect Intensity", min_value=0.1, max_value=2.0, step=0.1, value=0.5)
-    elif filter_choice == "Magic Eraser":
-        filter_param = st.slider("Eraser Threshold", min_value=0.5, max_value=2.0, step=0.1, value=1.0)
+    elif filter_choice == "Brightness":
+        filter_param = st.slider("Brightness Adjustment", min_value=-100, max_value=100, step=5, value=0)
+    elif filter_choice == "Tint":
+        filter_param = st.slider("Tint Level", min_value=0, max_value=255, step=1, value=100)
     else:
-        filter_param = 0  # No parameter needed for "None"
+        filter_param = 0  # No filter parameter required for 'None'
 
-    # Process the video and apply filter
+    # Process the video and show the result dynamically as filter is applied
     if st.button("Apply Filter and Download"):
         # 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
 
-        # Open video file with OpenCV
+        # Open the video file
         cap = cv2.VideoCapture(video_path)
         frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
         frame_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
         fps = cap.get(cv2.CAP_PROP_FPS)
 
-        # Create output file
+        # Output path for the edited video
         output_path = os.path.join("output", "edited_video.mp4")
         os.makedirs(os.path.dirname(output_path), exist_ok=True)
         fourcc = cv2.VideoWriter_fourcc(*'mp4v')
         out = cv2.VideoWriter(output_path, fourcc, fps, (frame_width, frame_height))
 
-        # Read and process frames
+        # Read the video frame by frame and apply the selected filter
         while cap.isOpened():
             ret, frame = cap.read()
             if not ret:
@@ -88,7 +84,7 @@ if video_file:
             processed_frame = apply_filter(frame, filter_choice, filter_param)
             out.write(processed_frame)
 
-        # Release the video objects
+        # Release video objects
         cap.release()
         out.release()