File size: 2,398 Bytes
c336648 | 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 | import sys
import os
import shutil
import gradio
# Get the paths
module_path = os.path.join(sys.path[1],"scripts")
detect_module_path = os.path.join(sys.path[0], "detect_extension.py") # assuming the script is in the current directory
# Form the destination path
destination_path = os.path.join(module_path, "detect_extension.py")
# Check if the source file exists and is a file (not a directory),
# and if the file doesn't already exist at the destination,
# and if the module path is a valid directory
if os.path.isfile(detect_module_path) and not os.path.isfile(destination_path) and os.path.isdir(module_path):
# Copy the file
shutil.copyfile(detect_module_path, destination_path)
# Check the version of the gradio, if it is less than 3.28.1 then cancel the installation
is_right_version = False
# Getting the Gradio version
current_version = gradio.__version__
# Converting versions into tuples of numbers for easy comparison
def version_to_tuple(version_string):
return tuple(map(int, version_string.split('.')))
minimum_required_version = "3.28.1"
second_required_version = "3.32.0"
# Checking the Gradio version
if version_to_tuple(current_version) < version_to_tuple(minimum_required_version):
print("\nPlease update webui to the latest version for the canvas-zoom extension to work properly, supported versions from 1.1 \n")
elif version_to_tuple(current_version) <= version_to_tuple(second_required_version):
source_dir_name = 'v1_1_v1_5_1'
is_right_version = True
else:
source_dir_name = ''
is_right_version = True
if is_right_version:
# Moving files with replacement
canvasZoomPath = sys.path[0]
gradioPath = os.path.dirname(gradio.__file__)
source_dir = os.path.join(canvasZoomPath,"dist", source_dir_name, 'templates', 'frontend')
if not os.path.exists(source_dir):
canvasZoomPath = os.path.dirname(os.path.realpath(__file__))
source_dir = os.path.join(canvasZoomPath,"dist", source_dir_name, 'templates', 'frontend')
destination_dir = os.path.join(gradioPath, 'templates', 'frontend')
# Deleting the existing "templates" folder in the gradio folder, if it exists
if os.path.exists(destination_dir):
shutil.rmtree(destination_dir)
# Moving the "templates" folder from the canvasZoom folder to the gradio folder
shutil.copytree(source_dir, destination_dir)
|