Upload folder using huggingface_hub

Dockerfile

@@ -0,0 +1,14 @@
+FROM python:3.12-slim
+
+RUN useradd -m -u 1000 user
+WORKDIR /app
+
+COPY --chown=user requirements.txt /app/requirements.txt
+RUN pip install --no-cache-dir -U pip && pip install --no-cache-dir -r /app/requirements.txt
+
+COPY --chown=user . /app
+
+USER user
+ENV HOME=/home/user PATH=/home/user/.local/bin:$PATH
+
+CMD ["waitress-serve", "--host=0.0.0.0", "--port=7860", "python_executor:app"]

README.md

@@ -1,10 +1,15 @@
 ---
-title: Kgot Python Executor
-emoji: 📚
-colorFrom: red
-colorTo: green
+title: KGoT Python Executor
+emoji: 🐍
+colorFrom: blue
+colorTo: gray
 sdk: docker
-pinned: false
+app_port: 7860
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+Remote Python executor for Knowledge Graph of Thoughts.
+
+Exposes:
+- `GET /` basic service info
+- `GET /health` health check
+- `POST /run` code execution endpoint compatible with the KGoT Python tool

python_executor.py

@@ -0,0 +1,74 @@
+import re
+import subprocess
+import sys
+from concurrent.futures import ThreadPoolExecutor, TimeoutError
+
+import requests
+from flask import Flask, jsonify, request
+from langchain_experimental.utilities import PythonREPL
+
+app = Flask(__name__)
+PYTHON_EXECUTOR_HOST = "0.0.0.0"
+PYTHON_EXECUTOR_PORT = 7860
+
+
+def is_standard_lib(package: str) -> bool:
+    return package in sys.stdlib_module_names
+
+
+def install(package: str) -> None:
+    subprocess.check_call([sys.executable, "-m", "pip", "install", package])
+
+
+@app.get("/")
+def root():
+    return {"service": "kgot-python-executor", "port": PYTHON_EXECUTOR_PORT}
+
+
+@app.get("/health")
+def health():
+    return {"status": "ok"}
+
+
+@app.route('/run', methods=['POST'])
+def run_code() -> tuple[requests.Response, int]:
+    timeout_seconds = 240
+
+    required_modules = request.json.get('required_modules', [])
+    if required_modules:
+        for module in required_modules:
+            if not is_standard_lib(module):
+                install(module)
+
+    code = request.json.get('code')
+    if not code:
+        return jsonify({"error": "No code provided"}), 400
+
+    python_repl = PythonREPL()
+
+    def execute_code() -> tuple[requests.Response, int]:
+        def is_error_string(s: str) -> bool:
+            error_pattern = re.compile(r'^[a-zA-Z_]+Error\((.*)\)$')
+            return bool(error_pattern.match(s))
+
+        result = python_repl.run(code)
+        if is_error_string(result):
+            return {"error": result}, 400
+        return {"output": result}, 200
+
+    executor = ThreadPoolExecutor(max_workers=1)
+    future = executor.submit(execute_code)
+
+    try:
+        result, status_code = future.result(timeout=timeout_seconds)
+    except TimeoutError:
+        return jsonify({"error": "Code execution timed out"}), 408
+    except Exception as e:
+        return jsonify({"error": str(e)}), 400
+
+    return jsonify(result), status_code
+
+
+if __name__ == '__main__':
+    from waitress import serve
+    serve(app, host=PYTHON_EXECUTOR_HOST, port=PYTHON_EXECUTOR_PORT)

requirements.txt

@@ -0,0 +1,4 @@
+flask
+waitress
+requests
+langchain_experimental