fi0r1/roop
1
0
Fork 0
mirror of https://github.com/s0md3v/roop.git synced 2025-12-06 18:08:29 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Simplify is_image and is_video, close preview on source/target change, show preview slider on video only, fix start button error

Browse Source
This commit is contained in:
henryruhs 2023-06-12 00:30:15 +02:00
parent b78164895b
commit 1d2d565aef
2 changed files with 17 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
roop/ui.py
View File

@ -96,8 +96,7 @@ def create_preview(parent) -> ctk.CTkToplevel:
preview_label = ctk.CTkLabel(preview, text=None)
preview_label.pack(fill='both', expand=True)
preview_slider = ctk.CTkSlider(preview, from_=0, to=0, border_width=10, command=lambda frame_value: update_preview(frame_value))
preview_slider.pack(fill='x')
preview_slider = ctk.CTkSlider(preview, from_=0, to=0, command=lambda frame_value: update_preview(frame_value))
return preview
@ -110,6 +109,7 @@ def update_status(text: str) -> None:
def select_source_path() -> None:
global RECENT_DIRECTORY_SOURCE
PREVIEW.withdraw()
source_path = ctk.filedialog.askopenfilename(title='Select an face image', initialdir=RECENT_DIRECTORY_SOURCE)
if is_image(source_path):
roop.globals.source_path = source_path
@ -126,6 +126,7 @@ def select_source_path() -> None:
def select_target_path() -> None:
global RECENT_DIRECTORY_TARGET
PREVIEW.withdraw()
target_path = ctk.filedialog.askopenfilename(title='Select an image or video target', initialdir=RECENT_DIRECTORY_TARGET)
if is_image(target_path):
roop.globals.target_path = target_path
@ -152,6 +153,8 @@ def select_output_path(start):
output_path = ctk.filedialog.asksaveasfilename(title='Save image output', initialfile='output.png', initialdir=RECENT_DIRECTORY_OUTPUT)
elif is_video(roop.globals.target_path):
output_path = ctk.filedialog.asksaveasfilename(title='Save video output', initialfile='output.mp4', initialdir=RECENT_DIRECTORY_OUTPUT)
else:
output_path = None
if output_path:
roop.globals.output_path = output_path
RECENT_DIRECTORY_OUTPUT = os.path.dirname(roop.globals.output_path)
@ -182,16 +185,20 @@ def render_video_preview(video_path: str, dimensions: Tuple[int, int] = None, fr
def toggle_preview() -> None:
if PREVIEW.state() == 'normal':
PREVIEW.withdraw()
else:
elif roop.globals.source_path and roop.globals.target_path:
init_preview()
update_preview(0)
update_preview()
PREVIEW.deiconify()
def init_preview() -> None:
if roop.globals.target_path:
if is_image(roop.globals.target_path):
preview_slider.pack_forget()
if is_video(roop.globals.target_path):
video_frame_total = get_video_frame_total(roop.globals.target_path)
preview_slider.configure(to=video_frame_total)
preview_slider.pack(fill='x')
preview_slider.set(0)
def update_preview(frame_number: int = 0) -> None:

19
roop/utilities.py
View File

@ -1,4 +1,5 @@
import glob
import mimetypes
import os
import shutil
import subprocess
@ -94,25 +95,15 @@ def has_image_extension(image_path: str) -> bool:
def is_image(image_path: str) -> bool:
if image_path and os.path.isfile(image_path):
try:
image = Image.open(image_path)
image.verify()
return True
except Exception:
pass
mimetype, _ = mimetypes.guess_type(image_path)
return mimetype and mimetype.startswith('image/')
return False
def is_video(video_path: str) -> bool:
if video_path and os.path.isfile(video_path):
try:
capture = cv2.VideoCapture(video_path)
if capture.isOpened():
is_video, _ = capture.read()
capture.release()
return is_video
except Exception:
pass
mimetype, _ = mimetypes.guess_type(video_path)
return mimetype and mimetype.startswith('video/')
return False