Fix asyncio event loop management in app.py for Gunicorn workers

app.py

@@ -5,21 +5,17 @@ from proxy_lite import Runner, RunnerConfig
 import os
 import logging
 
-# Configure logging for better visibility in Space logs
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
 
 app = Flask(__name__)
 
-# Global runner instance. It will be initialized once.
 _runner = None
 
 async def initialize_runner():
-    """Initializes the Proxy-lite Runner asynchronously."""
     global _runner
     if _runner is None:
         logger.info("Initializing Proxy-lite Runner...")
-        # Retrieve Hugging Face API Token from environment variables (set as Space Secret)
         hf_api_token = os.environ.get("HF_API_TOKEN")
         if not hf_api_token:
             logger.error("HF_API_TOKEN environment variable not set. Cannot initialize Runner.")
@@ -29,7 +25,7 @@ async def initialize_runner():
             "environment": {
                 "name": "webbrowser",
                 "homepage": "https://www.google.com",
-                "headless": True, # Runs browser without a visible UI
+                "headless": True,
             },
             "solver": {
                 "name": "simple",
@@ -38,9 +34,8 @@ async def initialize_runner():
                     "client": {
                         "name": "convergence",
                         "model_id": "convergence-ai/proxy-lite-3b",
-                        # THIS IS THE HUGGING FACE INFERENCE API ENDPOINT
                         "api_base": "https://api-inference.huggingface.co/models/convergence-ai/proxy-lite-3b",
-                        "api_key": hf_api_token # Use the token from environment variable
+                        "api_key": hf_api_token
                     }
                 }
             }
@@ -49,15 +44,27 @@ async def initialize_runner():
         logger.info("Proxy-lite Runner initialized successfully.")
     return _runner
 
-# Helper function to run asynchronous code within a synchronous Flask route
+# --- MODIFIED run_async_task FUNCTION ---
 def run_async_task(coro):
-    """Helper to run async coroutines in a synchronous context (like Flask routes)."""
-    loop = asyncio.get_event_loop()
+    """
+    Helper to run async coroutines in a synchronous context (like Flask routes).
+    Ensures an event loop exists and runs the coroutine.
+    """
+    try:
+        # Try to get the running loop (for current thread/greenlet)
+        loop = asyncio.get_running_loop()
+    except RuntimeError:
+        # If no loop is running, create a new one for this thread
+        loop = asyncio.new_event_loop()
+        asyncio.set_event_loop(loop)
+    
+    # Run the coroutine until it completes
     return loop.run_until_complete(coro)
+# --- END MODIFIED run_async_task FUNCTION ---
+
 
 @app.route('/run_proxy_task', methods=['POST'])
 def run_proxy_task_endpoint():
-    """API endpoint to receive a task and execute it with Proxy-lite."""
     data = request.json
     task = data.get('task')
     if not task:
@@ -66,32 +73,23 @@ def run_proxy_task_endpoint():
 
     logger.info(f"Received task for proxy-lite: '{task}'")
     try:
-        # Initialize runner if not already done
         runner = run_async_task(initialize_runner())
-        # Run the task using the proxy-lite runner
         result = run_async_task(runner.run(task))
 
-        logger.info(f"Proxy-lite task completed. Output: {result[:200]}...") # Log truncated output
-        # Return the result as a JSON response
+        logger.info(f"Proxy-lite task completed. Output: {result[:200]}...")
         return jsonify({"output": result})
     except Exception as e:
-        logger.exception(f"Error processing task '{task}':") # Logs full traceback for debugging
+        logger.exception(f"Error processing task '{task}':")
         return jsonify({"error": f"An error occurred: {str(e)}. Check logs for details."}), 500
 
 @app.route('/')
 def root():
-    """Basic root endpoint for health check."""
     logger.info("Root endpoint accessed.")
     return "Proxy-lite API is running. Send POST requests to /run_proxy_task with a 'task' in JSON body."
 
 if __name__ == '__main__':
-    # Check if HF_API_TOKEN is set during local development/testing
     if not os.environ.get("HF_API_TOKEN"):
         logger.error("HF_API_TOKEN environment variable is not set. Please set it for local testing.")
-        # For local execution, you might choose to exit or just log
-        # For deployment on Hugging Face Spaces, it MUST be set as a Secret.
-        # exit(1) # Uncomment this if you want to force exit without token locally
-
+        exit(1)
     logger.info("Starting Flask development server on 0.0.0.0:7860...")
-    # Hugging Face Spaces requires binding to 0.0.0.0 and listening on port 7860
-    app.run(host='0.0.0.0', port=7860, debug=True) # debug=True for local, disable for production
+    app.run(host='0.0.0.0', port=7860, debug=True)