apps/roop
1
0
Fork 0
mirror of https://github.com/s0md3v/roop.git synced 2024-08-24 08:31:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Functionional style ui

Browse Source
This commit is contained in:
K1llM@n 2023-06-03 17:47:59 +03:00
parent 6288cdce65
commit 2c91270090
2 changed files with 95 additions and 90 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
roop/core.py
View File

@ -122,6 +122,7 @@ def preview_video(video_path):
cap.release() cap.release()
return (amount_of_frames, frame) return (amount_of_frames, frame)
def status(string): def status(string):
value = "Status: " + string value = "Status: " + string
if 'cli_mode' in args: if 'cli_mode' in args:
@ -130,7 +131,7 @@ def status(string):
ui.update_status_label(value) ui.update_status_label(value)
def start(): def start(preview_callback = None):
if not args['source_img'] or not os.path.isfile(args['source_img']): if not args['source_img'] or not os.path.isfile(args['source_img']):
print("\n[WARNING] Please select an image containing a face.") print("\n[WARNING] Please select an image containing a face.")
return return
@ -173,7 +174,7 @@ def start():
key=lambda x: int(x.split(sep)[-1].replace(".png", "")) key=lambda x: int(x.split(sep)[-1].replace(".png", ""))
)) ))
status("swapping in progress...") status("swapping in progress...")
process_video(args['source_img'], args["frame_paths"], ui.preview.update if ui.preview else None) process_video(args['source_img'], args["frame_paths"], preview_callback)
status("creating video...") status("creating video...")
create_video(video_name, exact_fps, output_dir) create_video(video_name, exact_fps, output_dir)
status("adding audio...") status("adding audio...")
@ -215,6 +216,7 @@ def create_test_preview(frame_number):
None None
) )
def run(): def run():
global all_faces, keep_frames, limit_fps global all_faces, keep_frames, limit_fps

179
roop/ui.py
View File

@ -8,88 +8,83 @@ import threading
from roop.utils import is_img from roop.utils import is_img
class PreviewWindow: max_preview_size = 800
def __init__(self, master):
self.max_preview_size = 800
self.master = master def create_preview(parent):
self.window = tk.Toplevel(self.master) global preview_image_frame, preview_frame_slider, test_button
# Override close button
self.window.protocol("WM_DELETE_WINDOW", self.hide)
self.window.withdraw()
self.window.title("Preview")
self.window.configure(bg="red")
self.window.resizable(width=False, height=False)
self.visible = False preview_window = tk.Toplevel(parent)
self.frame = tk.Frame(self.window, background="#2d3436") # Override close button
self.frame.pack(fill='both', side='left', expand='True') preview_window.protocol("WM_DELETE_WINDOW", hide_preview)
preview_window.withdraw()
# Preview image preview_window.title("Preview")
self.img_label = tk.Label(self.frame) preview_window.configure(bg="red")
self.img_label.pack(side='top') preview_window.resizable(width=False, height=False)
# Bottom frame frame = tk.Frame(preview_window, background="#2d3436")
buttons_frame = tk.Frame(self.frame, background="#2d3436") frame.pack(fill='both', side='left', expand='True')
buttons_frame.pack(fill='both', side='bottom')
self.current_frame = tk.IntVar()
self.frame_slider = tk.Scale(
buttons_frame,
from_=0,
to=0,
orient='horizontal',
variable=self.current_frame,
command=self.slider_changed
)
self.frame_slider.pack(fill='both', side='left', expand='True')
self.test_button = tk.Button(buttons_frame, text="Test", bg="#f1c40f", relief="flat", width=15, borderwidth=0, highlightthickness=0)
self.test_button.pack( side='right', fill='y')
def init_slider(self, frames_count, change_handler):
self.frame_change = change_handler
self.frame_slider.configure(to=frames_count)
self.frame_slider.set(0)
def slider_changed(self, event):
self.frame_change(self.frame_slider.get())
def set_preview_handler(self, test_handler):
self.test_button.config(command = test_handler)
# Show the window
def show(self):
self.visible = True
self.window.deiconify()
# Hide the window # Preview image
def hide(self): preview_image_frame = tk.Label(frame)
self.visible = False preview_image_frame.pack(side='top')
self.window.withdraw()
def update(self, frame): # Bottom frame
if not self.visible: buttons_frame = tk.Frame(frame, background="#2d3436")
return buttons_frame.pack(fill='both', side='bottom')
img = Image.fromarray(frame) current_frame = tk.IntVar()
width, height = img.size preview_frame_slider = tk.Scale(
aspect_ratio = 1 buttons_frame,
if width > height: from_=0,
aspect_ratio = self.max_preview_size / width to=0,
else: orient='horizontal',
aspect_ratio = self.max_preview_size / height variable=current_frame
img = img.resize( )
( preview_frame_slider.pack(fill='both', side='left', expand='True')
int(width * aspect_ratio),
int(height * aspect_ratio) test_button = tk.Button(buttons_frame, text="Test", bg="#f1c40f", relief="flat", width=15, borderwidth=0, highlightthickness=0)
), test_button.pack(side='right', fill='y')
Image.ANTIALIAS return preview_window
)
photo_img = ImageTk.PhotoImage(img)
self.img_label.configure(image=photo_img) def show_preview():
self.img_label.image = photo_img preview.deiconify()
preview_visible.set(True)
def hide_preview():
preview.withdraw()
preview_visible.set(False)
def set_preview_handler(test_handler):
test_button.config(command = test_handler)
def init_slider(frames_count, change_handler):
preview_frame_slider.configure(to=frames_count, command=lambda value: change_handler(preview_frame_slider.get()))
preview_frame_slider.set(0)
def update_preview(frame):
img = Image.fromarray(frame)
width, height = img.size
aspect_ratio = 1
if width > height:
aspect_ratio = max_preview_size / width
else:
aspect_ratio = max_preview_size / height
img = img.resize(
(
int(width * aspect_ratio),
int(height * aspect_ratio)
),
Image.ANTIALIAS
)
photo_img = ImageTk.PhotoImage(img)
preview_image_frame.configure(image=photo_img)
preview_image_frame.image = photo_img
def select_face(select_face_handler: Callable[[str], None]): def select_face(select_face_handler: Callable[[str], None]):
@ -101,15 +96,17 @@ def select_face(select_face_handler: Callable[[str], None]):
def update_slider_handler(get_video_frame, video_path): def update_slider_handler(get_video_frame, video_path):
return lambda frame_number: preview.update(get_video_frame(video_path, frame_number)) return lambda frame_number: update_preview(get_video_frame(video_path, frame_number))
def test_preview(create_test_preview): def test_preview(create_test_preview):
frame = create_test_preview(preview.current_frame.get()) frame = create_test_preview(preview_frame_slider.get())
preview.update(frame) update_preview(frame)
def update_slider(get_video_frame, create_test_preview, video_path, frames_amount): def update_slider(get_video_frame, create_test_preview, video_path, frames_amount):
preview.init_slider(frames_amount, update_slider_handler(get_video_frame, video_path)) init_slider(frames_amount, update_slider_handler(get_video_frame, video_path))
preview.set_preview_handler(lambda: preview_thread(test_preview(create_test_preview))) set_preview_handler(lambda: preview_thread(lambda: test_preview(create_test_preview)))
def analyze_target(select_target_handler: Callable[[str], Tuple[int, Any]], target_path: tk.StringVar, frames_amount: tk.IntVar): def analyze_target(select_target_handler: Callable[[str], Tuple[int, Any]], target_path: tk.StringVar, frames_amount: tk.IntVar):
@ -118,7 +115,7 @@ def analyze_target(select_target_handler: Callable[[str], Tuple[int, Any]], targ
amount, frame = select_target_handler(path) amount, frame = select_target_handler(path)
frames_amount.set(amount) frames_amount.set(amount)
preview_target(frame) preview_target(frame)
preview.update(frame) update_preview(frame)
def select_target(select_target_handler: Callable[[str], Tuple[int, Any]], target_path: tk.StringVar, frames_amount: tk.IntVar): def select_target(select_target_handler: Callable[[str], Tuple[int, Any]], target_path: tk.StringVar, frames_amount: tk.IntVar):
@ -136,6 +133,7 @@ def save_file(save_file_handler: Callable[[str], None], target_path: str):
return save_file_handler(asksaveasfilename(initialfile=filename, defaultextension=ext, filetypes=[("All Files","*.*"),("Videos","*.mp4")])) return save_file_handler(asksaveasfilename(initialfile=filename, defaultextension=ext, filetypes=[("All Files","*.*"),("Videos","*.mp4")]))
return None return None
def toggle_all_faces(toggle_all_faces_handler: Callable[[int], None], variable: tk.IntVar): def toggle_all_faces(toggle_all_faces_handler: Callable[[int], None], variable: tk.IntVar):
if toggle_all_faces_handler: if toggle_all_faces_handler:
return lambda: toggle_all_faces_handler(variable.get()) return lambda: toggle_all_faces_handler(variable.get())
@ -202,13 +200,14 @@ def preview_thread(thread_function):
def open_preview_window(get_video_frame, target_path): def open_preview_window(get_video_frame, target_path):
if (preview.visible): if preview_visible.get():
preview.hide() hide_preview()
else: else:
preview.show() show_preview()
if target_path: if target_path:
frame = get_video_frame(target_path) frame = get_video_frame(target_path)
preview.update(frame) update_preview(frame)
def preview_face(path): def preview_face(path):
img = Image.open(path) img = Image.open(path)
@ -217,6 +216,7 @@ def preview_face(path):
face_label.configure(image=photo_img) face_label.configure(image=photo_img)
face_label.image = photo_img face_label.image = photo_img
def preview_target(frame): def preview_target(frame):
img = Image.fromarray(frame) img = Image.fromarray(frame)
img = img.resize((180, 180), Image.ANTIALIAS) img = img.resize((180, 180), Image.ANTIALIAS)
@ -224,10 +224,12 @@ def preview_target(frame):
target_label.configure(image=photo_img) target_label.configure(image=photo_img)
target_label.image = photo_img target_label.image = photo_img
def update_status_label(value): def update_status_label(value):
status_label["text"] = value status_label["text"] = value
window.update() window.update()
def init( def init(
initial_values: dict, initial_values: dict,
select_face_handler: Callable[[str], None], select_face_handler: Callable[[str], None],
@ -240,7 +242,7 @@ def init(
get_video_frame: Callable[[str, int], None], get_video_frame: Callable[[str, int], None],
create_test_preview: Callable[[int], Any], create_test_preview: Callable[[int], Any],
): ):
global window, preview, face_label, target_label, status_label global window, preview, preview_visible, face_label, target_label, status_label
window = tk.Tk() window = tk.Tk()
window.geometry("600x700") window.geometry("600x700")
@ -248,11 +250,12 @@ def init(
window.configure(bg="#2d3436") window.configure(bg="#2d3436")
window.resizable(width=False, height=False) window.resizable(width=False, height=False)
preview_visible = tk.BooleanVar(window, False)
target_path = tk.StringVar() target_path = tk.StringVar()
frames_amount = tk.IntVar() frames_amount = tk.IntVar()
# Preview window # Preview window
preview = PreviewWindow(window) preview = create_preview(window)
# Contact information # Contact information
support_link = tk.Label(window, text="Donate to project <3", fg="#fd79a8", bg="#2d3436", cursor="hand2", font=("Arial", 8)) support_link = tk.Label(window, text="Donate to project <3", fg="#fd79a8", bg="#2d3436", cursor="hand2", font=("Arial", 8))
@ -298,7 +301,7 @@ def init(
frames_checkbox.place(x=60,y=450,width=240,height=31) frames_checkbox.place(x=60,y=450,width=240,height=31)
# Start button # Start button
start_button = create_button(window, "Start", lambda: [save_file(save_file_handler, target_path.get()), preview_thread(start)]) start_button = create_button(window, "Start", lambda: [save_file(save_file_handler, target_path.get()), preview_thread(lambda: start(update_preview))])
start_button.place(x=170,y=560,width=120,height=49) start_button.place(x=170,y=560,width=120,height=49)
# Preview button # Preview button