inital deployment1

app.py

@@ -10,22 +10,8 @@ from PIL import Image
 device = "cuda" if torch.cuda.is_available() else "cpu"
 realtime_pipe = StableDiffusionImg2ImgPipeline.from_pretrained("radames/Real-Time-Latent-Consistency-Model").to(device)
 
-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", prompt="A futuristic landscape"):
-    """Process a single frame by applying a color filter and real-time latent consistency model."""
-    frame = apply_color_filter(frame, filter_type)
+def process_frame(frame, prompt="A futuristic landscape"):
+    """Process a single frame using the real-time latent consistency model."""
     
     # Convert frame to PIL image
     image = Image.fromarray(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)).resize((512, 512))
@@ -34,15 +20,15 @@ def process_frame(frame, filter_type="None", prompt="A futuristic landscape"):
     result = realtime_pipe(prompt=prompt, image=image, strength=0.5, guidance_scale=7.5).images[0]
     return np.array(result)
 
-def video_stream(filter_type, prompt):
-    """Captures video feed from webcam, applies color filter, and sends to the AI model."""
+def video_stream(prompt):
+    """Captures video feed from webcam and sends to the AI model."""
     cap = cv2.VideoCapture(0)
     while cap.isOpened():
         ret, frame = cap.read()
         if not ret:
             break
         
-        frame = process_frame(frame, filter_type, prompt)
+        frame = process_frame(frame, prompt)
         yield frame  # Return processed frame
     cap.release()
 
@@ -53,10 +39,10 @@ with gr.Blocks() as demo:
     with gr.Row():
         webcam_feed = gr.Camera(streaming=True, label="Live Webcam")
         processed_image = gr.Image(label="AI-Enhanced Webcam Feed")
+        canvas = gr.Image(interactive=True, label="Canvas - Edit Processed Image")
     
-    filter_selector = gr.Radio(["None", "Red", "Green", "Blue"], label="Color Filter")
     prompt_input = gr.Textbox(label="Real-Time Latent Consistency Model Prompt", value="A futuristic landscape")
     
-    webcam_feed.change(fn=video_stream, inputs=[filter_selector, prompt_input], outputs=processed_image)
+    webcam_feed.change(fn=video_stream, inputs=[prompt_input], outputs=[processed_image, canvas])
 
 demo.launch(share=True)