Initial commit for proxy-lite API

Dockerfile

@@ -0,0 +1,71 @@
+# Dockerfile
+# Use a Python base image that is compatible with Playwright's system dependencies
+FROM python:3.9-slim-buster
+
+# Set the working directory inside the container
+WORKDIR /app
+
+# Install system dependencies required by Playwright's Chromium browser
+# These are common dependencies for running headless Chrome/Chromium.
+# This list might need minor adjustments based on specific runtime errors.
+RUN apt-get update && apt-get install -y \
+    # Core libraries for graphics/rendering
+    fonts-liberation \
+    libappindicator3-1 \
+    libasound2 \
+    libatk-bridge2.0-0 \
+    libatk1.0-0 \
+    libatspi2.0-0 \
+    libcairo2 \
+    libcups2 \
+    libdbus-1-3 \
+    libdrm2 \
+    libgdk-pixbuf2.0-0 \
+    libglib2.0-0 \
+    libgtk-3-0 \
+    libnspr4 \
+    libnss3 \
+    libpangocairo-1.0-0 \
+    libxcomposite1 \
+    libxdamage1 \
+    libxext6 \
+    libxfixes3 \
+    libxrandr2 \
+    libxrender1 \
+    libxss1 \
+    libxtst6 \
+    # Specific to Chromium/GPU (even if not using GPU, these are for display stack)
+    libgbm-dev \
+    libasound2-dev \
+    # xvfb provides a virtual display server, often necessary for headless browsers
+    xvfb \
+    # Ensure Chromium is installed on the system (Playwright uses its own, but sometimes useful)
+    chromium \
+    # Clean up apt caches to reduce image size
+    && rm -rf /var/lib/apt/lists/*
+
+# Copy Python dependencies and install them
+# This layer is cached efficiently if requirements.txt doesn't change
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy your application code into the container
+COPY . .
+
+# Install Playwright browser binaries within the container
+# This downloads Chromium into the container's Playwright-managed location.
+RUN playwright install chromium
+
+# Set environment variables for Playwright
+# PLAYWRIGHT_BROWSERS_PATH: Tells Playwright where to find the installed browsers.
+# DISPLAY, XDG_RUNTIME_DIR: Often needed for headless browser environments.
+ENV PLAYWRIGHT_BROWSERS_PATH=/ms-playwright/
+ENV DISPLAY=:99
+ENV XDG_RUNTIME_DIR=/tmp
+
+# Expose the port your Flask app will listen on. Hugging Face Spaces expects 7860.
+EXPOSE 7860
+
+# Define the command to run your Flask application using Gunicorn for production
+# Hugging Face Spaces will execute this command to start your web service.
+CMD exec gunicorn --bind 0.0.0.0:7860 --workers 2 --worker-class gevent app:app --timeout 300

Procfile

@@ -0,0 +1 @@
+web: gunicorn --bind 0.0.0.0:7860 --workers 2 --worker-class gevent app:app --timeout 300

app.py

@@ -0,0 +1,97 @@
+# app.py
+import asyncio
+from flask import Flask, request, jsonify
+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.")
+            raise ValueError("HF_API_TOKEN environment variable not set. Please set it as a Space secret.")
+
+        config = RunnerConfig.from_dict({
+            "environment": {
+                "name": "webbrowser",
+                "homepage": "https://www.google.com",
+                "headless": True, # Runs browser without a visible UI
+            },
+            "solver": {
+                "name": "simple",
+                "agent": {
+                    "name": "proxy_lite",
+                    "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
+                    }
+                }
+            }
+        })
+        _runner = Runner(config)
+        logger.info("Proxy-lite Runner initialized successfully.")
+    return _runner
+
+# Helper function to run asynchronous code within a synchronous Flask route
+def run_async_task(coro):
+    """Helper to run async coroutines in a synchronous context (like Flask routes)."""
+    loop = asyncio.get_event_loop()
+    return loop.run_until_complete(coro)
+
+@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:
+        logger.warning("Received request without 'task' field. Returning 400.")
+        return jsonify({"error": "No 'task' provided in request body"}), 400
+
+    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
+        return jsonify({"output": result})
+    except Exception as e:
+        logger.exception(f"Error processing task '{task}':") # Logs full traceback for debugging
+        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
+
+    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

requirements.txt

@@ -0,0 +1,6 @@
+Flask
+proxy-lite
+playwright
+playwright-stealth==1.0.6
+gunicorn 
+gevent