import gradio as gr
import tempfile
from PIL import Image
import uuid 
import os

# Assuming 'my_directory' is a directory in the current working directory
#directory_name = 'temp_dir'
#absolute_path1 = os.path.abspath(directory_name)
#print(f"The absolute path of '{directory_name}' is: {absolute_path1}")
#/home/user/app/temp_dir

# Access the environment variable
os.environ['GRADIO_TEMP_DIR'] = '/data'
gradio_temp_dir = os.environ['GRADIO_TEMP_DIR'] 
print(f'GRADIO_TEMP_DIR is = {gradio_temp_dir}')
absolute_path1 = os.path.abspath(gradio_temp_dir)
print(f"The absolute path of gradio_temp_dir is = {absolute_path1}")

#os.makedirs('/data', exist_ok=True)

#current_directory = os.getcwd()
#print(f'current_directory is = {current_directory}')
#new_directory_name = 'temp_dir' #os.environ['GRADIO_TEMP_DIR']

# Constructing the path to the new directory
#new_directory_path = os.path.join(current_directory, new_directory_name)
#print(f"The path to the new directory is: {new_directory_path}")

#os.environ['GRADIO_TEMP_DIR'] = new_directory_path

######

# Get the path of the system's temporary directory
temp_directory = tempfile.gettempdir()
print(f"System's temporary directory is: {temp_directory}")


def create():
    """
    Create a blank image with the specified dimensions, color, and filename.
    """
    color='blue'
    width=512
    height=512
    # Create a temporary file
    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
    print(f"Temporary file created at: {temp_file.name}")
    #filename = f'{temp_file.name}.png'
    # Create a new image with the given mode and size
    image = Image.new("RGB", (width, height), color)
    # Save the image to disk
    image.save(temp_file, format='PNG')


    # Specify the directory you want to search in, e.g., the current directory
    directory = 'temp_dir'
    # List all entries in the directory
    entries = os.listdir(directory)
    print(f"listdir of temp_dir directory = {entries}")
    temp_directory_file_list = os.listdir(temp_directory)
    print(f'temp_directory_file_list is = {temp_directory_file_list}')
    return temp_file.name, temp_directory_file_list


with gr.Blocks(delete_cache=(5,10),
               theme=gr.themes.Base(radius_size="none")) as demo:
    with gr.Row():
        inp = gr.Textbox(placeholder="What is your name?")
        out = gr.Image(type='filepath')
    btn = gr.Button("GO!")
    btn.click(create,inputs=[],outputs=[out, inp])

#filepath11 = demo.move_resource_to_block_cache('/content/testing/mp333.png')
#print(f'filepath11 ^^^^^^= {filepath11}')
demo.launch(debug=False)