Revert "Merge pull request #685 from KRSHH/main"
This reverts commitpull/691/headd4e5b8078d, reversing changes made toc08bec22e3.
							parent
							
								
									3d741bd269
								
							
						
					
					
						commit
						60e27f4755
					
				|  | @ -34,5 +34,4 @@ headless = None | ||||||
| log_level = 'error' | log_level = 'error' | ||||||
| fp_ui: Dict[str, bool] = {} | fp_ui: Dict[str, bool] = {} | ||||||
| camera_input_combobox = None | camera_input_combobox = None | ||||||
| webcam_preview_running = False | webcam_preview_running = False | ||||||
| opacity = 100  |  | ||||||
|  | @ -25,7 +25,6 @@ from modules.utilities import ( | ||||||
|     has_image_extension, |     has_image_extension, | ||||||
| ) | ) | ||||||
| 
 | 
 | ||||||
| modules.globals.face_opacity = 100 |  | ||||||
| os.environ["QT_AUTO_SCREEN_SCALE_FACTOR"] = "1" | os.environ["QT_AUTO_SCREEN_SCALE_FACTOR"] = "1" | ||||||
| os.environ["QT_SCREEN_SCALE_FACTORS"] = "1" | os.environ["QT_SCREEN_SCALE_FACTORS"] = "1" | ||||||
| os.environ["QT_SCALE_FACTOR"] = "1" | os.environ["QT_SCALE_FACTOR"] = "1" | ||||||
|  | @ -455,29 +454,6 @@ def create_root( | ||||||
|     ) |     ) | ||||||
|     remove_donate_button.pack(side="right", padx=(10, 0)) |     remove_donate_button.pack(side="right", padx=(10, 0)) | ||||||
| 
 | 
 | ||||||
|     # Add opacity slider |  | ||||||
|     opacity_frame = ctk.CTkFrame(options_column, fg_color="#2a2d2e") |  | ||||||
|     opacity_frame.pack(pady=5, anchor="w", fill="x") |  | ||||||
| 
 |  | ||||||
|     opacity_label = ctk.CTkLabel( |  | ||||||
|         opacity_frame, text="Face Opacity:", font=("Roboto", 14, "bold") |  | ||||||
|     ) |  | ||||||
|     opacity_label.pack(side="left", padx=(0, 10)) |  | ||||||
| 
 |  | ||||||
|     opacity_slider = ctk.CTkSlider( |  | ||||||
|         opacity_frame, |  | ||||||
|         from_=0, |  | ||||||
|         to=100, |  | ||||||
|         number_of_steps=100, |  | ||||||
|         command=update_opacity, |  | ||||||
|         fg_color=("gray75", "gray25"), |  | ||||||
|         progress_color="#3a7ebf", |  | ||||||
|         button_color="#3a7ebf", |  | ||||||
|         button_hover_color="#2b5d8b", |  | ||||||
|     ) |  | ||||||
|     opacity_slider.pack(side="left", fill="x", expand=True) |  | ||||||
|     opacity_slider.set(modules.globals.face_opacity) |  | ||||||
| 
 |  | ||||||
|     main_frame.grid_columnconfigure((0, 2), weight=1) |     main_frame.grid_columnconfigure((0, 2), weight=1) | ||||||
|     main_frame.grid_rowconfigure((0, 1, 2), weight=1) |     main_frame.grid_rowconfigure((0, 1, 2), weight=1) | ||||||
| 
 | 
 | ||||||
|  | @ -994,83 +970,64 @@ def webcam_preview(root: ctk.CTk): | ||||||
|         create_source_target_popup_for_webcam(root, modules.globals.souce_target_map) |         create_source_target_popup_for_webcam(root, modules.globals.souce_target_map) | ||||||
| 
 | 
 | ||||||
| 
 | 
 | ||||||
| # 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(): | def create_webcam_preview(): | ||||||
|     global preview_label, PREVIEW |     global preview_label, PREVIEW | ||||||
| 
 | 
 | ||||||
|     camera = cv2.VideoCapture(0) |     camera = cv2.VideoCapture( | ||||||
|     camera.set(cv2.CAP_PROP_FRAME_WIDTH, PREVIEW_DEFAULT_WIDTH) |         0 | ||||||
|     camera.set(cv2.CAP_PROP_FRAME_HEIGHT, PREVIEW_DEFAULT_HEIGHT) |     )  # Use index for the webcam (adjust the index accordingly if necessary) | ||||||
|     camera.set(cv2.CAP_PROP_FPS, 60) |     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 | ||||||
| 
 | 
 | ||||||
|     PREVIEW.deiconify() |     preview_label.configure( | ||||||
|  |         width=PREVIEW_DEFAULT_WIDTH, height=PREVIEW_DEFAULT_HEIGHT | ||||||
|  |     )  # Reset the preview image before startup | ||||||
| 
 | 
 | ||||||
|     # Remove any existing widgets in PREVIEW window |     PREVIEW.deiconify()  # Open 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) |     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: |     while camera: | ||||||
|         ret, frame = camera.read() |         ret, frame = camera.read() | ||||||
|         if not ret: |         if not ret: | ||||||
|             break |             break | ||||||
| 
 | 
 | ||||||
|         temp_frame = frame.copy() |         temp_frame = frame.copy()  # Create a copy of the frame | ||||||
| 
 | 
 | ||||||
|         if modules.globals.live_mirror: |         if modules.globals.live_mirror: | ||||||
|             temp_frame = cv2.flip(temp_frame, 1) |             temp_frame = cv2.flip(temp_frame, 1)  # horizontal flipping | ||||||
|  | 
 | ||||||
|  |         if modules.globals.live_resizable: | ||||||
|  |             temp_frame = fit_image_to_size( | ||||||
|  |                 temp_frame, PREVIEW.winfo_width(), PREVIEW.winfo_height() | ||||||
|  |             ) | ||||||
| 
 | 
 | ||||||
|         if not modules.globals.map_faces: |         if not modules.globals.map_faces: | ||||||
|  |             # Select and save face image only once | ||||||
|             if source_image is None and modules.globals.source_path: |             if source_image is None and modules.globals.source_path: | ||||||
|                 source_image = get_one_face(cv2.imread(modules.globals.source_path)) |                 source_image = get_one_face(cv2.imread(modules.globals.source_path)) | ||||||
| 
 | 
 | ||||||
|             original_frame = temp_frame.copy() |  | ||||||
|             for frame_processor in frame_processors: |             for frame_processor in frame_processors: | ||||||
|                 temp_frame = frame_processor.process_frame(source_image, temp_frame) |                 temp_frame = frame_processor.process_frame(source_image, temp_frame) | ||||||
| 
 |  | ||||||
|             # Apply opacity |  | ||||||
|             opacity = modules.globals.face_opacity / 100 |  | ||||||
|             temp_frame = cv2.addWeighted( |  | ||||||
|                 temp_frame, opacity, original_frame, 1 - opacity, 0 |  | ||||||
|             ) |  | ||||||
|         else: |         else: | ||||||
|             modules.globals.target_path = None |             modules.globals.target_path = None | ||||||
| 
 | 
 | ||||||
|             for frame_processor in frame_processors: |             for frame_processor in frame_processors: | ||||||
|                 temp_frame = frame_processor.process_frame_v2(temp_frame) |                 temp_frame = frame_processor.process_frame_v2(temp_frame) | ||||||
| 
 | 
 | ||||||
|         image = cv2.cvtColor(temp_frame, cv2.COLOR_BGR2RGB) |         image = cv2.cvtColor( | ||||||
|  |             temp_frame, cv2.COLOR_BGR2RGB | ||||||
|  |         )  # Convert the image to RGB format to display it with Tkinter | ||||||
|         image = Image.fromarray(image) |         image = Image.fromarray(image) | ||||||
|         image = ImageOps.contain( |         image = ImageOps.contain( | ||||||
|             image, |             image, (temp_frame.shape[1], temp_frame.shape[0]), Image.LANCZOS | ||||||
|             (preview_frame.winfo_width(), preview_frame.winfo_height()), |  | ||||||
|             Image.LANCZOS, |  | ||||||
|         ) |         ) | ||||||
|         image = ctk.CTkImage(image, size=image.size) |         image = ctk.CTkImage(image, size=image.size) | ||||||
|         preview_label.configure(image=image) |         preview_label.configure(image=image) | ||||||
|  | @ -1080,7 +1037,7 @@ def create_webcam_preview(): | ||||||
|             break |             break | ||||||
| 
 | 
 | ||||||
|     camera.release() |     camera.release() | ||||||
|     PREVIEW.withdraw() |     PREVIEW.withdraw()  # Close preview window when loop is finished | ||||||
| 
 | 
 | ||||||
| 
 | 
 | ||||||
| def create_source_target_popup_for_webcam(root: ctk.CTk, map: list) -> None: | def create_source_target_popup_for_webcam(root: ctk.CTk, map: list) -> None: | ||||||
|  |  | ||||||
		Loading…
	
		Reference in New Issue