Upload folder using huggingface_hub

Dockerfile

@@ -1,14 +1,38 @@
 FROM python:3.12-slim
 
+# Install system dependencies for audio/image processing
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    ffmpeg \
+    libsndfile1 \
+    && rm -rf /var/lib/apt/lists/*
+
 RUN useradd -m -u 1000 user
 WORKDIR /app
 
+# -------------------------------------------
+# Step 1: Install PyTorch CPU-only (pinned to 2.5.1 for PyG compatibility)
+# -------------------------------------------
+RUN pip install --no-cache-dir -U pip && \
+    pip install --no-cache-dir torch==2.5.1+cpu torchvision==0.20.1+cpu --index-url https://download.pytorch.org/whl/cpu
+
+# -------------------------------------------
+# Step 2: Install PyTorch Geometric + extensions (CPU)
+# -------------------------------------------
+RUN pip install --no-cache-dir torch_geometric && \
+    pip install --no-cache-dir pyg_lib torch_scatter torch_sparse torch_cluster -f https://data.pyg.org/whl/torch-2.5.1+cpu.html
+
+# -------------------------------------------
+# Step 3: Install remaining requirements
+# -------------------------------------------
 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
+RUN pip install --no-cache-dir -r /app/requirements.txt
 
+# -------------------------------------------
+# Step 4: Copy application
+# -------------------------------------------
 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"]
+CMD ["waitress-serve", "--host=0.0.0.0", "--port=7860", "python_executor:app"]

README.md

@@ -7,9 +7,64 @@ sdk: docker
 app_port: 7860
 ---
 
-Remote Python executor for Knowledge Graph of Thoughts.
+Remote Python executor for Knowledge Graph of Thoughts with comprehensive ML/data science capabilities.
 
-Exposes:
-- `GET /` basic service info
-- `GET /health` health check
-- `POST /run` code execution endpoint compatible with the KGoT Python tool
+## Hardware
+- **CPU Basic** (Free): 2 vCPU, 16GB RAM, 50GB ephemeral disk
+
+## Included Libraries
+
+### Core Data
+- numpy, pandas
+
+### NLP
+- nltk, spacy, transformers, textblob, gensim, fastembed
+
+### Vision
+- pillow, opencv-python-headless, torchvision, scikit-image
+
+### Advanced Data
+- polars, duckdb, sktime
+
+### Time Series & Graph
+- torch (CPU), torch_geometric, networkx
+- pyg_lib, torch_scatter, torch_sparse, torch_cluster
+
+### Audio
+- pydub, soundfile, librosa
+
+### Documents
+- python-docx, csvkit, rich, svgwrite, svgpathtools, PyMuPDF
+
+### Utilities
+- pydantic, tqdm, ipython, click, tenacity, ujson, msgpack, xxhash, zstandard, httpx
+
+### ML
+- scikit-learn, joblib
+
+### Visualization
+- matplotlib, seaborn
+
+## API Endpoints
+
+- `GET /` - Basic service info
+- `GET /health` - Health check
+- `POST /run` - Code execution endpoint
+
+### POST /run
+```python
+{
+   "required_modules": ["list", "of", "packages"],  # Optional - auto-install
+   "code": "python code to execute"
+}
+```
+
+Returns:
+```python
+{
+    "output": "execution output"
+}
+```
+
+## Timeout
+- Maximum execution time: 240 seconds

requirements.txt

@@ -1,4 +1,90 @@
-flask
-waitress
-requests
-langchain_experimental
+# ===========================================
+# KGoT Python Executor - Enhanced Requirements
+# ===========================================
+
+# Existing - Flask server
+flask>=3.0.0
+waitress>=3.0.0
+langchain_experimental>=0.0.60
+requests>=2.31.0
+
+# -------------------------------------------
+# Core Data
+# -------------------------------------------
+numpy>=1.26.0
+pandas>=2.2.0
+
+# -------------------------------------------
+# NLP
+# -------------------------------------------
+nltk>=3.8
+spacy>=3.7.0
+transformers>=4.40.0
+textblob>=0.18.0
+gensim>=4.3.0
+fastembed>=0.2.0
+
+# -------------------------------------------
+# Vision
+# -------------------------------------------
+pillow>=10.0.0
+opencv-python-headless>=4.9.0
+scikit-image>=0.22.0
+# torchvision installed via Dockerfile (CPU-only)
+
+# -------------------------------------------
+# Advanced Data
+# -------------------------------------------
+polars>=0.20.0
+duckdb>=0.10.0
+sktime>=0.26.0
+
+# -------------------------------------------
+# Time Series & Graph
+# -------------------------------------------
+networkx>=3.2
+# torch, torch_geometric, pyg_lib, torch_scatter, torch_sparse, torch_cluster
+# installed via Dockerfile for CPU-only versions
+
+# -------------------------------------------
+# Audio
+# -------------------------------------------
+pydub>=0.25.0
+soundfile>=0.12.0
+librosa>=0.10.0
+
+# -------------------------------------------
+# Documents
+# -------------------------------------------
+python-docx>=1.1.0
+csvkit>=2.0.0
+rich>=13.0.0
+svgwrite>=1.4.0
+svgpathtools>=1.6.0
+PyMuPDF>=1.24.0
+
+# -------------------------------------------
+# Utilities
+# -------------------------------------------
+pydantic>=2.6.0
+tqdm>=4.66.0
+ipython>=8.20.0
+click>=8.1.0
+tenacity>=8.2.0
+ujson>=5.9.0
+msgpack>=1.0.7
+xxhash>=3.4.0
+zstandard>=0.22.0
+httpx>=0.27.0
+
+# -------------------------------------------
+# ML
+# -------------------------------------------
+scikit-learn>=1.4.0
+joblib>=1.3.0
+
+# -------------------------------------------
+# Visualization
+# -------------------------------------------
+matplotlib>=3.8.0
+seaborn>=0.13.0