Spaces:
Sleeping
Sleeping
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) |