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

# Access the environment variable
print(os.environ['GRADIO_TEMP_DIR'])
 
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')
    return temp_file.name


with gr.Blocks(delete_cache=(2,2),
               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)

demo.launch(debug=False)