Update app.py

app.py

@@ -24,6 +24,8 @@ print(f"The path to the new directory is: {new_directory_path}")
 
 os.environ['GRADIO_TEMP_DIR'] = new_directory_path
 
+######
+
 def create():
     """
     Create a blank image with the specified dimensions, color, and filename.
@@ -39,7 +41,21 @@ def create():
     image = Image.new("RGB", (width, height), color)
     # Save the image to disk
     image.save(temp_file, format='PNG')
-    return temp_file.name
+
+
+    # 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) 
+    # Filter out hidden directories
+    hidden_directories = [d for d in entries if os.path.isdir(os.path.join(directory, d)) and d.startswith('.')]
+    print("------Hidden directories:------")
+    for d in hidden_directories:
+        print(d)
+        hidden_directories_listing = os.listdir(d)
+        print(os.listdir(d))
+    print("-------------------------------")
+    return temp_file.name, hidden_directories_listing
 
 
 with gr.Blocks(delete_cache=(2,2),
@@ -48,6 +64,6 @@ with gr.Blocks(delete_cache=(2,2),
         inp = gr.Textbox(placeholder="What is your name?")
         out = gr.Image(type='filepath')
     btn = gr.Button("GO!")
-    btn.click(create,inputs=[],outputs=out)
+    btn.click(create,inputs=[],outputs=[out, inp])
 
 demo.launch(debug=False)