File size: 1,134 Bytes
25a41e9
9f34051
 
c7a2e27
 
 
 
 
 
25a41e9
9f34051
 
 
 
25a41e9
9f34051
 
 
795c1e9
9f34051
 
 
 
 
c7a2e27
 
 
 
 
 
 
 
 
25a41e9
9f34051
 
 
c7a2e27
 
 
9f34051
c7a2e27
9f34051
25a41e9
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
import gradio as gr
import tempfile
from PIL import Image
import os

# Get the path to the temporary directory
temp_dir = tempfile.gettempdir()
print(f"Temporary directory : {temp_dir}")
    

def create():
    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}")
    
    # 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')

    # List all files in the temporary directory
    print('_________________')
    temp_files = os.listdir(temp_dir)
    print(f"Temporary files and directories in '{temp_dir}':")
    for file in temp_files:
        print(file)
    print('_________________')
    return temp_file.name, temp_files

# usage:

with gr.Blocks(delete_cache=(2,4)) as demo:
    with gr.Row():
        tb = gr.Textbox()
        out = gr.Image(type='filepath')
    btn = gr.Button("GO!")
    btn.click(create,inputs=[],outputs=[out, tb])
    
demo.launch()