Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions shared/cli_args.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,6 +61,7 @@ def parse_wgp_args(family_handlers: Sequence[str], config_filename: str, default
add("--server-name", type=str, default="", help="Server name")
add("--gpu", type=str, default="", help="Default GPU Device")
add("--open-browser", action="store_true", help="open browser")
add("--run-as-chrome-app", type=str, default="", help="open as app using provided chrome exe location")
add("--t2v", action="store_true", help="text to video mode")
add("--i2v", action="store_true", help="image to video mode")
add("--t2v-14B", action="store_true", help="text to video mode 14B model")
Expand Down
41 changes: 34 additions & 7 deletions wgp.py
Original file line number Diff line number Diff line change
Expand Up @@ -1795,6 +1795,10 @@ def clear_queue_action(state):
def quit_application():
print("Save and Quit requested...")
clear_startup_lock()
global chrome_proc
if chrome_proc:
chrome_proc.kill()
chrome_proc = None
autosave_queue()
import signal
os.kill(os.getpid(), signal.SIGINT)
Expand Down Expand Up @@ -12152,14 +12156,37 @@ def clear_startup_lock():
if len(server_name) == 0:
server_name = os.getenv("SERVER_NAME", "localhost")
demo = create_ui()
clear_startup_lock()
clear_startup_lock()
if server_name.startswith("http"):
url = server_name
else:
url = "http://" + server_name
url= url + ":" + str(server_port)
if len(args.run_as_chrome_app) >0:
def run_chrome(command, callback):
import subprocess
def cleanup():
global chrome_proc
print("Cleaning up Chrome...")
chrome_proc.kill()
def target():
global chrome_proc
chrome_proc = subprocess.Popen(command)
chrome_proc.communicate()
callback()
atexit.register(cleanup)
thread = threading.Thread(target=target,daemon=True)
thread.start()
def finish():
print("App has been closed, shutting down server...")
quit_application()

chrome_exe=args.run_as_chrome_app
profile=os.path.join( Path.cwd(), "appModeProfile")
run_chrome([chrome_exe,"--user-data-dir="+profile, "--app="+url], finish)
if args.open_browser:
import webbrowser
if server_name.startswith("http"):
url = server_name
else:
url = "http://" + server_name
webbrowser.open(url + ":" + str(server_port), new = 0, autoraise = True)
import webbrowser
webbrowser.open(url, new = 0, autoraise = True)
demo.launch(
favicon_path="favicon.png",
server_name=server_name,
Expand Down