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

Implemented separate path memory for each file dialog to enhance user navigation

Browse Source
This commit is contained in:
skrusiogit 2023-06-17 20:13:21 +02:00 committed by GitHub
parent 272be1e6d8
commit 684d3a5c19
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 18 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
roop/ui.py
View File

@ -1,3 +1,4 @@
import os
import tkinter as tk
from typing import Any, Callable, Tuple
from PIL import Image, ImageTk
@ -10,6 +11,10 @@ from roop.utils import is_img
max_preview_size = 800
last_target_path = ""
last_output_path = ""
last_face_path = ""
def create_preview(parent):
global preview_image_frame, preview_frame_slider, test_button
@ -88,8 +93,10 @@ def update_preview(frame):
def select_face(select_face_handler: Callable[[str], None]):
global last_face_path
if select_face_handler:
path = filedialog.askopenfilename(title="Select a face")
path = filedialog.askopenfilename(initialdir=last_face_path, title="Select a face")
last_face_path = os.path.dirname(path) # Aktualisieren Sie den Pfad für den nächsten Dialog
preview_face(path)
return select_face_handler(path)
return None
@ -109,8 +116,10 @@ def update_slider(get_video_frame, create_test_preview, video_path, frames_amoun
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):
path = filedialog.askopenfilename(title="Select a target")
def analyze_target(select_target_handler: Callable[[str], Tuple[int, Any]], target_path: tk.StringVar, frames_amount: tk.IntVar):
global last_target_path
path = filedialog.askopenfilename(initialdir=last_target_path, title="Select a target")
last_target_path = os.path.dirname(path) # Aktualisieren Sie den Pfad für den nächsten Dialog
target_path.set(path)
amount, frame = select_target_handler(path)
frames_amount.set(amount)
@ -124,16 +133,20 @@ def select_target(select_target_handler: Callable[[str], Tuple[int, Any]], targe
def save_file(save_file_handler: Callable[[str], None], target_path: str):
global last_output_path
filename, ext = 'output.mp4', '.mp4'
if is_img(target_path):
filename, ext = 'output.png', '.png'
if save_file_handler:
return save_file_handler(asksaveasfilename(initialfile=filename, defaultextension=ext, filetypes=[("All Files","*.*"),("Videos","*.mp4")]))
output_path = asksaveasfilename(initialdir=last_output_path, initialfile=filename, defaultextension=ext, filetypes=[("All Files","*.*"),("Videos","*.mp4")])
last_output_path = os.path.dirname(output_path) # Aktualisieren Sie den Pfad für den nächsten Dialog
return save_file_handler(output_path)
return None
def toggle_all_faces(toggle_all_faces_handler: Callable[[int], None], variable: tk.IntVar):
if toggle_all_faces_handler:
return lambda: toggle_all_faces_handler(variable.get())
@ -312,4 +325,4 @@ def init(
status_label = tk.Label(window, width=580, justify="center", text="Status: waiting for input...", fg="#2ecc71", bg="#2d3436")
status_label.place(x=10,y=640,width=580,height=30)
return window
return window