Update app.py

app.py

@@ -92,12 +92,14 @@ with gr.Blocks(title="Auto App Builder") as iface:
     btn.click(
         fn=run_app,
         inputs=[inp, enh, gr.State(""), state],
-        outputs=[code_out, zip_file, preview, file_selector, state]
+        outputs=[code_out, zip_file, preview, file_selector, state],
+        postprocess=update_file_tree
     )
     btn_enh.click(
         fn=run_app,
         inputs=[inp, enh, gr.State("Enhance"), state],
-        outputs=[code_out, zip_file, preview, file_selector, state]
+        outputs=[code_out, zip_file, preview, file_selector, state],
+        postprocess=update_file_tree
     )
 
     file_selector.change(