Update app.py

app.py

@@ -3,6 +3,38 @@ import tempfile
 from PIL import Image
 import uuid 
 import os
+import shutil
+from pathlib import Path
+import random
+
+
+def create_local():
+    length = 100
+    text_string = ''.join(random.choice(string.ascii_letters + string.digits) for _ in range(length))
+    #rand_num = random.randint(1, 10000)
+    file_name = os.path.join('temp_dir', f'random_file_1.txt')
+    with open(file_name, 'w') as f:
+        for _ in range(10):
+            f.write(text_string + '\n')
+    
+
+def move():
+    # Get the path of the system's temporary directory
+    temp_directory = tempfile.gettempdir()
+    
+    # Specify the path of the file you want to move
+    file_path = '/temp_dir/random_file1.txt'
+    
+    # Construct the new path in the temporary directory with the same filename
+    new_path = os.path.join(temp_directory, os.path.basename(file_path))
+    
+    print(f'new_path - {new_path}')
+    
+    # Move the file
+    shutil.move(file_path, new_path)
+    
+    print(f"Moved the file to: {new_path}")
+
 
 # Assuming 'my_directory' is a directory in the current working directory
 #directory_name = 'temp_dir'
@@ -17,19 +49,6 @@ 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()
@@ -57,7 +76,7 @@ def create():
     directory = 'temp_dir'
     # List all entries in the directory
     entries = os.listdir(directory)
-    print(f"listdir of temp_dir directory = {entries}")
+    #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
@@ -69,9 +88,15 @@ with gr.Blocks(delete_cache=(5,10),
         inp = gr.Textbox(placeholder="What is your name?")
         out = gr.Image(type='filepath')
         gr.FileExplorer(root="/data", label="Persistent storage")
-    btn = gr.Button("GO!")
+    with gr.Row():
+        with gr.Column():
+            btn = gr.Button("Generate temp files!")
+            btn1 = gr.Button("Generate local files!")
+        gr.FileExplorer(label="local storage")
+        btn2 = gr.Button("Move files to temp dir!")
+        
     btn.click(create,inputs=[],outputs=[out, inp])
+    btn1.click(create_local,inputs=[],outputs=[])
+    btn2.click(move,inputs=[],outputs=[])
 
-#filepath11 = demo.move_resource_to_block_cache('/content/testing/mp333.png')
-#print(f'filepath11 ^^^^^^= {filepath11}')
 demo.launch(debug=False)