changing behavior after failure

app.py

@@ -219,27 +219,11 @@ function() {
                     console.log("Error detected!");
                     showBSOD('Error');
                     clearInterval(resultsInterval);
-                    window.agentFailed = true;
                     break;
                 }
             }
         }, 1000);
     };
-
-    // Add a new function to conditionally go to interactive mode
-    window.shouldSwitchToInteractive = function() {
-        const resultsElements = document.querySelectorAll('textarea, .output-text');
-        for (let elem of resultsElements) {
-            const content = elem.value || elem.innerText || '';
-            // If we see an error message or our flag is set, don't switch to interactive
-            if (content.includes('Error running agent') || window.agentFailed === true) {
-                console.log("Error detected, not switching to interactive mode");
-                return false;
-            }
-        }
-        console.log("No errors detected, switching to interactive mode");
-        return true;
-    };
     
     
     // Start monitoring for timeouts immediately
@@ -253,7 +237,6 @@ function() {
         if (e.target.tagName === 'BUTTON') {
             if (e.target.innerText === "Let's go!") {
                 resetBSOD();
-                window.agentFailed = false;
             }
             setTimeout(monitorForErrors, 3000);
         }
@@ -487,6 +470,17 @@ with gr.Blocks(css=custom_css, js=custom_js) as demo:
     def set_interactive_mode(request: gr.Request):
         return update_html(True, request)
     
+
+    # Function to check result and conditionally set interactive mode
+    def check_and_set_interactive(result, request: gr.Request):
+        if result and not result.startswith("Error running agent"):
+            # Only set interactive mode if no error
+            return update_html(True, request)
+        else:
+            # Return the current HTML to avoid changing the display
+            # This will keep the BSOD visible
+            return gr.update()
+
     # Chain the events
     # 1. Set view-only mode when button is clicked
     view_only_event = update_btn.click(
@@ -494,21 +488,19 @@ with gr.Blocks(css=custom_css, js=custom_js) as demo:
         inputs=[task_input], 
         outputs=html_output
     )
-    
+
     # 2. Then run the agent task
     task_result = view_only_event.then(
         fn=run_agent_task,
         inputs=[task_input],
         outputs=results_output
     )
-    
-    # # 3. Then set back to interactive mode
+
+    # 3. Then check the result and conditionally set to interactive mode
     task_result.then(
-        fn=set_interactive_mode,
-        inputs=None,  # No inputs needed here
-        outputs=html_output,
-        _js="() => window.shouldSwitchToInteractive() ? [] : null"
-        
+        fn=check_and_set_interactive,
+        inputs=[results_output],  # Pass the result text to check
+        outputs=html_output
     )
     
     # Load the sandbox on app start with initial HTML