Import Failed - No Matter What I Do? about comfyui-artgallery HOT 5 OPEN

def modify_js_file(file_path, new_content):
    try:
        with open(file_path, 'r', encoding='utf-8') as file:
            content = file.read()

        # Check if new content is already included
        if "image_upload_artist" not in content:
            insert_position = content.find('nodeData.input.required.upload = ["IMAGEUPLOAD"];')
            if insert_position != -1:
                insert_position += len('nodeData.input.required.upload = ["IMAGEUPLOAD"];')
                content = content[:insert_position] + new_content + content[insert_position:]

                # Backup the original file
                backup_path = file_path + '.backup'
                with open(backup_path, 'w', encoding='utf-8') as backup_file:
                    backup_file.write(content)

                # Write modified content back to file
                with open(file_path, 'w', encoding='utf-8') as file:
                    file.write(content)
                print(f"File '{file_path}' updated successfully.✅")
            else:
                print("Original code block not found.❌")
        else:
            print("File already contains the necessary modifications.✅")
    except Exception as e:
        print(f"An error occurred: {e}")

def modify_wedgets_js_file(file_path, new_content, new_content_2):
    try:
        with open(file_path, 'r', encoding='utf-8') as file:
            content = file.read()

        # Check if the file already contains the content to be added
        if "ARTISTS_IMAGEUPLOAD" not in content:
            # Find the position of the original code
            insert_position = content.find('return (display==="slider") ? "slider" : "number"')
            if insert_position != -1:
                # Insert new code after the original code
                insert_position += len('return (display==="slider") ? "slider" : "number"')
                content = content[:insert_position] + new_content + content[insert_position:]

            insert_position_2 = content.find('return { widget: uploadWidget };')
            if insert_position_2 != -1:
                # Insert new code after the original code
                insert_position_2 += len('return { widget: uploadWidget };')
                content = content[:insert_position_2] + new_content_2 + content[insert_position_2:]

                # Backup the original file
                backup_path = file_path + '.backup'
                with open(backup_path, 'w', encoding='utf-8') as backup_file:
                    backup_file.write(content)

                # Write back the file
                with open(file_path, 'w', encoding='utf-8') as file:
                    file.write(content)
                print(f"File '{file_path}' updated successfully.✅")
            else:
                print("Original code block not found.❌")
        else:
            print("File already contains the necessary modifications.✅")
    except Exception as e:
        print(f"An error occurred: {e}")

def modify_py_file(file_path, new_content, search_line, function_content, search_function):
    try:
        with open(file_path, 'r', encoding='utf-8') as file:
            lines = file.readlines()

        # Prepare key lines of new content and function content for comparison
        new_content_key_line = new_content.strip().split('\n')[0]
        function_content_key_line = function_content.strip().split('\n')[0]

        # Check if new content already exists
        if new_content_key_line not in "".join(lines):
            for index, line in enumerate(lines):
                if search_line in line:
                    lines.insert(index + 1, new_content)
                    break

        # Check if function modification already exists
        if function_content_key_line not in "".join(lines):
            function_start = False
            for index, line in enumerate(lines):
                if search_function in line:
                    function_start = True
                if function_start and "return None" in line:
                    lines.insert(index, function_content)
                    break

        # Backup the original file
        backup_path = file_path + '.backup'
        with open(backup_path, 'w', encoding='utf-8') as backup_file:
            backup_file.writelines(lines)

        # Write back the modified content
        with open(file_path, 'w', encoding='utf-8') as file:
            file.writelines(lines)
        print(f"File '{file_path}' updated successfully.✅")

    except Exception as e:
        print(f"An error occurred: {e}")