Update app.py

app.py

@@ -9,6 +9,19 @@ import random
 import string
 import glob
 
+
+# Set 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}")
+
+# Get the path of the system's temporary directory
+temp_directory = tempfile.gettempdir()
+print(f"System's temporary directory is: {temp_directory}")
+
 def create_local():
     length = 100
     text_string = ''.join(random.choice(string.ascii_letters + string.digits) for _ in range(length))
@@ -23,7 +36,7 @@ def movefiles():
     # Get the path of the system's temporary directory
     temp_directory = tempfile.gettempdir()
     
-    # Specify the path of the file you want to move
+    # Specify the path of the file to be moved
     directory_name = 'temp_dir'
     absolute_path = os.path.abspath(directory_name)
     file_path = 'random_file_1.txt'
@@ -40,25 +53,6 @@ def movefiles():
     print(f"Moved the file to: {new_path}")
 
 
-# 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}")
-
-
-# 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.
@@ -69,18 +63,12 @@ def create():
     # 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