Fix for working with SDNext

Gradio-v4-updates
Danil Boldyrev 2024-02-01 10:29:40 +03:00
parent 26450143c5
commit 55cbff67c0
1 changed files with 25 additions and 19 deletions

View File

@ -16,15 +16,22 @@ script_path = os.path.realpath(__file__)
requirements_path = os.path.join(sys.path[2], "requirements.txt")
config_path = os.path.join(sys.path[2], "config.json")
# Define vars
gradio_version = None
should_script_work = False
# Read the version of 'gradio' from 'requirements.txt'
if os.path.isfile(requirements_path):
with open(requirements_path, "r") as file:
lines = file.readlines()
gradio_line = next((line for line in lines if line.startswith("gradio")), None)
gradio_version = gradio_line.split("==")[1].strip() if gradio_line else None
should_script_work = True
# Check if config.json exists
if os.path.isfile(config_path):
# In case we have not found anything, we skip this script.
if should_script_work:
# Check if config.json exists
if os.path.isfile(config_path):
with open(config_path, 'r') as config_file:
config = json.load(config_file)
# Check if "canvas-zoom" is in the list of disabled_extensions
@ -37,10 +44,9 @@ if os.path.isfile(config_path):
launch.run(f'"{python}" -m pip install --force-reinstall --no-deps gradio=={gradio_version}', desc=f"Uninstalling modified gradio for canvas-zoom", errdesc=f"Couldn't uninstall canvas-zoom", live=False)
os.remove(script_path)
# Check if the folder exists
if not os.path.exists(canvasZoomPath) and gradio_version is not None:
# Check if the folder exists
if not os.path.exists(canvasZoomPath) and gradio_version is not None:
# If the folder does not exist and we found 'gradio' version, uninstall 'gradio' and install it again
launch.run(f'"{python}" -m pip install --force-reinstall --no-deps gradio=={gradio_version}', desc=f"Uninstalling modified gradio for canvas-zoom", errdesc=f"Couldn't uninstall canvas-zoom", live=False)
# Deleting the file
os.remove(script_path)