import os import time import gradio as gr import roop.globals import roop.metadata import roop.utilities as util import ui.globals as uii from ui.tabs.faceswap_tab import faceswap_tab from ui.tabs.livecam_tab import livecam_tab from ui.tabs.facemgr_tab import facemgr_tab from ui.tabs.extras_tab import extras_tab from ui.tabs.settings_tab import settings_tab roop.globals.keep_fps = None roop.globals.keep_frames = None roop.globals.skip_audio = None roop.globals.use_batch = None def prepare_environment(): roop.globals.output_path = os.path.abspath(os.path.join(os.getcwd(), "output")) os.makedirs(roop.globals.output_path, exist_ok=True) if not roop.globals.CFG.use_os_temp_folder: os.environ["TEMP"] = os.environ["TMP"] = os.path.abspath(os.path.join(os.getcwd(), "temp")) os.makedirs(os.environ["TEMP"], exist_ok=True) os.environ["GRADIO_TEMP_DIR"] = os.environ["TEMP"] def run(): from roop.core import decode_execution_providers, set_display_ui prepare_environment() set_display_ui(show_msg) roop.globals.execution_providers = decode_execution_providers([roop.globals.CFG.provider]) print(f'Using provider {roop.globals.execution_providers} - Device:{util.get_device()}') run_server = True uii.ui_restart_server = False mycss = """ span {color: var(--block-info-text-color)} #fixedheight { max-height: 238.4px; overflow-y: auto !important; } """ uii.ui_live_cam_active = roop.globals.CFG.live_cam_start_active while run_server: server_name = roop.globals.CFG.server_name if server_name is None or len(server_name) < 1: server_name = None server_port = roop.globals.CFG.server_port if server_port <= 0: server_port = None ssl_verify = False if server_name == '0.0.0.0' else True with gr.Blocks(title=f'{roop.metadata.name} {roop.metadata.version}', theme=roop.globals.CFG.selected_theme, css=mycss) as ui: with gr.Row(variant='compact'): gr.Markdown(f"### [{roop.metadata.name} {roop.metadata.version}](https://github.com/C0untFloyd/roop-unleashed)") gr.HTML(util.create_version_html(), elem_id="versions") faceswap_tab() livecam_tab() facemgr_tab() extras_tab() settings_tab() uii.ui_restart_server = False try: ui.queue().launch(inbrowser=True, server_name=server_name, server_port=server_port, share=roop.globals.CFG.server_share, ssl_verify=ssl_verify, prevent_thread_lock=True, show_error=True) except Exception as e: print(f'Exception {e} when launching Gradio Server!') uii.ui_restart_server = True run_server = False try: while uii.ui_restart_server == False: time.sleep(1.0) except (KeyboardInterrupt, OSError): print("Keyboard interruption in main thread... closing server.") run_server = False ui.close() def show_msg(msg: str): gr.Info(msg)