update create_webcam_preview

README.md

@@ -371,11 +371,11 @@ For the latest experimental builds and features, see the [experimental branch](h
 
 **TODO:**
 
+- [x] Support multiple faces 
 - [ ] Develop a version for web app/service
+- [ ] UI/UX enhancements for desktop app
 - [ ] Speed up model loading
 - [ ] Speed up real-time face swapping
-- [x] Support multiple faces 
-- [x] UI/UX enhancements for desktop app
 
 This is an open-source project developed in our free time. Updates may be delayed.
 

modules/globals.py

@@ -35,4 +35,3 @@ log_level = 'error'
 fp_ui: Dict[str, bool] = {}
 camera_input_combobox = None
 webcam_preview_running = False
-opacity = 100 

modules/ui.py

@@ -3,7 +3,6 @@ import webbrowser
 import customtkinter as ctk
 from typing import Callable, Tuple
 import cv2
-from cv2_enumerate_cameras import enumerate_cameras
 from PIL import Image, ImageOps
 import tkinterdnd2 as tkdnd
 import time
@@ -26,7 +25,6 @@ from modules.utilities import (
     has_image_extension,
 )
 
-modules.globals.face_opacity = 100
 os.environ["QT_AUTO_SCREEN_SCALE_FACTOR"] = "1"
 os.environ["QT_SCREEN_SCALE_FACTORS"] = "1"
 os.environ["QT_SCALE_FACTOR"] = "1"
@@ -404,33 +402,11 @@ def create_root(
     )
     preview_button.pack(side="left", padx=10, expand=True)
 
-    # --- Camera Selection ---
-    camera_label = ctk.CTkLabel(root, text="Select Camera:")
-    camera_label.place(relx=0.4, rely=0.86, relwidth=0.2, relheight=0.05)
-    available_cameras = get_available_cameras()
-    # Convert camera indices to strings for CTkOptionMenu
-    available_camera_indices, available_camera_strings = available_cameras
-    camera_variable = ctk.StringVar(
-        value=available_camera_strings[0]
-        if available_camera_strings
-        else "No cameras found"
-    )
-    camera_optionmenu = ctk.CTkOptionMenu(
-        root, variable=camera_variable, values=available_camera_strings
-    )
-    camera_optionmenu.place(relx=0.65, rely=0.86, relwidth=0.2, relheight=0.05)
-    # --- End Camera Selection ---
-
     live_button = ModernButton(
         button_frame,
         text="Live",
         cursor="hand2",
-        command=lambda: webcam_preview(
+        command=lambda: webcam_preview(root),
-            root,
-            available_camera_indices[
-                available_camera_strings.index(camera_variable.get())
-            ],
-        ),
     )
     live_button.pack(side="left", padx=10, expand=True)
 
@@ -706,6 +682,29 @@ def create_preview(parent: ctk.CTkToplevel) -> ctk.CTkToplevel:
     )
     preview_slider.pack(fill="x", padx=20, pady=10)
 
+    last_update_time = 0
+    debounce_delay = 0.1  # Adjust this delay as needed (in seconds)
+
+    def on_key_press(event):
+        nonlocal last_update_time
+
+        current_time = time.time()
+        if current_time - last_update_time > debounce_delay:
+            current_frame = int(preview_slider.get())
+            if event.keysym == "Left":
+                new_frame = max(0, current_frame - 1)
+            elif event.keysym == "Right":
+                new_frame = min(int(preview_slider.cget("to")), current_frame + 1)
+            else:
+                return  # Ignore other key presses
+
+            preview_slider.set(new_frame)
+            update_preview(new_frame)
+            last_update_time = current_time
+
+    preview.bind("<Left>", on_key_press)
+    preview.bind("<Right>", on_key_press)
+
     return preview
 
 
@@ -903,6 +902,11 @@ def init_preview() -> None:
         preview_slider.configure(to=video_frame_total)
         preview_slider.pack(fill="x")
         preview_slider.set(0)
+    # Disable slider if it's an image
+    if is_image(modules.globals.target_path):
+        preview_slider.configure(state="disabled")
+    else:
+        preview_slider.configure(state="normal")
 
 
 def update_preview(frame_number: int = 0) -> None:
@@ -955,75 +959,42 @@ def update_preview(frame_number: int = 0) -> None:
         PREVIEW.deiconify()
 
 
-def webcam_preview(root: ctk.CTk, camera_index: int):
+def webcam_preview(root: ctk.CTk):
     if not modules.globals.map_faces:
         if modules.globals.source_path is None:
             # No image selected
             return
-        create_webcam_preview(camera_index)
+        create_webcam_preview()
     else:
         modules.globals.souce_target_map = []
         create_source_target_popup_for_webcam(root, modules.globals.souce_target_map)
 
 
-def get_available_cameras():
+def create_webcam_preview():
-    """Returns a list of available camera names and indices."""
-    camera_indices = []
-    camera_names = []
-
-    for camera in enumerate_cameras():
-        cap = cv2.VideoCapture(camera.index)
-        if cap.isOpened():
-            camera_indices.append(camera.index)
-            camera_names.append(camera.name)
-            cap.release()
-    return (camera_indices, camera_names)
-
-
-# Add this function to update the opacity value
-def update_opacity(value):
-    modules.globals.face_opacity = int(value)
-
-
-# Modify the create_webcam_preview function to include the slider
-def create_webcam_preview(camera_index):
     global preview_label, PREVIEW
+    stream_url = 'http://192.168.0.103:4747'
+    camera = cv2.VideoCapture(
+        # stream_url
+        #7
+        1, cv2.CAP_DSHOW
+    )  # Use index for the webcam (adjust the index accordingly if necessary)
+    camera.set(
+        cv2.CAP_PROP_FRAME_WIDTH, PREVIEW_DEFAULT_WIDTH
+    )  # Set the width of the resolution
+    camera.set(
+        cv2.CAP_PROP_FRAME_HEIGHT, PREVIEW_DEFAULT_HEIGHT
+    )  # Set the height of the resolution
+    camera.set(cv2.CAP_PROP_FPS, 60)  # Set the frame rate of the webcam
 
-    camera = cv2.VideoCapture(camera_index)
+    preview_label.configure(
-    if not camera.isOpened():
+        width=PREVIEW_DEFAULT_WIDTH, height=PREVIEW_DEFAULT_HEIGHT
-        update_status(f"Error: Could not open camera with index {camera_index}")
+    )  # Reset the preview image before startup
-        return
-    camera.set(cv2.CAP_PROP_FRAME_WIDTH, PREVIEW_DEFAULT_WIDTH)
-    camera.set(cv2.CAP_PROP_FRAME_HEIGHT, PREVIEW_DEFAULT_HEIGHT)
-    camera.set(cv2.CAP_PROP_FPS, 60)
 
-    PREVIEW.deiconify()
+    PREVIEW.deiconify()  # Open preview window
-
-    # Remove any existing widgets in PREVIEW window
-    for widget in PREVIEW.winfo_children():
-        widget.destroy()
-
-    # Create a main frame to hold all widgets
-    main_frame = ctk.CTkFrame(PREVIEW)
-    main_frame.pack(fill="both", expand=True)
-
-    # Create a frame for the preview label
-    preview_frame = ctk.CTkFrame(main_frame)
-    preview_frame.pack(fill="both", expand=True, padx=10, pady=10)
-
-    preview_label = ctk.CTkLabel(preview_frame, text="")
-    preview_label.pack(fill="both", expand=True)
 
     frame_processors = get_frame_processors_modules(modules.globals.frame_processors)
 
-    source_image = None
+    source_image = None  # Initialize variable for the selected face image
-
-    def update_frame_size(event):
-        nonlocal temp_frame
-        if modules.globals.live_resizable:
-            temp_frame = fit_image_to_size(temp_frame, event.width, event.height)
-
-    preview_frame.bind("<Configure>", update_frame_size)
 
     while camera:
         ret, frame = camera.read()
@@ -1041,6 +1012,7 @@ def create_webcam_preview(camera_index):
             )
 
         if not modules.globals.map_faces:
+            # Select and save face image only once
             if source_image is None and modules.globals.source_path:
                 source_image = get_one_face(cv2.imread(modules.globals.source_path))
 

requirements.txt

@@ -2,7 +2,6 @@
 
 numpy>=1.23.5,<2
 opencv-python==4.8.1.78
-cv2_enumerate_cameras==1.1.15
 onnx==1.16.0
 insightface==0.7.3
 psutil==5.9.8