Update app.py

app.py

@@ -60,6 +60,7 @@ class WebSocketInputOutput:
         """Reads a line from stdin, waiting for input from WebSocket."""
         try:
             print("DEBUG: Sending input_request to client.")
+            # CORRECTED: Access send_text via self.websocket
             await self.websocket.send_text(json.dumps({"type": "input_request"}))
             line = await self.input_queue.get() # Wait for input from client
             print(f"DEBUG: Received input from queue: {line.strip()[:50]}...")
@@ -391,12 +392,12 @@ class SecurePythonExecutor:
             
             # --- START TEST 1 MODIFICATIONS ---
             # DO NOT redirect sys.stdin/stdout/stderr here yet
-            print(f"DEBUG: Executor thread '{threading.current_thread().name}' started. (No IO redirected yet)") # UPDATED PRINT
+            print(f"DEBUG: Executor thread '{threading.current_thread().name}' started. (No IO redirected yet)")
             result['success'] = True
             # Do not call exec yet. Just write directly to output_buffer for testing.
-            websocket_io.output_buffer.write("DEBUG: Test message from executor thread (no exec).\n") # Write to buffer directly
+            websocket_io.output_buffer.write("DEBUG: Test message from executor thread (no exec).\n")
             # Do not call asyncio.run(websocket_io.write) here. The main loop will check buffer.
-            print("DEBUG: _execute_in_thread finished with direct buffer write.") # ADDED THIS LINE
+            print("DEBUG: _execute_in_thread finished with direct buffer write.")
             # --- END TEST 1 MODIFICATIONS ---
 
         # Use the event loop's default ThreadPoolExecutor to run the synchronous code
@@ -438,7 +439,8 @@ 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']}).")
-            await websocket_io.send_text(json.dumps({"type": "execution_complete", "result": result}))
+            # CORRECTED: Access send_text via websocket_io.websocket
+            await websocket_io.websocket.send_text(json.dumps({"type": "execution_complete", "result": result}))
             print("DEBUG: execution_complete message sent.")
         except WebSocketDisconnect:
             print("DEBUG: WebSocketDisconnect when sending execution_complete.")
@@ -484,17 +486,18 @@ async def websocket_endpoint(websocket: WebSocket):
 
             if data["type"] == "code":
                 code = data["content"]
-                print(f"Received code from {websocket_io.websocket.client}: {code[:50]}...") # Use websocket_io.websocket.client here
+                print(f"Received code from {websocket_io.websocket.client}: {code[:50]}...")
                 # Execute code asynchronously in the background
                 asyncio.create_task(executor.execute_code(code, websocket_io))
             elif data["type"] == "input":
                 user_input = data["content"]
-                print(f"Received input from {websocket_io.websocket.client}: {user_input.strip()[:50]}...") # Use websocket_io.websocket.client here
+                print(f"Received input from {websocket_io.websocket.client}: {user_input.strip()[:50]}...")
                 await websocket_io.input_queue.put(user_input + "\n") # Add newline for readline
             elif data["type"] == "ping":
                 # Respond to pings to keep connection alive
-                await websocket.send_text(json.dumps({"type": "pong"}))
-                print(f"DEBUG: Sent pong to {websocket_io.websocket.client}") # Use websocket_io.websocket.client here
+                # CORRECTED: Access send_text via websocket_io.websocket
+                await websocket_io.websocket.send_text(json.dumps({"type": "pong"}))
+                print(f"DEBUG: Sent pong to {websocket_io.websocket.client}")
             else:
                 await websocket_io.write(f"Unknown message type: {data['type']}\n")
 
@@ -512,4 +515,29 @@ async def websocket_endpoint(websocket: WebSocket):
                 websocket_io.input_queue.get_nowait()
             except asyncio.QueueEmpty:
                 pass
-        print(f"WebSocket connection closed for {websocket_io.websocket.client}")
+        print(f"WebSocket connection closed for {websocket_io.websocket.client}")
+
+# This block is for local development using `uvicorn`
+# Hugging Face Spaces will use `uvicorn` directly via the Dockerfile CMD
+if __name__ == "__main__":
+    import uvicorn
+    print("Starting Interactive Python Code Executor API (FastAPI)...")
+    print(f"Python version: {sys.version}")
+    print(f"Working directory: {os.getcwd()}")
+    
+    server_name = os.getenv('FASTAPI_SERVER_NAME', '0.0.0.0')
+    server_port = int(os.getenv('FASTAPI_SERVER_PORT', 7860))
+    
+    print(f"Starting server on {server_name}:{server_port}")
+    
+    try:
+        uvicorn.run(
+            "app:app",  # app:app means the `app` object in `app.py`
+            host=server_name,
+            port=server_port,
+            reload=False, # Set to True for local dev to auto-reload on code changes
+            log_level="info"
+        )
+    except Exception as e:
+        print(f"Failed to start server: {e}")
+        sys.exit(1)