import gradio as gr
import os
import modules.config
import modules.html
import modules.meta_parser

from tkinter import Tk, filedialog


def process_directories(directory_paths):
    if not directory_paths:
        return "No directories selected."

    results = []
    for directory in directory_paths:
        # List files in the directory
        files = os.listdir(directory)
        results.append(f"Contents of {directory}:\n" + "\n".join(files))

    return "\n\n".join(results)


def update_visibility(x):
    # Add more updates for other components
    return [gr.update(visible=x), gr.update(visible=x)]


def list_to_string(filenames):
    # Join the filenames list into a comma-separated string
    file_list = ', '.join(filenames)
    return file_list


def on_browse(data_type):
    root = Tk()
    root.attributes("-topmost", True)
    root.withdraw()
    if data_type == "Files":
        filenames = filedialog.askopenfilenames()
        if len(filenames) > 0:
            root.destroy()
            file_list = list_to_string(filenames)
            return file_list
        else:
            filename = "Files not seleceted"
            root.destroy()
            return None

    elif data_type == "Folder":
        filename = filedialog.askdirectory()
        if filename:
            if os.path.isdir(filename):
                root.destroy()
                return str(filename)
            else:
                root.destroy()
                return str(filename)
        else:
            filename = "Folder not seleceted"
            root.destroy()
            return None


def on_file_change(files, data_type):
    if files and data_type == "Files":
        return gr.update(visible=True), gr.update(), gr.update(value=True)

    # If no files are selected, hide file explorer and clear input_path
    if not files and data_type == "Files":
        return gr.update(visible=False), gr.update(value=""), gr.update(value=False)

    if data_type == "Folder":
        return gr.update(visible=False), gr.update(), gr.update(value=True)

    return gr.update(visible=False), gr.update(), gr.update(value=False)


def on_input_change(input, file_explorer):
    if input:
        # Verify with normalised version of path
        input_path = os.path.normpath(os.path.realpath(input))

        if os.path.isdir(os.path.realpath(input_path)):
            # Return an empty list if input_path is a directory
            return None, gr.update(visible=True), gr.update(value=True)
    else:
        # Return an empty list if input_path is empty
        return None, gr.update(visible=False), gr.update(value=False)

    # Initialize a dictionary to track unique file names and their paths
    unique_file_paths = {}

    # Process the input_path string
    if input_path:
        # Clean up the input path string and split it into a list of file paths
        file_paths_list = input_path.strip("()").replace("'", "").split(", ")
        # Extract file names and ensure uniqueness
        for path in file_paths_list:
            file_name = os.path.basename(path)
            unique_file_paths[file_name] = path

    # Process file_explorer items if provided
    if file_explorer:
        # Extract 'orig_name' from each file_explorer object and ensure uniqueness
        for item in file_explorer:
            sanitized_path = item.orig_name
            file_name = os.path.basename(sanitized_path)
            # Store the path, replacing any existing path with the same file name
            unique_file_paths[file_name] = sanitized_path

    # Convert the dictionary values back to a list of unique file paths
    if len(unique_file_paths.values()) > 0:
        return list(unique_file_paths.values()), gr.update(visible=False), gr.update(value=True)
    else:
        return None, gr.update(visible=False), gr.update(value=False)


def on_click_clear():
    return None, None, gr.update(visible=False), gr.update(visible=False)

# Function to set prompts based on the selected type


def update_prompts(selected_type):
    # Ensure selected_type is a valid key and exists in the dictionary
    if selected_type in modules.config.default_enhance_prompts:
        positive_prompt = modules.config.default_enhance_prompts[selected_type]['positive']
        negative_prompt = modules.config.default_enhance_prompts[selected_type]['negative']
        return positive_prompt, negative_prompt
    else:
        # Returning default or empty values
        return "Default positive prompt", "Default negative prompt"


def on_selection_change(selected_type):
    # Get prompts based on selected_type
    positive_prompt, negative_prompt = update_prompts(selected_type[0])

    # Return the prompts
    return positive_prompt, negative_prompt