Update app.py

app.py

@@ -934,16 +934,74 @@ with gr.Blocks(title="Axon Pro — Python AI IDE") as demo:
     find_all_btn.click(do_find_all, find_query, find_results)
     replace_btn.click(do_replace, [editor, find_query, replace_query, find_case],
                       [editor, find_results, structure_view])
+# ... [Keep your existing FIND tab code] ... 
 
-    # Display
-    capture_btn.click(capture_display, None, display_image)
-    auto_capture.change(toggle_auto, auto_capture, auto_timer)
-    auto_timer.tick(auto_capture_tick, None, display_image)
+                # ═══ MISSING TAB: GUI DISPLAY ═══
+                with gr.Tab("🖥 DISPLAY", id="display-tab"):
+                    display_image = gr.Image(label="Virtual Display Output", interactive=False, type="filepath")
+                    with gr.Row():
+                        capture_btn = gr.Button("📸 CAPTURE SCREEN", variant="primary")
+                        auto_capture = gr.Checkbox(label="Auto-refresh (2s)")
+                    # Hidden timer for auto-refresh
+                    auto_timer = gr.Timer(value=2.0, active=False)
+
+    # ═══════════════════════════════════════
+    # BRIDGE FUNCTIONS (These were missing)
+    # ═══════════════════════════════════════
+    
+    # Wrappers for Find/Replace
+    def do_find(code, query, case):
+        return finder.find_in_file(code, query, case)
+
+    def do_find_all(query):
+        return finder.find_in_all_files(fs.files, query)
+
+    def do_replace(code, find, replace, case):
+        new_code, msg = finder.replace_in_file(code, find, replace, case)
+        # We need to update the editor, the results box, AND the structure view
+        new_structure = analyzer.analyze(new_code)
+        return new_code, msg, new_structure
+
+    # Wrappers for Display
+    def capture_display():
+        path = vdisplay.capture()
+        if not path:
+            return None 
+        return path
+
+    def toggle_auto(checkbox_state):
+        return gr.Timer(active=checkbox_state)
+
+    def auto_capture_tick():
+        return capture_display()
+
+    # ═══════════════════════════════════════
+    # EVENT WIRING
+    # ═══════════════════════════════════════
+    
+    # Find & Replace Events
+    find_btn.click(fn=do_find, inputs=[editor, find_query, find_case], outputs=find_results)
+    find_all_btn.click(fn=do_find_all, inputs=[find_query], outputs=find_results)
+    
+    replace_btn.click(fn=do_replace, 
+                      inputs=[editor, find_query, replace_query, find_case],
+                      outputs=[editor, find_results, structure_view])
+
+    # Display Events
+    capture_btn.click(fn=capture_display, inputs=None, outputs=display_image)
+    auto_capture.change(fn=toggle_auto, inputs=auto_capture, outputs=auto_timer)
+    auto_timer.tick(fn=auto_capture_tick, inputs=None, outputs=display_image)
+
+    # Note: Ensure you have wired up the other buttons from the previous 
+    # part of the code (Run, Stop, Complete, etc.) or they won't work either!
 
 if __name__ == "__main__":
     import atexit
     atexit.register(terminal.cleanup)
     atexit.register(gui_mgr.stop)
     atexit.register(vdisplay.cleanup)
-    demo.launch(server_name="0.0.0.0", server_port=7860,
-                theme=theme, css=css, ssr_mode=False)
+    
+    # Use queue=True for timers and long running tasks
+    demo.queue().launch(server_name="0.0.0.0", server_port=7860, 
+                        theme=theme, css=css, ssr_mode=False,
+                        allowed_paths=["/tmp"])