Spaces:
Paused
Paused
Upload 12 files
Browse files- .gitignore +125 -0
- Dockerfile +47 -0
- README.md +273 -11
- app.py +606 -0
- best.onnx +3 -0
- best_upright.onnx +3 -0
- bestspiral.onnx +3 -0
- data.yaml +115 -0
- data_upright.yaml +12 -0
- dataspiral.yaml +12 -0
- requirements.txt +22 -0
- test-api.py +343 -0
.gitignore
ADDED
|
@@ -0,0 +1,125 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Byte-compiled / optimized / DLL files
|
| 2 |
+
__pycache__/
|
| 3 |
+
*.py[cod]
|
| 4 |
+
*$py.class
|
| 5 |
+
|
| 6 |
+
# C extensions
|
| 7 |
+
*.so
|
| 8 |
+
|
| 9 |
+
# Distribution / packaging
|
| 10 |
+
.Python
|
| 11 |
+
build/
|
| 12 |
+
develop-eggs/
|
| 13 |
+
dist/
|
| 14 |
+
downloads/
|
| 15 |
+
eggs/
|
| 16 |
+
.eggs/
|
| 17 |
+
lib/
|
| 18 |
+
lib64/
|
| 19 |
+
parts/
|
| 20 |
+
sdist/
|
| 21 |
+
var/
|
| 22 |
+
wheels/
|
| 23 |
+
pip-wheel-metadata/
|
| 24 |
+
share/python-wheels/
|
| 25 |
+
*.egg-info/
|
| 26 |
+
.installed.cfg
|
| 27 |
+
*.egg
|
| 28 |
+
MANIFEST
|
| 29 |
+
|
| 30 |
+
# PyInstaller
|
| 31 |
+
*.manifest
|
| 32 |
+
*.spec
|
| 33 |
+
|
| 34 |
+
# Installer logs
|
| 35 |
+
pip-log.txt
|
| 36 |
+
pip-delete-this-directory.txt
|
| 37 |
+
|
| 38 |
+
# Unit test / coverage reports
|
| 39 |
+
htmlcov/
|
| 40 |
+
.tox/
|
| 41 |
+
.nox/
|
| 42 |
+
.coverage
|
| 43 |
+
.coverage.*
|
| 44 |
+
.cache
|
| 45 |
+
nosetests.xml
|
| 46 |
+
coverage.xml
|
| 47 |
+
*.cover
|
| 48 |
+
*.py,cover
|
| 49 |
+
.hypothesis/
|
| 50 |
+
.pytest_cache/
|
| 51 |
+
|
| 52 |
+
# Virtual environments
|
| 53 |
+
.env
|
| 54 |
+
.venv
|
| 55 |
+
env/
|
| 56 |
+
venv/
|
| 57 |
+
ENV/
|
| 58 |
+
env.bak/
|
| 59 |
+
venv.bak/
|
| 60 |
+
|
| 61 |
+
# Spyder project settings
|
| 62 |
+
.spyderproject
|
| 63 |
+
.spyproject
|
| 64 |
+
|
| 65 |
+
# Rope project settings
|
| 66 |
+
.ropeproject
|
| 67 |
+
|
| 68 |
+
# mkdocs documentation
|
| 69 |
+
/site
|
| 70 |
+
|
| 71 |
+
# mypy
|
| 72 |
+
.mypy_cache/
|
| 73 |
+
.dmypy.json
|
| 74 |
+
dmypy.json
|
| 75 |
+
|
| 76 |
+
# Model files (large, should be uploaded manually to HF)
|
| 77 |
+
*.onnx
|
| 78 |
+
*.pt
|
| 79 |
+
*.pth
|
| 80 |
+
*.h5
|
| 81 |
+
*.pb
|
| 82 |
+
|
| 83 |
+
# Image files (test data)
|
| 84 |
+
*.jpg
|
| 85 |
+
*.jpeg
|
| 86 |
+
*.png
|
| 87 |
+
*.gif
|
| 88 |
+
*.bmp
|
| 89 |
+
*.tiff
|
| 90 |
+
|
| 91 |
+
# Debug images folder
|
| 92 |
+
debug_images/
|
| 93 |
+
debug_*/
|
| 94 |
+
|
| 95 |
+
# IDE settings
|
| 96 |
+
.vscode/
|
| 97 |
+
.idea/
|
| 98 |
+
*.swp
|
| 99 |
+
*.swo
|
| 100 |
+
*~
|
| 101 |
+
|
| 102 |
+
# OS generated files
|
| 103 |
+
.DS_Store
|
| 104 |
+
.DS_Store?
|
| 105 |
+
._*
|
| 106 |
+
.Spotlight-V100
|
| 107 |
+
.Trashes
|
| 108 |
+
ehthumbs.db
|
| 109 |
+
Thumbs.db
|
| 110 |
+
|
| 111 |
+
# Logs
|
| 112 |
+
*.log
|
| 113 |
+
logs/
|
| 114 |
+
|
| 115 |
+
# Temporary files
|
| 116 |
+
*.tmp
|
| 117 |
+
*.temp
|
| 118 |
+
*.cache
|
| 119 |
+
|
| 120 |
+
# Local configuration files
|
| 121 |
+
.env.local
|
| 122 |
+
config.local.*
|
| 123 |
+
|
| 124 |
+
# HF Spaces specific
|
| 125 |
+
.git/
|
Dockerfile
ADDED
|
@@ -0,0 +1,47 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Multi-stage Docker build untuk FunCaptcha Server di HF Spaces
|
| 2 |
+
# Optimasi untuk ukuran minimal dan performa maksimal
|
| 3 |
+
|
| 4 |
+
FROM python:3.11-slim as base
|
| 5 |
+
|
| 6 |
+
# Install system dependencies yang diperlukan (minimal)
|
| 7 |
+
RUN apt-get update && apt-get install -y \
|
| 8 |
+
libglib2.0-0 \
|
| 9 |
+
libsm6 \
|
| 10 |
+
libxext6 \
|
| 11 |
+
libxrender-dev \
|
| 12 |
+
libgomp1 \
|
| 13 |
+
libglib2.0-0 \
|
| 14 |
+
&& rm -rf /var/lib/apt/lists/* \
|
| 15 |
+
&& apt-get clean
|
| 16 |
+
|
| 17 |
+
# Set working directory
|
| 18 |
+
WORKDIR /app
|
| 19 |
+
|
| 20 |
+
# Copy requirements first (untuk layer caching yang lebih baik)
|
| 21 |
+
COPY requirements.txt .
|
| 22 |
+
|
| 23 |
+
# Install Python dependencies dengan optimasi
|
| 24 |
+
RUN pip install --no-cache-dir --upgrade pip && \
|
| 25 |
+
pip install --no-cache-dir -r requirements.txt && \
|
| 26 |
+
pip cache purge
|
| 27 |
+
|
| 28 |
+
# Copy aplikasi
|
| 29 |
+
COPY . .
|
| 30 |
+
|
| 31 |
+
# Set environment variables untuk optimasi
|
| 32 |
+
ENV PYTHONUNBUFFERED=1
|
| 33 |
+
ENV PYTHONDONTWRITEBYTECODE=1
|
| 34 |
+
ENV OMP_NUM_THREADS=2
|
| 35 |
+
ENV MKL_NUM_THREADS=2
|
| 36 |
+
ENV OPENBLAS_NUM_THREADS=2
|
| 37 |
+
ENV NUMEXPR_NUM_THREADS=2
|
| 38 |
+
|
| 39 |
+
# Expose port
|
| 40 |
+
EXPOSE 7860
|
| 41 |
+
|
| 42 |
+
# Health check untuk monitoring
|
| 43 |
+
HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
|
| 44 |
+
CMD curl -f http://localhost:7860/health || exit 1
|
| 45 |
+
|
| 46 |
+
# Run aplikasi
|
| 47 |
+
CMD ["python", "-m", "uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]
|
README.md
CHANGED
|
@@ -1,11 +1,273 @@
|
|
| 1 |
-
-
|
| 2 |
-
|
| 3 |
-
|
| 4 |
-
|
| 5 |
-
|
| 6 |
-
|
| 7 |
-
|
| 8 |
-
|
| 9 |
-
--
|
| 10 |
-
|
| 11 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# π§© FunCaptcha Solver API - Hugging Face Spaces
|
| 2 |
+
|
| 3 |
+
> **High-performance FunCaptcha solver dengan fuzzy matching dan API key authentication**
|
| 4 |
+
|
| 5 |
+
## π Features
|
| 6 |
+
|
| 7 |
+
- β
**FastAPI** - High performance async web framework
|
| 8 |
+
- β
**Docker optimization** - Multi-stage build untuk size minimal
|
| 9 |
+
- β
**Fuzzy label matching** - Handle variasi label seperti "ice cream" vs "ice"
|
| 10 |
+
- β
**API key authentication** - Secure access menggunakan HF Secrets
|
| 11 |
+
- β
**Response caching** - Fast responses untuk requests yang sama
|
| 12 |
+
- β
**Multi-model support** - Support berbagai jenis CAPTCHA challenges
|
| 13 |
+
- β
**Memory efficient** - Optimized untuk HF Spaces environment
|
| 14 |
+
- β
**Auto health checks** - Built-in monitoring
|
| 15 |
+
|
| 16 |
+
## π Supported Challenge Types
|
| 17 |
+
|
| 18 |
+
| Challenge Type | Description | Response |
|
| 19 |
+
|---|---|---|
|
| 20 |
+
| `pick_the` | Pick specific objects dari images | `{status, box, confidence}` |
|
| 21 |
+
| `upright` | Find correctly oriented objects | `{status, button_index, confidence}` |
|
| 22 |
+
|
| 23 |
+
## π§ Deployment ke Hugging Face Spaces
|
| 24 |
+
|
| 25 |
+
### 1. Create New Space
|
| 26 |
+
|
| 27 |
+
1. Buka [Hugging Face Spaces](https://huggingface.co/spaces)
|
| 28 |
+
2. Click **"Create new Space"**
|
| 29 |
+
3. Pilih:
|
| 30 |
+
- **Space name**: `funcaptcha-solver-api` (atau nama pilihan Anda)
|
| 31 |
+
- **License**: `mit`
|
| 32 |
+
- **Space SDK**: `Docker`
|
| 33 |
+
- **Space visibility**: `Private` (recommended)
|
| 34 |
+
|
| 35 |
+
### 2. Upload Files
|
| 36 |
+
|
| 37 |
+
Upload semua files dalam folder ini ke Space repository:
|
| 38 |
+
|
| 39 |
+
```
|
| 40 |
+
hf-funcaptcha-deployment/
|
| 41 |
+
βββ Dockerfile # Docker configuration
|
| 42 |
+
βββ app.py # Main FastAPI application
|
| 43 |
+
βββ requirements.txt # Python dependencies
|
| 44 |
+
βββ README.md # Documentation (this file)
|
| 45 |
+
βββ data.yaml # Class names untuk default model
|
| 46 |
+
βββ best.onnx # ONNX model file (upload manually)
|
| 47 |
+
βββ test-api.py # Testing script
|
| 48 |
+
```
|
| 49 |
+
|
| 50 |
+
### 3. Setup API Key (CRITICAL!)
|
| 51 |
+
|
| 52 |
+
1. Di Space settings, buka tab **"Settings"**
|
| 53 |
+
2. Scroll ke **"Repository secrets"**
|
| 54 |
+
3. Add new secret:
|
| 55 |
+
- **Name**: `FUNCAPTCHA_API_KEY`
|
| 56 |
+
- **Value**: `your-secure-api-key-here` (generate strong key)
|
| 57 |
+
|
| 58 |
+
> β οΈ **PENTING**: Tanpa API key, aplikasi tidak akan start!
|
| 59 |
+
|
| 60 |
+
### 4. Upload Model Files
|
| 61 |
+
|
| 62 |
+
Upload model files ke Space repository:
|
| 63 |
+
|
| 64 |
+
- `best.onnx` - Main detection model
|
| 65 |
+
- `data.yaml` - Class names configuration
|
| 66 |
+
- `bestspiral.onnx` (optional) - Spiral galaxy model
|
| 67 |
+
- `dataspiral.yaml` (optional) - Spiral galaxy classes
|
| 68 |
+
- `best_Upright.onnx` (optional) - Upright detection model
|
| 69 |
+
- `data_upright.yaml` (optional) - Upright detection classes
|
| 70 |
+
|
| 71 |
+
### 5. Deploy & Test
|
| 72 |
+
|
| 73 |
+
1. Space akan auto-deploy setelah files uploaded
|
| 74 |
+
2. Wait untuk build process selesai (~5-10 minutes)
|
| 75 |
+
3. Test endpoint: `https://your-space-url.hf.space/health`
|
| 76 |
+
|
| 77 |
+
## π Authentication
|
| 78 |
+
|
| 79 |
+
Semua API endpoints require **Bearer token authentication**:
|
| 80 |
+
|
| 81 |
+
```bash
|
| 82 |
+
# Example request
|
| 83 |
+
curl -X POST "https://your-space-url.hf.space/solve" \\
|
| 84 |
+
-H "Authorization: Bearer YOUR_API_KEY" \\
|
| 85 |
+
-H "Content-Type: application/json" \\
|
| 86 |
+
-d '{
|
| 87 |
+
"challenge_type": "pick_the",
|
| 88 |
+
"image_b64": "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAA...",
|
| 89 |
+
"target_label": "ice cream"
|
| 90 |
+
}'
|
| 91 |
+
```
|
| 92 |
+
|
| 93 |
+
## π API Documentation
|
| 94 |
+
|
| 95 |
+
### Endpoints
|
| 96 |
+
|
| 97 |
+
| Endpoint | Method | Description | Auth Required |
|
| 98 |
+
|---|---|---|---|
|
| 99 |
+
| `/` | GET | Root info & status | β |
|
| 100 |
+
| `/health` | GET | Health check | β |
|
| 101 |
+
| `/solve` | POST | Solve FunCaptcha | β
|
|
| 102 |
+
| `/docs` | GET | Interactive API docs | β |
|
| 103 |
+
|
| 104 |
+
### Request Format
|
| 105 |
+
|
| 106 |
+
```json
|
| 107 |
+
{
|
| 108 |
+
"challenge_type": "pick_the",
|
| 109 |
+
"image_b64": "data:image/png;base64,iVBORw0KGgo...",
|
| 110 |
+
"target_label": "ice cream"
|
| 111 |
+
}
|
| 112 |
+
```
|
| 113 |
+
|
| 114 |
+
### Response Format
|
| 115 |
+
|
| 116 |
+
```json
|
| 117 |
+
{
|
| 118 |
+
"status": "success",
|
| 119 |
+
"box": [120.5, 80.3, 150.0, 100.0],
|
| 120 |
+
"confidence": 0.89,
|
| 121 |
+
"processing_time": 0.245,
|
| 122 |
+
"message": null
|
| 123 |
+
}
|
| 124 |
+
```
|
| 125 |
+
|
| 126 |
+
## π§ͺ Testing
|
| 127 |
+
|
| 128 |
+
### Local Testing
|
| 129 |
+
|
| 130 |
+
```bash
|
| 131 |
+
# Install dependencies
|
| 132 |
+
pip install -r requirements.txt
|
| 133 |
+
|
| 134 |
+
# Set API key
|
| 135 |
+
export FUNCAPTCHA_API_KEY="your-test-key"
|
| 136 |
+
|
| 137 |
+
# Run server
|
| 138 |
+
python app.py
|
| 139 |
+
|
| 140 |
+
# Test with provided script
|
| 141 |
+
python test-api.py
|
| 142 |
+
```
|
| 143 |
+
|
| 144 |
+
### Production Testing
|
| 145 |
+
|
| 146 |
+
```python
|
| 147 |
+
import requests
|
| 148 |
+
import base64
|
| 149 |
+
|
| 150 |
+
# Load test image
|
| 151 |
+
with open("test_image.png", "rb") as f:
|
| 152 |
+
image_b64 = base64.b64encode(f.read()).decode()
|
| 153 |
+
|
| 154 |
+
# Make request
|
| 155 |
+
response = requests.post(
|
| 156 |
+
"https://your-space-url.hf.space/solve",
|
| 157 |
+
headers={"Authorization": "Bearer YOUR_API_KEY"},
|
| 158 |
+
json={
|
| 159 |
+
"challenge_type": "pick_the",
|
| 160 |
+
"image_b64": f"data:image/png;base64,{image_b64}",
|
| 161 |
+
"target_label": "ice cream"
|
| 162 |
+
}
|
| 163 |
+
)
|
| 164 |
+
|
| 165 |
+
print(response.json())
|
| 166 |
+
```
|
| 167 |
+
|
| 168 |
+
## βοΈ Configuration
|
| 169 |
+
|
| 170 |
+
### Model Configuration
|
| 171 |
+
|
| 172 |
+
Edit `CONFIGS` in `app.py` untuk custom models:
|
| 173 |
+
|
| 174 |
+
```python
|
| 175 |
+
CONFIGS = {
|
| 176 |
+
'default': {
|
| 177 |
+
'model_path': 'best.onnx',
|
| 178 |
+
'yaml_path': 'data.yaml',
|
| 179 |
+
'input_size': 640,
|
| 180 |
+
'confidence_threshold': 0.4,
|
| 181 |
+
'nms_threshold': 0.2
|
| 182 |
+
}
|
| 183 |
+
# Add more models...
|
| 184 |
+
}
|
| 185 |
+
```
|
| 186 |
+
|
| 187 |
+
### Environment Variables
|
| 188 |
+
|
| 189 |
+
| Variable | Description | Required |
|
| 190 |
+
|---|---|---|
|
| 191 |
+
| `FUNCAPTCHA_API_KEY` | API key untuk authentication | β
|
|
| 192 |
+
| `OMP_NUM_THREADS` | CPU threads untuk optimization | β (default: 2) |
|
| 193 |
+
| `CACHE_MAX_SIZE` | Maximum cache entries | β (default: 100) |
|
| 194 |
+
|
| 195 |
+
## π Monitoring & Debugging
|
| 196 |
+
|
| 197 |
+
### Health Check
|
| 198 |
+
|
| 199 |
+
```bash
|
| 200 |
+
curl https://your-space-url.hf.space/health
|
| 201 |
+
```
|
| 202 |
+
|
| 203 |
+
### Logs
|
| 204 |
+
|
| 205 |
+
- Check HF Spaces logs dalam space interface
|
| 206 |
+
- Logs include processing times, cache hits, errors
|
| 207 |
+
|
| 208 |
+
### Performance Metrics
|
| 209 |
+
|
| 210 |
+
- **Processing time**: Biasanya < 300ms per request
|
| 211 |
+
- **Memory usage**: ~500MB dengan 1 model loaded
|
| 212 |
+
- **Cache hit rate**: Displayed in `/health` endpoint
|
| 213 |
+
|
| 214 |
+
## π¨ Troubleshooting
|
| 215 |
+
|
| 216 |
+
### Common Issues
|
| 217 |
+
|
| 218 |
+
1. **"API key not found"**
|
| 219 |
+
- Solution: Set `FUNCAPTCHA_API_KEY` di Space secrets
|
| 220 |
+
|
| 221 |
+
2. **"Model file not found"**
|
| 222 |
+
- Solution: Upload model files (.onnx, .yaml) ke repository
|
| 223 |
+
|
| 224 |
+
3. **"401 Unauthorized"**
|
| 225 |
+
- Solution: Check API key dalam request header
|
| 226 |
+
|
| 227 |
+
4. **Slow responses**
|
| 228 |
+
- Check: Model loading, enable caching
|
| 229 |
+
- Monitor: Memory usage dalam Space logs
|
| 230 |
+
|
| 231 |
+
### Debug Mode
|
| 232 |
+
|
| 233 |
+
Untuk debugging, set logging level:
|
| 234 |
+
|
| 235 |
+
```python
|
| 236 |
+
import logging
|
| 237 |
+
logging.basicConfig(level=logging.DEBUG)
|
| 238 |
+
```
|
| 239 |
+
|
| 240 |
+
## π Performance Optimization
|
| 241 |
+
|
| 242 |
+
- β
**Model caching** - Models loaded once, reused
|
| 243 |
+
- β
**Response caching** - Identical requests cached
|
| 244 |
+
- β
**CPU optimization** - ONNX dengan CPU-specific settings
|
| 245 |
+
- β
**Memory efficiency** - Minimal memory footprint
|
| 246 |
+
- β
**Async operations** - Non-blocking request handling
|
| 247 |
+
|
| 248 |
+
## π Security
|
| 249 |
+
|
| 250 |
+
- β
**API key authentication** via Bearer tokens
|
| 251 |
+
- β
**CORS protection** configured
|
| 252 |
+
- β
**Input validation** dengan Pydantic models
|
| 253 |
+
- β
**Error handling** tanpa expose internal details
|
| 254 |
+
|
| 255 |
+
## π Support
|
| 256 |
+
|
| 257 |
+
- **Issues**: Report dalam repository issues
|
| 258 |
+
- **Updates**: Check Space status dan rebuild jika needed
|
| 259 |
+
- **Performance**: Monitor via `/health` endpoint
|
| 260 |
+
|
| 261 |
+
---
|
| 262 |
+
|
| 263 |
+
> π― **Ready to deploy?** Upload files, set API key, dan Space siap digunakan!
|
| 264 |
+
|
| 265 |
+
## π Client Integration
|
| 266 |
+
|
| 267 |
+
Update client untuk menggunakan Space URL:
|
| 268 |
+
|
| 269 |
+
```javascript
|
| 270 |
+
// funcaptcha-blob-fixed.js
|
| 271 |
+
const SOLVER_SERVER_URL = 'https://your-space-url.hf.space/solve';
|
| 272 |
+
const API_KEY = 'your-api-key-here';
|
| 273 |
+
```
|
app.py
ADDED
|
@@ -0,0 +1,606 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""
|
| 2 |
+
π FunCaptcha Solver API - Hugging Face Spaces Deployment
|
| 3 |
+
Optimized for speed, memory efficiency, and scalability
|
| 4 |
+
|
| 5 |
+
Features:
|
| 6 |
+
- FastAPI async operations
|
| 7 |
+
- API key authentication via HF secrets
|
| 8 |
+
- Fuzzy label matching
|
| 9 |
+
- Memory-efficient model loading
|
| 10 |
+
- ONNX CPU optimization
|
| 11 |
+
- Request caching for performance
|
| 12 |
+
"""
|
| 13 |
+
|
| 14 |
+
import os
|
| 15 |
+
import io
|
| 16 |
+
import base64
|
| 17 |
+
import hashlib
|
| 18 |
+
import asyncio
|
| 19 |
+
from datetime import datetime
|
| 20 |
+
from typing import Optional, Dict, Any, List
|
| 21 |
+
import logging
|
| 22 |
+
|
| 23 |
+
import cv2
|
| 24 |
+
import numpy as np
|
| 25 |
+
import onnxruntime as ort
|
| 26 |
+
from PIL import Image
|
| 27 |
+
import yaml
|
| 28 |
+
import difflib
|
| 29 |
+
|
| 30 |
+
from fastapi import FastAPI, HTTPException, Depends, status
|
| 31 |
+
from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
|
| 32 |
+
from fastapi.middleware.cors import CORSMiddleware
|
| 33 |
+
from pydantic import BaseModel, Field
|
| 34 |
+
import uvicorn
|
| 35 |
+
|
| 36 |
+
# Configure logging
|
| 37 |
+
logging.basicConfig(level=logging.INFO)
|
| 38 |
+
logger = logging.getLogger(__name__)
|
| 39 |
+
|
| 40 |
+
# =================================================================
|
| 41 |
+
# CONFIGURATION & MODELS
|
| 42 |
+
# =================================================================
|
| 43 |
+
|
| 44 |
+
class FunCaptchaRequest(BaseModel):
|
| 45 |
+
"""Request model untuk FunCaptcha solving"""
|
| 46 |
+
challenge_type: str = Field(..., description="Type of challenge (pick_the, upright)")
|
| 47 |
+
image_b64: str = Field(..., description="Base64 encoded image")
|
| 48 |
+
target_label: Optional[str] = Field(None, description="Target label untuk pick_the challenges")
|
| 49 |
+
|
| 50 |
+
class FunCaptchaResponse(BaseModel):
|
| 51 |
+
"""Response model untuk FunCaptcha solving"""
|
| 52 |
+
status: str = Field(..., description="Status: success, not_found, error")
|
| 53 |
+
box: Optional[List[float]] = Field(None, description="Bounding box coordinates [x, y, w, h]")
|
| 54 |
+
button_index: Optional[int] = Field(None, description="Button index untuk upright challenges")
|
| 55 |
+
confidence: Optional[float] = Field(None, description="Detection confidence")
|
| 56 |
+
message: Optional[str] = Field(None, description="Additional message")
|
| 57 |
+
processing_time: Optional[float] = Field(None, description="Processing time in seconds")
|
| 58 |
+
|
| 59 |
+
# =================================================================
|
| 60 |
+
# AUTHENTICATION
|
| 61 |
+
# =================================================================
|
| 62 |
+
|
| 63 |
+
security = HTTPBearer()
|
| 64 |
+
|
| 65 |
+
def get_api_key_from_secrets() -> str:
|
| 66 |
+
"""Get API key dari Hugging Face Secrets"""
|
| 67 |
+
api_key = os.getenv("FUNCAPTCHA_API_KEY")
|
| 68 |
+
if not api_key:
|
| 69 |
+
logger.error("FUNCAPTCHA_API_KEY not found in environment variables")
|
| 70 |
+
raise ValueError("API key tidak ditemukan dalam HF Secrets")
|
| 71 |
+
return api_key
|
| 72 |
+
|
| 73 |
+
def verify_api_key(credentials: HTTPAuthorizationCredentials = Depends(security)) -> bool:
|
| 74 |
+
"""Verify API key dari request header"""
|
| 75 |
+
expected_key = get_api_key_from_secrets()
|
| 76 |
+
if credentials.credentials != expected_key:
|
| 77 |
+
raise HTTPException(
|
| 78 |
+
status_code=status.HTTP_401_UNAUTHORIZED,
|
| 79 |
+
detail="Invalid API key",
|
| 80 |
+
headers={"WWW-Authenticate": "Bearer"}
|
| 81 |
+
)
|
| 82 |
+
return True
|
| 83 |
+
|
| 84 |
+
# =================================================================
|
| 85 |
+
# MODEL CONFIGURATION & MANAGEMENT
|
| 86 |
+
# =================================================================
|
| 87 |
+
|
| 88 |
+
CONFIGS = {
|
| 89 |
+
'default': {
|
| 90 |
+
'model_path': 'best.onnx',
|
| 91 |
+
'yaml_path': 'data.yaml',
|
| 92 |
+
'input_size': 640,
|
| 93 |
+
'confidence_threshold': 0.4,
|
| 94 |
+
'nms_threshold': 0.2
|
| 95 |
+
},
|
| 96 |
+
'spiral_galaxy': {
|
| 97 |
+
'model_path': 'bestspiral.onnx',
|
| 98 |
+
'yaml_path': 'dataspiral.yaml',
|
| 99 |
+
'input_size': 416,
|
| 100 |
+
'confidence_threshold': 0.30,
|
| 101 |
+
'nms_threshold': 0.45
|
| 102 |
+
},
|
| 103 |
+
'upright': {
|
| 104 |
+
'model_path': 'best_Upright.onnx',
|
| 105 |
+
'yaml_path': 'data_upright.yaml',
|
| 106 |
+
'input_size': 640,
|
| 107 |
+
'confidence_threshold': 0.45,
|
| 108 |
+
'nms_threshold': 0.45
|
| 109 |
+
}
|
| 110 |
+
}
|
| 111 |
+
|
| 112 |
+
MODEL_ROUTING = [
|
| 113 |
+
(['spiral', 'galaxy'], 'spiral_galaxy')
|
| 114 |
+
]
|
| 115 |
+
|
| 116 |
+
# Global cache untuk models dan responses
|
| 117 |
+
LOADED_MODELS: Dict[str, Dict[str, Any]] = {}
|
| 118 |
+
RESPONSE_CACHE: Dict[str, Dict[str, Any]] = {}
|
| 119 |
+
CACHE_MAX_SIZE = 100
|
| 120 |
+
|
| 121 |
+
class ModelManager:
|
| 122 |
+
"""Manager untuk loading dan caching models"""
|
| 123 |
+
|
| 124 |
+
@staticmethod
|
| 125 |
+
async def get_model(config_key: str) -> Optional[Dict[str, Any]]:
|
| 126 |
+
"""Load model dengan caching untuk efficiency"""
|
| 127 |
+
if config_key not in LOADED_MODELS:
|
| 128 |
+
logger.info(f"Loading model: {config_key}")
|
| 129 |
+
|
| 130 |
+
try:
|
| 131 |
+
config = CONFIGS[config_key]
|
| 132 |
+
|
| 133 |
+
# Check if files exist
|
| 134 |
+
if not os.path.exists(config['model_path']):
|
| 135 |
+
logger.warning(f"Model file not found: {config['model_path']}")
|
| 136 |
+
return None
|
| 137 |
+
|
| 138 |
+
if not os.path.exists(config['yaml_path']):
|
| 139 |
+
logger.warning(f"YAML file not found: {config['yaml_path']}")
|
| 140 |
+
return None
|
| 141 |
+
|
| 142 |
+
# Load ONNX session dengan CPU optimization
|
| 143 |
+
providers = ['CPUExecutionProvider']
|
| 144 |
+
session_options = ort.SessionOptions()
|
| 145 |
+
session_options.intra_op_num_threads = 2 # Optimize untuk CPU
|
| 146 |
+
session_options.inter_op_num_threads = 2
|
| 147 |
+
session_options.execution_mode = ort.ExecutionMode.ORT_SEQUENTIAL
|
| 148 |
+
session_options.graph_optimization_level = ort.GraphOptimizationLevel.ORT_ENABLE_ALL
|
| 149 |
+
|
| 150 |
+
session = ort.InferenceSession(
|
| 151 |
+
config['model_path'],
|
| 152 |
+
providers=providers,
|
| 153 |
+
sess_options=session_options
|
| 154 |
+
)
|
| 155 |
+
|
| 156 |
+
# Load class names
|
| 157 |
+
with open(config['yaml_path'], 'r', encoding='utf-8') as file:
|
| 158 |
+
class_names = yaml.safe_load(file)['names']
|
| 159 |
+
|
| 160 |
+
LOADED_MODELS[config_key] = {
|
| 161 |
+
'session': session,
|
| 162 |
+
'class_names': class_names,
|
| 163 |
+
'input_name': session.get_inputs()[0].name,
|
| 164 |
+
'input_size': config['input_size'],
|
| 165 |
+
'confidence': config['confidence_threshold'],
|
| 166 |
+
'nms': config.get('nms_threshold', 0.45)
|
| 167 |
+
}
|
| 168 |
+
|
| 169 |
+
logger.info(f"β
Model loaded successfully: {config_key}")
|
| 170 |
+
|
| 171 |
+
except Exception as e:
|
| 172 |
+
logger.error(f"β Error loading model {config_key}: {e}")
|
| 173 |
+
return None
|
| 174 |
+
|
| 175 |
+
return LOADED_MODELS[config_key]
|
| 176 |
+
|
| 177 |
+
# =================================================================
|
| 178 |
+
# IMAGE PROCESSING & UTILITIES
|
| 179 |
+
# =================================================================
|
| 180 |
+
|
| 181 |
+
def preprocess_image(image_bytes: bytes, input_size: int) -> np.ndarray:
|
| 182 |
+
"""Preprocess image untuk ONNX inference dengan optimasi memory"""
|
| 183 |
+
image = Image.open(io.BytesIO(image_bytes)).convert('RGB')
|
| 184 |
+
image_np = np.array(image)
|
| 185 |
+
h, w, _ = image_np.shape
|
| 186 |
+
|
| 187 |
+
scale = min(input_size / w, input_size / h)
|
| 188 |
+
new_w, new_h = int(w * scale), int(h * scale)
|
| 189 |
+
|
| 190 |
+
resized_image = cv2.resize(image_np, (new_w, new_h))
|
| 191 |
+
padded_image = np.full((input_size, input_size, 3), 114, dtype=np.uint8)
|
| 192 |
+
|
| 193 |
+
# Calculate padding
|
| 194 |
+
y_offset = (input_size - new_h) // 2
|
| 195 |
+
x_offset = (input_size - new_w) // 2
|
| 196 |
+
|
| 197 |
+
padded_image[y_offset:y_offset + new_h, x_offset:x_offset + new_w, :] = resized_image
|
| 198 |
+
|
| 199 |
+
# Convert untuk ONNX
|
| 200 |
+
input_tensor = padded_image.astype(np.float32) / 255.0
|
| 201 |
+
input_tensor = np.transpose(input_tensor, (2, 0, 1))
|
| 202 |
+
input_tensor = np.expand_dims(input_tensor, axis=0)
|
| 203 |
+
|
| 204 |
+
return input_tensor
|
| 205 |
+
|
| 206 |
+
def fuzzy_match_label(target_label: str, class_names: List[str], threshold: float = 0.6) -> Optional[str]:
|
| 207 |
+
"""Fuzzy matching untuk label variations"""
|
| 208 |
+
target_normalized = target_label.lower().strip()
|
| 209 |
+
|
| 210 |
+
# Dictionary untuk common variations
|
| 211 |
+
label_variants = {
|
| 212 |
+
'ice cream': ['ice cream', 'icecream', 'ice'],
|
| 213 |
+
'hotdog': ['hot dog', 'hotdog', 'hot-dog'],
|
| 214 |
+
'hot dog': ['hot dog', 'hotdog', 'hot-dog'],
|
| 215 |
+
'sunglasses': ['sunglasses', 'sun glasses', 'sunglass'],
|
| 216 |
+
'sun glasses': ['sunglasses', 'sun glasses', 'sunglass']
|
| 217 |
+
}
|
| 218 |
+
|
| 219 |
+
# 1. Exact match
|
| 220 |
+
if target_normalized in class_names:
|
| 221 |
+
return target_normalized
|
| 222 |
+
|
| 223 |
+
# 2. Check known variants
|
| 224 |
+
for main_label, variants in label_variants.items():
|
| 225 |
+
if target_normalized in variants and main_label in class_names:
|
| 226 |
+
return main_label
|
| 227 |
+
|
| 228 |
+
# 3. Fuzzy matching
|
| 229 |
+
best_matches = difflib.get_close_matches(
|
| 230 |
+
target_normalized,
|
| 231 |
+
[name.lower() for name in class_names],
|
| 232 |
+
n=3,
|
| 233 |
+
cutoff=threshold
|
| 234 |
+
)
|
| 235 |
+
|
| 236 |
+
if best_matches:
|
| 237 |
+
for match in best_matches:
|
| 238 |
+
for class_name in class_names:
|
| 239 |
+
if class_name.lower() == match:
|
| 240 |
+
return class_name
|
| 241 |
+
|
| 242 |
+
# 4. Partial matching
|
| 243 |
+
for class_name in class_names:
|
| 244 |
+
if target_normalized in class_name.lower() or class_name.lower() in target_normalized:
|
| 245 |
+
return class_name
|
| 246 |
+
|
| 247 |
+
return None
|
| 248 |
+
|
| 249 |
+
def get_config_key_for_label(target_label: str) -> str:
|
| 250 |
+
"""Determine which model config to use"""
|
| 251 |
+
for keywords, config_key in MODEL_ROUTING:
|
| 252 |
+
if any(keyword in target_label for keyword in keywords):
|
| 253 |
+
return config_key
|
| 254 |
+
return 'default'
|
| 255 |
+
|
| 256 |
+
def get_button_index(x_center: float, y_center: float, img_width: int, img_height: int,
|
| 257 |
+
grid_cols: int = 3, grid_rows: int = 2) -> int:
|
| 258 |
+
"""Calculate button index dari coordinates"""
|
| 259 |
+
col = int(x_center // (img_width / grid_cols))
|
| 260 |
+
row = int(y_center // (img_height / grid_rows))
|
| 261 |
+
return row * grid_cols + col + 1
|
| 262 |
+
|
| 263 |
+
# =================================================================
|
| 264 |
+
# CACHING SYSTEM
|
| 265 |
+
# =================================================================
|
| 266 |
+
|
| 267 |
+
def get_cache_key(request_data: dict) -> str:
|
| 268 |
+
"""Generate cache key dari request data"""
|
| 269 |
+
cache_string = f"{request_data.get('challenge_type')}_{request_data.get('target_label')}_{request_data.get('image_b64', '')[:100]}"
|
| 270 |
+
return hashlib.md5(cache_string.encode()).hexdigest()
|
| 271 |
+
|
| 272 |
+
def get_cached_response(cache_key: str) -> Optional[dict]:
|
| 273 |
+
"""Get response dari cache jika ada"""
|
| 274 |
+
return RESPONSE_CACHE.get(cache_key)
|
| 275 |
+
|
| 276 |
+
def cache_response(cache_key: str, response: dict):
|
| 277 |
+
"""Cache response dengan size limit"""
|
| 278 |
+
if len(RESPONSE_CACHE) >= CACHE_MAX_SIZE:
|
| 279 |
+
# Remove oldest entry
|
| 280 |
+
oldest_key = next(iter(RESPONSE_CACHE))
|
| 281 |
+
del RESPONSE_CACHE[oldest_key]
|
| 282 |
+
|
| 283 |
+
RESPONSE_CACHE[cache_key] = response
|
| 284 |
+
|
| 285 |
+
# =================================================================
|
| 286 |
+
# CHALLENGE HANDLERS
|
| 287 |
+
# =================================================================
|
| 288 |
+
|
| 289 |
+
async def handle_pick_the_challenge(data: dict) -> dict:
|
| 290 |
+
"""Handle 'pick the' challenges dengan fuzzy matching"""
|
| 291 |
+
start_time = datetime.now()
|
| 292 |
+
|
| 293 |
+
target_label_original = data['target_label']
|
| 294 |
+
image_b64 = data['image_b64']
|
| 295 |
+
target_label = target_label_original
|
| 296 |
+
|
| 297 |
+
config_key = get_config_key_for_label(target_label)
|
| 298 |
+
|
| 299 |
+
if config_key == 'spiral_galaxy':
|
| 300 |
+
target_label = 'spiral'
|
| 301 |
+
|
| 302 |
+
model_data = await ModelManager.get_model(config_key)
|
| 303 |
+
if not model_data:
|
| 304 |
+
return {'status': 'error', 'message': f'Model {config_key} tidak ditemukan'}
|
| 305 |
+
|
| 306 |
+
try:
|
| 307 |
+
# Decode image
|
| 308 |
+
image_bytes = base64.b64decode(image_b64.split(',')[1])
|
| 309 |
+
|
| 310 |
+
# Fuzzy matching untuk label
|
| 311 |
+
matched_label = fuzzy_match_label(target_label, model_data['class_names'])
|
| 312 |
+
if not matched_label:
|
| 313 |
+
return {
|
| 314 |
+
'status': 'not_found',
|
| 315 |
+
'message': f'Label "{target_label}" tidak ditemukan dalam model',
|
| 316 |
+
'processing_time': (datetime.now() - start_time).total_seconds()
|
| 317 |
+
}
|
| 318 |
+
|
| 319 |
+
target_label = matched_label
|
| 320 |
+
|
| 321 |
+
# Preprocessing
|
| 322 |
+
input_tensor = preprocess_image(image_bytes, model_data['input_size'])
|
| 323 |
+
|
| 324 |
+
# Inference
|
| 325 |
+
outputs = model_data['session'].run(None, {model_data['input_name']: input_tensor})[0]
|
| 326 |
+
predictions = np.squeeze(outputs).T
|
| 327 |
+
|
| 328 |
+
# Process detections
|
| 329 |
+
boxes = []
|
| 330 |
+
confidences = []
|
| 331 |
+
class_ids = []
|
| 332 |
+
|
| 333 |
+
for pred in predictions:
|
| 334 |
+
class_scores = pred[4:]
|
| 335 |
+
class_id = np.argmax(class_scores)
|
| 336 |
+
max_confidence = class_scores[class_id]
|
| 337 |
+
|
| 338 |
+
if max_confidence > model_data['confidence']:
|
| 339 |
+
confidences.append(float(max_confidence))
|
| 340 |
+
class_ids.append(class_id)
|
| 341 |
+
box_model = pred[:4]
|
| 342 |
+
x_center, y_center, width, height = box_model
|
| 343 |
+
x1 = x_center - width / 2
|
| 344 |
+
y1 = y_center - height / 2
|
| 345 |
+
boxes.append([int(x1), int(y1), int(width), int(height)])
|
| 346 |
+
|
| 347 |
+
if not boxes:
|
| 348 |
+
return {
|
| 349 |
+
'status': 'not_found',
|
| 350 |
+
'processing_time': (datetime.now() - start_time).total_seconds()
|
| 351 |
+
}
|
| 352 |
+
|
| 353 |
+
# Non-Maximum Suppression
|
| 354 |
+
indices = cv2.dnn.NMSBoxes(
|
| 355 |
+
np.array(boxes),
|
| 356 |
+
np.array(confidences),
|
| 357 |
+
model_data['confidence'],
|
| 358 |
+
model_data['nms']
|
| 359 |
+
)
|
| 360 |
+
|
| 361 |
+
if len(indices) == 0:
|
| 362 |
+
return {
|
| 363 |
+
'status': 'not_found',
|
| 364 |
+
'processing_time': (datetime.now() - start_time).total_seconds()
|
| 365 |
+
}
|
| 366 |
+
|
| 367 |
+
# Find target
|
| 368 |
+
target_class_id = model_data['class_names'].index(target_label)
|
| 369 |
+
best_match_box = None
|
| 370 |
+
highest_score = 0
|
| 371 |
+
|
| 372 |
+
for i in indices.flatten():
|
| 373 |
+
if class_ids[i] == target_class_id:
|
| 374 |
+
current_score = confidences[i]
|
| 375 |
+
if current_score > highest_score:
|
| 376 |
+
highest_score = current_score
|
| 377 |
+
best_match_box = boxes[i]
|
| 378 |
+
|
| 379 |
+
if best_match_box is not None:
|
| 380 |
+
# Scale back to original coordinates
|
| 381 |
+
img = Image.open(io.BytesIO(image_bytes))
|
| 382 |
+
original_w, original_h = img.size
|
| 383 |
+
scale = min(model_data['input_size'] / original_w, model_data['input_size'] / original_h)
|
| 384 |
+
pad_x = (model_data['input_size'] - original_w * scale) / 2
|
| 385 |
+
pad_y = (model_data['input_size'] - original_h * scale) / 2
|
| 386 |
+
|
| 387 |
+
x_orig = (best_match_box[0] - pad_x) / scale
|
| 388 |
+
y_orig = (best_match_box[1] - pad_y) / scale
|
| 389 |
+
w_orig = best_match_box[2] / scale
|
| 390 |
+
h_orig = best_match_box[3] / scale
|
| 391 |
+
|
| 392 |
+
return {
|
| 393 |
+
'status': 'success',
|
| 394 |
+
'box': [x_orig, y_orig, w_orig, h_orig],
|
| 395 |
+
'confidence': highest_score,
|
| 396 |
+
'processing_time': (datetime.now() - start_time).total_seconds()
|
| 397 |
+
}
|
| 398 |
+
|
| 399 |
+
except Exception as e:
|
| 400 |
+
logger.error(f"Error in handle_pick_the_challenge: {e}")
|
| 401 |
+
return {
|
| 402 |
+
'status': 'error',
|
| 403 |
+
'message': str(e),
|
| 404 |
+
'processing_time': (datetime.now() - start_time).total_seconds()
|
| 405 |
+
}
|
| 406 |
+
|
| 407 |
+
return {
|
| 408 |
+
'status': 'not_found',
|
| 409 |
+
'processing_time': (datetime.now() - start_time).total_seconds()
|
| 410 |
+
}
|
| 411 |
+
|
| 412 |
+
async def handle_upright_challenge(data: dict) -> dict:
|
| 413 |
+
"""Handle 'upright' challenges"""
|
| 414 |
+
start_time = datetime.now()
|
| 415 |
+
|
| 416 |
+
try:
|
| 417 |
+
image_b64 = data['image_b64']
|
| 418 |
+
model_data = await ModelManager.get_model('upright')
|
| 419 |
+
|
| 420 |
+
if not model_data:
|
| 421 |
+
return {'status': 'error', 'message': 'Model upright tidak ditemukan'}
|
| 422 |
+
|
| 423 |
+
image_bytes = base64.b64decode(image_b64.split(',')[1])
|
| 424 |
+
reconstructed_image_pil = Image.open(io.BytesIO(image_bytes))
|
| 425 |
+
original_w, original_h = reconstructed_image_pil.size
|
| 426 |
+
|
| 427 |
+
input_tensor = preprocess_image(image_bytes, model_data['input_size'])
|
| 428 |
+
outputs = model_data['session'].run(None, {model_data['input_name']: input_tensor})[0]
|
| 429 |
+
|
| 430 |
+
predictions = np.squeeze(outputs).T
|
| 431 |
+
confident_preds = predictions[predictions[:, 4] > model_data['confidence']]
|
| 432 |
+
|
| 433 |
+
if len(confident_preds) == 0:
|
| 434 |
+
return {
|
| 435 |
+
'status': 'not_found',
|
| 436 |
+
'message': 'Tidak ada objek terdeteksi',
|
| 437 |
+
'processing_time': (datetime.now() - start_time).total_seconds()
|
| 438 |
+
}
|
| 439 |
+
|
| 440 |
+
best_detection = confident_preds[np.argmax(confident_preds[:, 4])]
|
| 441 |
+
box_model = best_detection[:4]
|
| 442 |
+
|
| 443 |
+
scale = min(model_data['input_size'] / original_w, model_data['input_size'] / original_h)
|
| 444 |
+
pad_x = (model_data['input_size'] - original_w * scale) / 2
|
| 445 |
+
pad_y = (model_data['input_size'] - original_h * scale) / 2
|
| 446 |
+
|
| 447 |
+
x_center_orig = (box_model[0] - pad_x) / scale
|
| 448 |
+
y_center_orig = (box_model[1] - pad_y) / scale
|
| 449 |
+
|
| 450 |
+
button_to_click = get_button_index(x_center_orig, y_center_orig, original_w, original_h)
|
| 451 |
+
|
| 452 |
+
return {
|
| 453 |
+
'status': 'success',
|
| 454 |
+
'button_index': button_to_click,
|
| 455 |
+
'confidence': float(best_detection[4]),
|
| 456 |
+
'processing_time': (datetime.now() - start_time).total_seconds()
|
| 457 |
+
}
|
| 458 |
+
|
| 459 |
+
except Exception as e:
|
| 460 |
+
logger.error(f"Error in handle_upright_challenge: {e}")
|
| 461 |
+
return {
|
| 462 |
+
'status': 'error',
|
| 463 |
+
'message': str(e),
|
| 464 |
+
'processing_time': (datetime.now() - start_time).total_seconds()
|
| 465 |
+
}
|
| 466 |
+
|
| 467 |
+
# =================================================================
|
| 468 |
+
# FASTAPI APPLICATION
|
| 469 |
+
# =================================================================
|
| 470 |
+
|
| 471 |
+
app = FastAPI(
|
| 472 |
+
title="π§© FunCaptcha Solver API",
|
| 473 |
+
description="High-performance FunCaptcha solver dengan fuzzy matching untuk Hugging Face Spaces",
|
| 474 |
+
version="1.0.0",
|
| 475 |
+
docs_url="/docs",
|
| 476 |
+
redoc_url="/redoc"
|
| 477 |
+
)
|
| 478 |
+
|
| 479 |
+
# CORS middleware
|
| 480 |
+
app.add_middleware(
|
| 481 |
+
CORSMiddleware,
|
| 482 |
+
allow_origins=["*"],
|
| 483 |
+
allow_credentials=True,
|
| 484 |
+
allow_methods=["*"],
|
| 485 |
+
allow_headers=["*"],
|
| 486 |
+
)
|
| 487 |
+
|
| 488 |
+
@app.get("/")
|
| 489 |
+
async def root():
|
| 490 |
+
"""Root endpoint dengan info API"""
|
| 491 |
+
return {
|
| 492 |
+
"service": "FunCaptcha Solver API",
|
| 493 |
+
"version": "1.0.0",
|
| 494 |
+
"status": "running",
|
| 495 |
+
"endpoints": {
|
| 496 |
+
"/solve": "POST - Solve FunCaptcha challenges",
|
| 497 |
+
"/health": "GET - Health check",
|
| 498 |
+
"/docs": "GET - API documentation"
|
| 499 |
+
},
|
| 500 |
+
"models_loaded": len(LOADED_MODELS),
|
| 501 |
+
"cache_size": len(RESPONSE_CACHE)
|
| 502 |
+
}
|
| 503 |
+
|
| 504 |
+
@app.get("/health")
|
| 505 |
+
async def health_check():
|
| 506 |
+
"""Health check endpoint"""
|
| 507 |
+
return {
|
| 508 |
+
"status": "healthy",
|
| 509 |
+
"service": "FunCaptcha Solver",
|
| 510 |
+
"models_loaded": len(LOADED_MODELS),
|
| 511 |
+
"available_models": list(CONFIGS.keys()),
|
| 512 |
+
"cache_entries": len(RESPONSE_CACHE)
|
| 513 |
+
}
|
| 514 |
+
|
| 515 |
+
@app.post("/solve", response_model=FunCaptchaResponse)
|
| 516 |
+
async def solve_funcaptcha(
|
| 517 |
+
request: FunCaptchaRequest,
|
| 518 |
+
authenticated: bool = Depends(verify_api_key)
|
| 519 |
+
) -> FunCaptchaResponse:
|
| 520 |
+
"""
|
| 521 |
+
π§© Solve FunCaptcha challenges
|
| 522 |
+
|
| 523 |
+
Supports:
|
| 524 |
+
- pick_the: Pick specific objects dari images
|
| 525 |
+
- upright: Find correctly oriented objects
|
| 526 |
+
|
| 527 |
+
Features:
|
| 528 |
+
- Fuzzy label matching
|
| 529 |
+
- Response caching
|
| 530 |
+
- Multi-model support
|
| 531 |
+
"""
|
| 532 |
+
|
| 533 |
+
# Generate cache key
|
| 534 |
+
request_dict = request.dict()
|
| 535 |
+
cache_key = get_cache_key(request_dict)
|
| 536 |
+
|
| 537 |
+
# Check cache first
|
| 538 |
+
cached_response = get_cached_response(cache_key)
|
| 539 |
+
if cached_response:
|
| 540 |
+
logger.info(f"Cache hit for challenge: {request.challenge_type}")
|
| 541 |
+
return FunCaptchaResponse(**cached_response)
|
| 542 |
+
|
| 543 |
+
# Process request
|
| 544 |
+
if request.challenge_type == 'pick_the':
|
| 545 |
+
if not request.target_label:
|
| 546 |
+
raise HTTPException(status_code=400, detail="target_label required for pick_the challenges")
|
| 547 |
+
result = await handle_pick_the_challenge(request_dict)
|
| 548 |
+
elif request.challenge_type == 'upright':
|
| 549 |
+
result = await handle_upright_challenge(request_dict)
|
| 550 |
+
else:
|
| 551 |
+
raise HTTPException(status_code=400, detail=f"Unsupported challenge type: {request.challenge_type}")
|
| 552 |
+
|
| 553 |
+
# Cache response
|
| 554 |
+
cache_response(cache_key, result)
|
| 555 |
+
|
| 556 |
+
logger.info(f"Challenge solved: {request.challenge_type} -> {result['status']}")
|
| 557 |
+
|
| 558 |
+
return FunCaptchaResponse(**result)
|
| 559 |
+
|
| 560 |
+
# =================================================================
|
| 561 |
+
# APPLICATION STARTUP
|
| 562 |
+
# =================================================================
|
| 563 |
+
|
| 564 |
+
@app.on_event("startup")
|
| 565 |
+
async def startup_event():
|
| 566 |
+
"""Initialize aplikasi saat startup"""
|
| 567 |
+
logger.info("π Starting FunCaptcha Solver API...")
|
| 568 |
+
|
| 569 |
+
# Verify API key ada
|
| 570 |
+
try:
|
| 571 |
+
api_key = get_api_key_from_secrets()
|
| 572 |
+
logger.info("β
API key loaded successfully")
|
| 573 |
+
except ValueError as e:
|
| 574 |
+
logger.error(f"β API key error: {e}")
|
| 575 |
+
raise e
|
| 576 |
+
|
| 577 |
+
# Preload default model jika ada
|
| 578 |
+
if os.path.exists('best.onnx') and os.path.exists('data.yaml'):
|
| 579 |
+
logger.info("Preloading default model...")
|
| 580 |
+
await ModelManager.get_model('default')
|
| 581 |
+
|
| 582 |
+
logger.info("β
FunCaptcha Solver API started successfully")
|
| 583 |
+
|
| 584 |
+
@app.on_event("shutdown")
|
| 585 |
+
async def shutdown_event():
|
| 586 |
+
"""Cleanup saat shutdown"""
|
| 587 |
+
logger.info("π Shutting down FunCaptcha Solver API...")
|
| 588 |
+
|
| 589 |
+
# Clear caches
|
| 590 |
+
LOADED_MODELS.clear()
|
| 591 |
+
RESPONSE_CACHE.clear()
|
| 592 |
+
|
| 593 |
+
logger.info("β
Cleanup completed")
|
| 594 |
+
|
| 595 |
+
# =================================================================
|
| 596 |
+
# DEVELOPMENT SERVER
|
| 597 |
+
# =================================================================
|
| 598 |
+
|
| 599 |
+
if __name__ == "__main__":
|
| 600 |
+
uvicorn.run(
|
| 601 |
+
"app:app",
|
| 602 |
+
host="0.0.0.0",
|
| 603 |
+
port=7860,
|
| 604 |
+
reload=False, # Disabled untuk production
|
| 605 |
+
workers=1 # Single worker untuk HF Spaces
|
| 606 |
+
)
|
best.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:39cdc33aa98866623a8cf3ca0e1a6d820c851d69ca428958e04f2a0cce233197
|
| 3 |
+
size 44825355
|
best_upright.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:af3c2c59ecd8ccd63b0b57cb0880099d06ff4c2d2329423189652f64ea2df894
|
| 3 |
+
size 12183823
|
bestspiral.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7e4b5cc002bf1fbfd51f41b9dcbc65933fc00cfa44bea5313566df4c6d41c3f1
|
| 3 |
+
size 44606239
|
data.yaml
ADDED
|
@@ -0,0 +1,115 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
names:
|
| 2 |
+
- alien
|
| 3 |
+
- anchor
|
| 4 |
+
- ant
|
| 5 |
+
- apple
|
| 6 |
+
- aquarium
|
| 7 |
+
- ball
|
| 8 |
+
- banana
|
| 9 |
+
- bat
|
| 10 |
+
- bear
|
| 11 |
+
- bee
|
| 12 |
+
- bike
|
| 13 |
+
- boat
|
| 14 |
+
- bread
|
| 15 |
+
- burger
|
| 16 |
+
- butterfly
|
| 17 |
+
- cactus
|
| 18 |
+
- calculator
|
| 19 |
+
- camel
|
| 20 |
+
- camera
|
| 21 |
+
- car
|
| 22 |
+
- cat
|
| 23 |
+
- cheese
|
| 24 |
+
- chicken
|
| 25 |
+
- computer
|
| 26 |
+
- controller
|
| 27 |
+
- couch
|
| 28 |
+
- cow
|
| 29 |
+
- crab
|
| 30 |
+
- crown
|
| 31 |
+
- deer
|
| 32 |
+
- dinosaur
|
| 33 |
+
- dog
|
| 34 |
+
- dolphin
|
| 35 |
+
- donut
|
| 36 |
+
- duck
|
| 37 |
+
- elephant
|
| 38 |
+
- fan
|
| 39 |
+
- fax
|
| 40 |
+
- fence
|
| 41 |
+
- fire
|
| 42 |
+
- flower
|
| 43 |
+
- fridge
|
| 44 |
+
- frog
|
| 45 |
+
- gazelle
|
| 46 |
+
- giraffe
|
| 47 |
+
- goat
|
| 48 |
+
- grapes
|
| 49 |
+
- guitar
|
| 50 |
+
- helicopter
|
| 51 |
+
- helmet
|
| 52 |
+
- horse
|
| 53 |
+
- hotdog
|
| 54 |
+
- ice cream
|
| 55 |
+
- kangaroo
|
| 56 |
+
- key
|
| 57 |
+
- koala
|
| 58 |
+
- ladybug
|
| 59 |
+
- lamp
|
| 60 |
+
- lion
|
| 61 |
+
- lobster
|
| 62 |
+
- lock
|
| 63 |
+
- money
|
| 64 |
+
- monkey
|
| 65 |
+
- mouse
|
| 66 |
+
- mushroom
|
| 67 |
+
- octopus
|
| 68 |
+
- owl
|
| 69 |
+
- panda
|
| 70 |
+
- pants
|
| 71 |
+
- parrot
|
| 72 |
+
- pencil
|
| 73 |
+
- pig
|
| 74 |
+
- pineapple
|
| 75 |
+
- pizza
|
| 76 |
+
- plane
|
| 77 |
+
- printer
|
| 78 |
+
- rabbit
|
| 79 |
+
- ram
|
| 80 |
+
- rhino
|
| 81 |
+
- ring
|
| 82 |
+
- scissors
|
| 83 |
+
- seal
|
| 84 |
+
- shark
|
| 85 |
+
- sheep
|
| 86 |
+
- shirt
|
| 87 |
+
- shoe
|
| 88 |
+
- snail
|
| 89 |
+
- snake
|
| 90 |
+
- snowman
|
| 91 |
+
- sock
|
| 92 |
+
- spaceship
|
| 93 |
+
- stapler
|
| 94 |
+
- starfish
|
| 95 |
+
- sunglasses
|
| 96 |
+
- toaster
|
| 97 |
+
- toater
|
| 98 |
+
- toilet
|
| 99 |
+
- train
|
| 100 |
+
- trophy
|
| 101 |
+
- turtle
|
| 102 |
+
- umbrella
|
| 103 |
+
- watch
|
| 104 |
+
- watermelon
|
| 105 |
+
- zebra
|
| 106 |
+
nc: 104
|
| 107 |
+
roboflow:
|
| 108 |
+
license: CC BY 4.0
|
| 109 |
+
project: kaptcha
|
| 110 |
+
url: https://universe.roboflow.com/krispcode-agr/kaptcha/dataset/2
|
| 111 |
+
version: 2
|
| 112 |
+
workspace: krispcode-agr
|
| 113 |
+
test: ../test/images
|
| 114 |
+
train: ../train/images
|
| 115 |
+
val: ../valid/images
|
data_upright.yaml
ADDED
|
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
names:
|
| 2 |
+
- upright
|
| 3 |
+
nc: 1
|
| 4 |
+
roboflow:
|
| 5 |
+
license: CC BY 4.0
|
| 6 |
+
project: funcaptcha
|
| 7 |
+
url: https://universe.roboflow.com/thebaconpug-gmail-com/funcaptcha/dataset/6
|
| 8 |
+
version: 6
|
| 9 |
+
workspace: thebaconpug-gmail-com
|
| 10 |
+
test: ../test/images
|
| 11 |
+
train: ../train/images
|
| 12 |
+
val: ../valid/images
|
dataspiral.yaml
ADDED
|
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
names:
|
| 2 |
+
- spiral
|
| 3 |
+
nc: 1
|
| 4 |
+
roboflow:
|
| 5 |
+
license: CC BY 4.0
|
| 6 |
+
project: sg-bhith
|
| 7 |
+
url: https://universe.roboflow.com/zoom-awet2/sg-bhith/dataset/1
|
| 8 |
+
version: 1
|
| 9 |
+
workspace: zoom-awet2
|
| 10 |
+
test: ../test/images
|
| 11 |
+
train: ../train/images
|
| 12 |
+
val: ../valid/images
|
requirements.txt
ADDED
|
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Optimized requirements for FunCaptcha HF Spaces Deployment
|
| 2 |
+
# Minimal dependencies untuk performa dan ukuran optimal
|
| 3 |
+
|
| 4 |
+
# Core web framework - Async FastAPI untuk performa tinggi
|
| 5 |
+
fastapi==0.104.1
|
| 6 |
+
uvicorn[standard]==0.24.0
|
| 7 |
+
|
| 8 |
+
# ML/AI dependencies - Versi stabil dan ringan
|
| 9 |
+
onnxruntime==1.16.3
|
| 10 |
+
opencv-python-headless==4.8.1.78
|
| 11 |
+
numpy==1.24.4
|
| 12 |
+
pillow==10.1.0
|
| 13 |
+
|
| 14 |
+
# Utility libraries - Minimal yang diperlukan
|
| 15 |
+
pyyaml==6.0.1
|
| 16 |
+
python-multipart==0.0.6
|
| 17 |
+
|
| 18 |
+
# Security & Authentication
|
| 19 |
+
python-jose[cryptography]==3.3.0
|
| 20 |
+
|
| 21 |
+
# Optional: Logging dan monitoring (sangat ringan)
|
| 22 |
+
structlog==23.2.0
|
test-api.py
ADDED
|
@@ -0,0 +1,343 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/usr/bin/env python3
|
| 2 |
+
"""
|
| 3 |
+
π§ͺ Test script untuk FunCaptcha Solver API
|
| 4 |
+
Test authentication, endpoints, dan response formats
|
| 5 |
+
"""
|
| 6 |
+
|
| 7 |
+
import os
|
| 8 |
+
import base64
|
| 9 |
+
import json
|
| 10 |
+
import requests
|
| 11 |
+
import time
|
| 12 |
+
from typing import Optional
|
| 13 |
+
import argparse
|
| 14 |
+
|
| 15 |
+
# Configuration
|
| 16 |
+
DEFAULT_LOCAL_URL = "http://localhost:7860"
|
| 17 |
+
DEFAULT_HF_URL = "https://your-space-name.hf.space"
|
| 18 |
+
DEFAULT_API_KEY = "test-key-123"
|
| 19 |
+
|
| 20 |
+
class FunCaptchaAPITester:
|
| 21 |
+
def __init__(self, base_url: str, api_key: str):
|
| 22 |
+
self.base_url = base_url.rstrip('/')
|
| 23 |
+
self.api_key = api_key
|
| 24 |
+
self.session = requests.Session()
|
| 25 |
+
|
| 26 |
+
def test_health_check(self) -> bool:
|
| 27 |
+
"""Test health check endpoint (no auth required)"""
|
| 28 |
+
print("π Testing health check endpoint...")
|
| 29 |
+
|
| 30 |
+
try:
|
| 31 |
+
response = self.session.get(f"{self.base_url}/health", timeout=10)
|
| 32 |
+
|
| 33 |
+
if response.status_code == 200:
|
| 34 |
+
data = response.json()
|
| 35 |
+
print(f"β
Health check passed")
|
| 36 |
+
print(f" Status: {data.get('status')}")
|
| 37 |
+
print(f" Models loaded: {data.get('models_loaded', 0)}")
|
| 38 |
+
print(f" Cache entries: {data.get('cache_entries', 0)}")
|
| 39 |
+
return True
|
| 40 |
+
else:
|
| 41 |
+
print(f"β Health check failed: {response.status_code}")
|
| 42 |
+
print(f" Response: {response.text}")
|
| 43 |
+
return False
|
| 44 |
+
|
| 45 |
+
except Exception as e:
|
| 46 |
+
print(f"β Health check error: {e}")
|
| 47 |
+
return False
|
| 48 |
+
|
| 49 |
+
def test_root_endpoint(self) -> bool:
|
| 50 |
+
"""Test root endpoint (no auth required)"""
|
| 51 |
+
print("π Testing root endpoint...")
|
| 52 |
+
|
| 53 |
+
try:
|
| 54 |
+
response = self.session.get(f"{self.base_url}/", timeout=10)
|
| 55 |
+
|
| 56 |
+
if response.status_code == 200:
|
| 57 |
+
data = response.json()
|
| 58 |
+
print(f"β
Root endpoint accessible")
|
| 59 |
+
print(f" Service: {data.get('service')}")
|
| 60 |
+
print(f" Version: {data.get('version')}")
|
| 61 |
+
return True
|
| 62 |
+
else:
|
| 63 |
+
print(f"β Root endpoint failed: {response.status_code}")
|
| 64 |
+
return False
|
| 65 |
+
|
| 66 |
+
except Exception as e:
|
| 67 |
+
print(f"β Root endpoint error: {e}")
|
| 68 |
+
return False
|
| 69 |
+
|
| 70 |
+
def test_authentication(self) -> bool:
|
| 71 |
+
"""Test API authentication"""
|
| 72 |
+
print("π Testing API authentication...")
|
| 73 |
+
|
| 74 |
+
# Test without API key
|
| 75 |
+
try:
|
| 76 |
+
response = self.session.post(f"{self.base_url}/solve", timeout=10, json={
|
| 77 |
+
"challenge_type": "pick_the",
|
| 78 |
+
"image_b64": "data:image/png;base64,test",
|
| 79 |
+
"target_label": "test"
|
| 80 |
+
})
|
| 81 |
+
|
| 82 |
+
if response.status_code == 401:
|
| 83 |
+
print("β
Authentication correctly rejects requests without API key")
|
| 84 |
+
else:
|
| 85 |
+
print(f"β οΈ Expected 401 for missing auth, got {response.status_code}")
|
| 86 |
+
|
| 87 |
+
except Exception as e:
|
| 88 |
+
print(f"β Auth test error: {e}")
|
| 89 |
+
return False
|
| 90 |
+
|
| 91 |
+
# Test with wrong API key
|
| 92 |
+
try:
|
| 93 |
+
headers = {"Authorization": "Bearer wrong-key"}
|
| 94 |
+
response = self.session.post(f"{self.base_url}/solve",
|
| 95 |
+
headers=headers, timeout=10, json={
|
| 96 |
+
"challenge_type": "pick_the",
|
| 97 |
+
"image_b64": "data:image/png;base64,test",
|
| 98 |
+
"target_label": "test"
|
| 99 |
+
})
|
| 100 |
+
|
| 101 |
+
if response.status_code == 401:
|
| 102 |
+
print("β
Authentication correctly rejects wrong API key")
|
| 103 |
+
return True
|
| 104 |
+
else:
|
| 105 |
+
print(f"β οΈ Expected 401 for wrong key, got {response.status_code}")
|
| 106 |
+
return False
|
| 107 |
+
|
| 108 |
+
except Exception as e:
|
| 109 |
+
print(f"β Wrong auth test error: {e}")
|
| 110 |
+
return False
|
| 111 |
+
|
| 112 |
+
def create_test_image_b64(self) -> str:
|
| 113 |
+
"""Create a simple test image in base64 format"""
|
| 114 |
+
# Simple 1x1 pixel PNG in base64
|
| 115 |
+
# This is a minimal valid PNG image
|
| 116 |
+
png_b64 = "iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNkYPhfDwAChwGA60e6kgAAAABJRU5ErkJggg=="
|
| 117 |
+
return f"data:image/png;base64,{png_b64}"
|
| 118 |
+
|
| 119 |
+
def test_solve_endpoint_pick_the(self) -> bool:
|
| 120 |
+
"""Test solve endpoint dengan pick_the challenge"""
|
| 121 |
+
print("π Testing solve endpoint (pick_the)...")
|
| 122 |
+
|
| 123 |
+
try:
|
| 124 |
+
headers = {"Authorization": f"Bearer {self.api_key}"}
|
| 125 |
+
test_image = self.create_test_image_b64()
|
| 126 |
+
|
| 127 |
+
payload = {
|
| 128 |
+
"challenge_type": "pick_the",
|
| 129 |
+
"image_b64": test_image,
|
| 130 |
+
"target_label": "ice cream"
|
| 131 |
+
}
|
| 132 |
+
|
| 133 |
+
start_time = time.time()
|
| 134 |
+
response = self.session.post(f"{self.base_url}/solve",
|
| 135 |
+
headers=headers,
|
| 136 |
+
json=payload,
|
| 137 |
+
timeout=30)
|
| 138 |
+
response_time = time.time() - start_time
|
| 139 |
+
|
| 140 |
+
print(f" Response time: {response_time:.2f}s")
|
| 141 |
+
print(f" Status code: {response.status_code}")
|
| 142 |
+
|
| 143 |
+
if response.status_code == 200:
|
| 144 |
+
data = response.json()
|
| 145 |
+
print("β
Solve endpoint accessible with valid auth")
|
| 146 |
+
print(f" Status: {data.get('status')}")
|
| 147 |
+
print(f" Processing time: {data.get('processing_time', 0):.3f}s")
|
| 148 |
+
|
| 149 |
+
if 'box' in data:
|
| 150 |
+
print(f" Box coordinates: {data['box']}")
|
| 151 |
+
if 'confidence' in data:
|
| 152 |
+
print(f" Confidence: {data['confidence']:.3f}")
|
| 153 |
+
|
| 154 |
+
return True
|
| 155 |
+
else:
|
| 156 |
+
print(f"β Solve endpoint failed: {response.status_code}")
|
| 157 |
+
print(f" Response: {response.text}")
|
| 158 |
+
return False
|
| 159 |
+
|
| 160 |
+
except Exception as e:
|
| 161 |
+
print(f"β Solve endpoint error: {e}")
|
| 162 |
+
return False
|
| 163 |
+
|
| 164 |
+
def test_solve_endpoint_upright(self) -> bool:
|
| 165 |
+
"""Test solve endpoint dengan upright challenge"""
|
| 166 |
+
print("π Testing solve endpoint (upright)...")
|
| 167 |
+
|
| 168 |
+
try:
|
| 169 |
+
headers = {"Authorization": f"Bearer {self.api_key}"}
|
| 170 |
+
test_image = self.create_test_image_b64()
|
| 171 |
+
|
| 172 |
+
payload = {
|
| 173 |
+
"challenge_type": "upright",
|
| 174 |
+
"image_b64": test_image
|
| 175 |
+
}
|
| 176 |
+
|
| 177 |
+
start_time = time.time()
|
| 178 |
+
response = self.session.post(f"{self.base_url}/solve",
|
| 179 |
+
headers=headers,
|
| 180 |
+
json=payload,
|
| 181 |
+
timeout=30)
|
| 182 |
+
response_time = time.time() - start_time
|
| 183 |
+
|
| 184 |
+
print(f" Response time: {response_time:.2f}s")
|
| 185 |
+
print(f" Status code: {response.status_code}")
|
| 186 |
+
|
| 187 |
+
if response.status_code == 200:
|
| 188 |
+
data = response.json()
|
| 189 |
+
print("β
Upright solve endpoint works")
|
| 190 |
+
print(f" Status: {data.get('status')}")
|
| 191 |
+
print(f" Processing time: {data.get('processing_time', 0):.3f}s")
|
| 192 |
+
|
| 193 |
+
if 'button_index' in data:
|
| 194 |
+
print(f" Button index: {data['button_index']}")
|
| 195 |
+
if 'confidence' in data:
|
| 196 |
+
print(f" Confidence: {data['confidence']:.3f}")
|
| 197 |
+
|
| 198 |
+
return True
|
| 199 |
+
else:
|
| 200 |
+
print(f"β Upright solve failed: {response.status_code}")
|
| 201 |
+
print(f" Response: {response.text}")
|
| 202 |
+
return False
|
| 203 |
+
|
| 204 |
+
except Exception as e:
|
| 205 |
+
print(f"β Upright solve error: {e}")
|
| 206 |
+
return False
|
| 207 |
+
|
| 208 |
+
def test_invalid_requests(self) -> bool:
|
| 209 |
+
"""Test invalid request handling"""
|
| 210 |
+
print("π Testing invalid request handling...")
|
| 211 |
+
|
| 212 |
+
headers = {"Authorization": f"Bearer {self.api_key}"}
|
| 213 |
+
|
| 214 |
+
# Test 1: Invalid challenge type
|
| 215 |
+
try:
|
| 216 |
+
response = self.session.post(f"{self.base_url}/solve",
|
| 217 |
+
headers=headers,
|
| 218 |
+
json={
|
| 219 |
+
"challenge_type": "invalid_type",
|
| 220 |
+
"image_b64": self.create_test_image_b64()
|
| 221 |
+
},
|
| 222 |
+
timeout=10)
|
| 223 |
+
|
| 224 |
+
if response.status_code == 400:
|
| 225 |
+
print("β
Correctly rejects invalid challenge type")
|
| 226 |
+
else:
|
| 227 |
+
print(f"β οΈ Expected 400 for invalid challenge type, got {response.status_code}")
|
| 228 |
+
|
| 229 |
+
except Exception as e:
|
| 230 |
+
print(f"β Invalid challenge type test error: {e}")
|
| 231 |
+
return False
|
| 232 |
+
|
| 233 |
+
# Test 2: Missing target_label for pick_the
|
| 234 |
+
try:
|
| 235 |
+
response = self.session.post(f"{self.base_url}/solve",
|
| 236 |
+
headers=headers,
|
| 237 |
+
json={
|
| 238 |
+
"challenge_type": "pick_the",
|
| 239 |
+
"image_b64": self.create_test_image_b64()
|
| 240 |
+
# Missing target_label
|
| 241 |
+
},
|
| 242 |
+
timeout=10)
|
| 243 |
+
|
| 244 |
+
if response.status_code == 400:
|
| 245 |
+
print("β
Correctly rejects pick_the without target_label")
|
| 246 |
+
else:
|
| 247 |
+
print(f"β οΈ Expected 400 for missing target_label, got {response.status_code}")
|
| 248 |
+
|
| 249 |
+
except Exception as e:
|
| 250 |
+
print(f"β Missing target_label test error: {e}")
|
| 251 |
+
return False
|
| 252 |
+
|
| 253 |
+
return True
|
| 254 |
+
|
| 255 |
+
def run_all_tests(self) -> bool:
|
| 256 |
+
"""Run all test cases"""
|
| 257 |
+
print("π Starting FunCaptcha API tests...")
|
| 258 |
+
print(f"π Base URL: {self.base_url}")
|
| 259 |
+
print(f"π API Key: {self.api_key[:8]}...")
|
| 260 |
+
print("=" * 60)
|
| 261 |
+
|
| 262 |
+
tests = [
|
| 263 |
+
("Health Check", self.test_health_check),
|
| 264 |
+
("Root Endpoint", self.test_root_endpoint),
|
| 265 |
+
("Authentication", self.test_authentication),
|
| 266 |
+
("Solve Pick-The", self.test_solve_endpoint_pick_the),
|
| 267 |
+
("Solve Upright", self.test_solve_endpoint_upright),
|
| 268 |
+
("Invalid Requests", self.test_invalid_requests)
|
| 269 |
+
]
|
| 270 |
+
|
| 271 |
+
results = []
|
| 272 |
+
for test_name, test_func in tests:
|
| 273 |
+
print(f"\\nπ Running test: {test_name}")
|
| 274 |
+
print("-" * 40)
|
| 275 |
+
try:
|
| 276 |
+
result = test_func()
|
| 277 |
+
results.append((test_name, result))
|
| 278 |
+
if result:
|
| 279 |
+
print(f"β
{test_name} PASSED")
|
| 280 |
+
else:
|
| 281 |
+
print(f"β {test_name} FAILED")
|
| 282 |
+
except Exception as e:
|
| 283 |
+
print(f"π₯ {test_name} CRASHED: {e}")
|
| 284 |
+
results.append((test_name, False))
|
| 285 |
+
|
| 286 |
+
time.sleep(1) # Brief pause between tests
|
| 287 |
+
|
| 288 |
+
# Summary
|
| 289 |
+
print("\\n" + "=" * 60)
|
| 290 |
+
print("π TEST SUMMARY")
|
| 291 |
+
print("=" * 60)
|
| 292 |
+
|
| 293 |
+
passed = sum(1 for _, result in results if result)
|
| 294 |
+
total = len(results)
|
| 295 |
+
|
| 296 |
+
for test_name, result in results:
|
| 297 |
+
status = "β
PASS" if result else "β FAIL"
|
| 298 |
+
print(f"{test_name:<20} {status}")
|
| 299 |
+
|
| 300 |
+
print(f"\\nπ― Overall: {passed}/{total} tests passed")
|
| 301 |
+
|
| 302 |
+
if passed == total:
|
| 303 |
+
print("π All tests PASSED! API is working correctly.")
|
| 304 |
+
return True
|
| 305 |
+
else:
|
| 306 |
+
print(f"β οΈ {total - passed} tests FAILED. Check configuration and deployment.")
|
| 307 |
+
return False
|
| 308 |
+
|
| 309 |
+
def main():
|
| 310 |
+
parser = argparse.ArgumentParser(description="Test FunCaptcha Solver API")
|
| 311 |
+
parser.add_argument("--url", default=DEFAULT_LOCAL_URL,
|
| 312 |
+
help=f"Base URL for API (default: {DEFAULT_LOCAL_URL})")
|
| 313 |
+
parser.add_argument("--api-key", default=DEFAULT_API_KEY,
|
| 314 |
+
help=f"API key for authentication (default: {DEFAULT_API_KEY})")
|
| 315 |
+
parser.add_argument("--hf", action="store_true",
|
| 316 |
+
help="Use default Hugging Face URL")
|
| 317 |
+
|
| 318 |
+
args = parser.parse_args()
|
| 319 |
+
|
| 320 |
+
# Use HF URL if --hf flag provided
|
| 321 |
+
if args.hf:
|
| 322 |
+
base_url = DEFAULT_HF_URL
|
| 323 |
+
print("π€ Using Hugging Face Spaces URL")
|
| 324 |
+
print("β οΈ Make sure to update DEFAULT_HF_URL in script dengan actual space URL!")
|
| 325 |
+
else:
|
| 326 |
+
base_url = args.url
|
| 327 |
+
|
| 328 |
+
# Check for API key in environment
|
| 329 |
+
api_key = os.getenv("FUNCAPTCHA_API_KEY", args.api_key)
|
| 330 |
+
|
| 331 |
+
if api_key == DEFAULT_API_KEY and args.hf:
|
| 332 |
+
print("β οΈ WARNING: Using default API key dengan HF Spaces!")
|
| 333 |
+
print(" Set FUNCAPTCHA_API_KEY environment variable atau use --api-key")
|
| 334 |
+
|
| 335 |
+
# Run tests
|
| 336 |
+
tester = FunCaptchaAPITester(base_url, api_key)
|
| 337 |
+
success = tester.run_all_tests()
|
| 338 |
+
|
| 339 |
+
# Exit with appropriate code
|
| 340 |
+
exit(0 if success else 1)
|
| 341 |
+
|
| 342 |
+
if __name__ == "__main__":
|
| 343 |
+
main()
|