mirror of
https://github.com/nod-ai/AMD-SHARK-Studio.git
synced 2026-02-19 11:56:43 -05:00
6285430d8a
* Moves the imports of winreg and Tk, into the functions that use them, with winreg behind a guard clause. This should hopefully mean that if you're not on Window or not using `ui=app` we won't trip over either of these due to them not being there.
106 lines
3.3 KiB
Python
106 lines
3.3 KiB
Python
import os
|
|
import sys
|
|
import webview
|
|
import webview.util
|
|
import socket
|
|
|
|
from contextlib import closing
|
|
from multiprocessing import Process
|
|
|
|
from apps.stable_diffusion.src import args
|
|
|
|
|
|
def webview2_installed():
|
|
if sys.platform != "win32":
|
|
return False
|
|
|
|
# On windows we want to ensure we have MS webview2 available so we don't fall back
|
|
# to MSHTML (aka ye olde Internet Explorer) which is deprecated by pywebview, and
|
|
# apparently causes SHARK not to load in properly.
|
|
|
|
# Checking these registry entries is how Microsoft says to detect a webview2 installation:
|
|
# https://learn.microsoft.com/en-us/microsoft-edge/webview2/concepts/distribution
|
|
import winreg
|
|
|
|
path = r"SOFTWARE\WOW6432Node\Microsoft\EdgeUpdate\Clients\{F3017226-FE2A-4295-8BDF-00C3A9A7E4C5}"
|
|
|
|
# only way can find if a registry entry even exists is to try and open it
|
|
try:
|
|
# check for an all user install
|
|
with winreg.OpenKey(
|
|
winreg.HKEY_LOCAL_MACHINE,
|
|
path,
|
|
0,
|
|
winreg.KEY_QUERY_VALUE | winreg.KEY_WOW64_64KEY,
|
|
) as registry_key:
|
|
value, type = winreg.QueryValueEx(registry_key, "pv")
|
|
|
|
# if it didn't exist, we want to continue on...
|
|
except WindowsError:
|
|
try:
|
|
# ...to check for a current user install
|
|
with winreg.OpenKey(
|
|
winreg.HKEY_CURRENT_USER,
|
|
path,
|
|
0,
|
|
winreg.KEY_QUERY_VALUE | winreg.KEY_WOW64_64KEY,
|
|
) as registry_key:
|
|
value, type = winreg.QueryValueEx(registry_key, "pv")
|
|
except WindowsError:
|
|
value = None
|
|
finally:
|
|
return (value is not None) and value != "" and value != "0.0.0.0"
|
|
|
|
|
|
def window(address):
|
|
from tkinter import Tk
|
|
|
|
window = Tk()
|
|
|
|
# get screen width and height of display and make it more reasonably
|
|
# sized as we aren't making it full-screen or maximized
|
|
width = int(window.winfo_screenwidth() * 0.81)
|
|
height = int(window.winfo_screenheight() * 0.91)
|
|
webview.create_window(
|
|
"SHARK AI Studio",
|
|
url=address,
|
|
width=width,
|
|
height=height,
|
|
text_select=True,
|
|
)
|
|
webview.start(private_mode=False, storage_path=os.getcwd())
|
|
|
|
|
|
def usable_port():
|
|
# Make sure we can actually use the port given in args.server_port. If
|
|
# not ask the OS for a port and return that as our port to use.
|
|
|
|
port = args.server_port
|
|
|
|
with closing(socket.socket(socket.AF_INET, socket.SOCK_STREAM)) as sock:
|
|
try:
|
|
sock.bind(("0.0.0.0", port))
|
|
except OSError:
|
|
with closing(
|
|
socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
) as sock:
|
|
sock.bind(("0.0.0.0", 0))
|
|
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
|
return sock.getsockname()[1]
|
|
|
|
return port
|
|
|
|
|
|
def launch(port):
|
|
# setup to launch as an app if app mode has been requested and we're able
|
|
# to do it, answering whether we succeeded.
|
|
if args.ui == "app" and (sys.platform != "win32" or webview2_installed()):
|
|
try:
|
|
t = Process(target=window, args=[f"http://localhost:{port}"])
|
|
t.start()
|
|
return True
|
|
except webview.util.WebViewException:
|
|
return False
|
|
else:
|
|
return False
|