Update app.py

app.py

@@ -405,28 +405,29 @@ class SecurePythonExecutor:
                     '__name__': '__main__',
                     '__doc__': None,
                 }
+                print("DEBUG: Restricted globals prepared.") # ADDED THIS LINE
                 
-                print(f"DEBUG: Entering exec context for code (thread): {code[:100]}...") # NEW
-                # Execute the code
+                print(f"DEBUG: Preparing to execute code (thread): {code[:100]}...")
                 exec(code, restricted_globals)
-                print("DEBUG: Code execution completed successfully in thread.") # NEW
+                print("DEBUG: Code execution completed successfully in thread.")
                 result['success'] = True
                 
             except EOFError as e:
                 # This happens if the WebSocket disconnects during input()
                 result['error'] = f"Input stream closed: {str(e)}"
                 result['success'] = False
-                print(f"ERROR: EOFError during execution in thread: {e}") # NEW
+                print(f"ERROR: EOFError during execution in thread: {e}")
             except Exception as e:
                 error_msg = f"{type(e).__name__}: {str(e)}\n{traceback.format_exc()}"
-                print(f"ERROR: Unhandled exception during execution in thread: {error_msg}") # NEW
+                print(f"ERROR: Unhandled exception during execution in thread: {error_msg}")
                 # Write error to WebSocket and internal buffer
                 try:
                     # Use asyncio.run for writing within a synchronous thread, if necessary
                     # This is generally safe if it's a new event loop per call or in a separate thread.
                     asyncio.run(websocket_io.write(f"Error: {error_msg}\n"))
+                    print("DEBUG: Attempted to write error via websocket_io.write from thread.")
                 except Exception as write_e:
-                    print(f"ERROR: Failed to write error to WebSocket from thread: {write_e}")
+                    print(f"ERROR: Failed to write error to WebSocket from thread (inner): {write_e}")
                 websocket_io.error_buffer.write(error_msg)
                 result['success'] = False
             finally:
@@ -447,7 +448,7 @@ class SecurePythonExecutor:
         thread.join(timeout=self.timeout)
         
         if thread.is_alive():
-            print(f"DEBUG: Thread '{thread.name}' timed out after {self.timeout} seconds.") # NEW
+            print(f"DEBUG: Thread '{thread.name}' timed out after {self.timeout} seconds.")
             result['error'] = f"Code execution timed out after {self.timeout} seconds"
             result['success'] = False
             # Attempt to send timeout message to client
@@ -456,7 +457,7 @@ class SecurePythonExecutor:
             except Exception:
                 pass # Ignore if websocket is already closed
         else:
-            print(f"DEBUG: Thread '{thread.name}' finished within timeout.") # NEW
+            print(f"DEBUG: Thread '{thread.name}' finished within timeout.")
             result['output'] = websocket_io.output_buffer.getvalue()
             if websocket_io.error_buffer.getvalue():
                 result['error'] = websocket_io.error_buffer.getvalue()
@@ -466,11 +467,11 @@ class SecurePythonExecutor:
         
         # Send a final message to the client indicating execution is complete
         try:
-            print(f"DEBUG: Attempting to send execution_complete message (success={result['success']}).") # NEW
+            print(f"DEBUG: Attempting to send execution_complete message (success={result['success']}).")
             await websocket_io.send_text(json.dumps({"type": "execution_complete", "result": result}))
             print("DEBUG: execution_complete message sent.")
         except WebSocketDisconnect:
-            print("DEBUG: WebSocketDisconnect when sending execution_complete.") # NEW
+            print("DEBUG: WebSocketDisconnect when sending execution_complete.")
             pass # Ignore if websocket is already closed
         except Exception as e:
             print(f"ERROR: Error sending execution_complete message: {e}") # Warning if this fails
@@ -489,13 +490,12 @@ app = FastAPI(
     version="1.0.0",
 )
 
-# Serve the HTML frontend
 # Serve the HTML frontend
 @app.get("/", response_class=HTMLResponse)
 async def get_index():
     """Serves the interactive frontend."""
     # This assumes index.html is in the same directory as app.py
-    # Explicitly copy index.html to /app/index.html to avoid any ambiguity with '.' [cite: 9]
+    # [cite_start]Explicitly copy index.html to /app/index.html to avoid any ambiguity with '.' [cite: 5, 6]
     with open("index.html", "r") as f:
         return HTMLResponse(content=f.read())