inital deployment1

app.py

@@ -21,9 +21,24 @@ def generate_image(prompt: str):
     image = pipe(prompt, num_inference_steps=4).images[0]
     return image
 
-def process_frame(frame):
-    """Process each frame (convert to grayscale as an example)"""
-    frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
+
+
+def apply_color_filter(frame, filter_type="None"):
+    """Apply a color filter to the frame."""
+    if filter_type == "Red":
+        frame[:, :, 1] = 0  # Remove green channel
+        frame[:, :, 2] = 0  # Remove blue channel
+    elif filter_type == "Green":
+        frame[:, :, 0] = 0  # Remove red channel
+        frame[:, :, 2] = 0  # Remove blue channel
+    elif filter_type == "Blue":
+        frame[:, :, 0] = 0  # Remove red channel
+        frame[:, :, 1] = 0  # Remove green channel
+    return frame
+
+def process_frame(frame, filter_type="None"):
+    """Process a single frame by applying a color filter."""
+    frame = apply_color_filter(frame, filter_type)
     return frame
 
 def video_stream(frame):