Spaces:
Build error
Build error
File size: 3,718 Bytes
da50507 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 | 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"]
os.environ["GRADIO_ANALYTICS_ENABLED"] = "0"
def run():
from roop.core import decode_execution_providers, set_display_ui
prepare_environment()
set_display_ui(show_msg)
if roop.globals.CFG.provider == "cuda" and util.has_cuda_device() == False:
roop.globals.CFG.provider = "cpu"
roop.globals.execution_providers = decode_execution_providers(
[roop.globals.CFG.provider]
)
gputype = util.get_device()
if gputype == "cuda":
util.print_cuda_info()
print(f"Using provider {roop.globals.execution_providers} - Device:{gputype}")
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;
}
.image-container.svelte-1l6wqyv {height: 100%}
"""
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,
delete_cache=(60, 86400),
) as ui:
with gr.Row(variant="compact"):
gr.Markdown(
f"### [{roop.metadata.name} {roop.metadata.version}](https://gitlab.com/roop-unleashed/roop-unleashed)"
)
gr.HTML(util.create_version_html(), elem_id="versions")
faceswap_tab()
livecam_tab()
facemgr_tab()
extras_tab()
settings_tab()
launch_browser = roop.globals.CFG.launch_browser
uii.ui_restart_server = False
try:
ui.queue().launch(
inbrowser=launch_browser,
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)
|