Update app.py

app.py

@@ -46,10 +46,10 @@ def run_app(user_input, enhance_prompt, action, state, progress=gr.Progress()):
         f.write(zip_bytes)
 
     progress(0.9, desc="Generating preview...")
-    preview_html = extract_react_preview(code_bundle["full_code"])
+    preview_html = extract_react_preview(code_bundle.get("full_code", ""))
 
     progress(1.0, desc="Done!")
-    return code_bundle["full_code"], "generated_app.zip", preview_html, code_bundle.get("files", {}), state
+    return code_bundle.get("full_code", ""), "generated_app.zip", preview_html, code_bundle.get("files", {}), state
 
 with gr.Blocks(title="Auto App Builder") as iface:
     gr.Markdown("# ⚡ Auto App Builder (Gemini Powered)\n_Powered by Google Gemini API_")
@@ -77,27 +77,35 @@ with gr.Blocks(title="Auto App Builder") as iface:
     def update_file_tree(code, zip_name, preview_path, files, state):
         if isinstance(files, dict) and len(files) > 0:
             keys = list(files.keys())
-            return gr.update(choices=keys, value=keys[0]), files[keys[0]]
+            default_key = keys[0] if keys else None
+            return gr.update(choices=keys, value=default_key), files.get(default_key, "")
         else:
             return gr.update(choices=[], value=None), ""
 
     btn.click(
         fn=run_app,
         inputs=[inp, enh, gr.State(""), state],
-        outputs=[code_out, zip_file, preview, file_selector, state],
-        postprocess=update_file_tree
+        outputs=[code_out, zip_file, preview, file_selector, state]
+    ).then(
+        fn=update_file_tree,
+        inputs=[code_out, zip_file, preview, state["bundle"]["files"] if state else {}, state],
+        outputs=[file_selector, file_viewer]
     )
+
     btn_enh.click(
         fn=run_app,
         inputs=[inp, enh, gr.State("Enhance"), state],
-        outputs=[code_out, zip_file, preview, file_selector, state],
-        postprocess=update_file_tree
+        outputs=[code_out, zip_file, preview, file_selector, state]
+    ).then(
+        fn=update_file_tree,
+        inputs=[code_out, zip_file, preview, state["bundle"]["files"] if state else {}, state],
+        outputs=[file_selector, file_viewer]
     )
 
     file_selector.change(
-        lambda name, state: state["bundle"]["files"].get(name, ""),
+        lambda name, state: state["bundle"]["files"].get(name, "") if state and "bundle" in state else "",
         inputs=[file_selector, state],
         outputs=file_viewer
     )
 
-iface.launch()
+iface.launch()