Update app.py

app.py

@@ -8,7 +8,15 @@ def generate(image, prompt, seed):
         result = subprocess.run(command, shell=True, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
         output_path = 'output'
         print("Output:", result.stdout)
-        return output_path
+        print output_path
+        # List all files and directories in the given directory
+        contents = os.listdir(directory_path)
+        
+        # Print the contents
+        for item in contents:
+            print(item)
+        
+        return "Done"
     except subprocess.CalledProcessError as e:
         print("Error:", e.stderr)
         return None
@@ -20,7 +28,8 @@ with gr.Blocks() as demo:
           textbox = gr.Textbox(show_label=False, value="a person facing the camera, making a hand gesture, indoor")
           seed = gr.Slider(minimum=0, maximum=1000000, value=643534)
           button = gr.Button()
-      output_image = gr.Image(show_label=False, type="filepath", interactive=False, height=512, width=512)
+      #output_image = gr.Image(show_label=False, type="filepath", interactive=False, height=512, width=512)
+      output_image = gr.Textbox()
     button.click(fn=generate, inputs=[image, textbox, seed], outputs=[output_image])
 
 demo.queue().launch(inline=False, share=True, debug=True)