ysharma's picture
ysharma HF Staff
Create app.py
9a9513f verified
raw
history blame
1.02 kB
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)