6bb76e3f1b
- Implemented OCR functionality using pytesseract for image and PDF text extraction. - Added Background Removal service using rembg for image processing. - Developed PDF Editor service for applying text annotations to PDF files. - Created corresponding API routes for OCR, Background Removal, and PDF Editor. - Added frontend components for OCR and Background Removal tools. - Integrated feature flagging for new tools, ensuring they are disabled by default. - Implemented comprehensive unit tests for OCR service, PDF editor, and background removal. - Updated documentation to reflect new features and usage instructions. - Added translations for new features in English, Arabic, and French.
65 lines
1.9 KiB
Python
65 lines
1.9 KiB
Python
"""Background removal route."""
|
|
from flask import Blueprint, request, jsonify, current_app
|
|
|
|
from app.extensions import limiter
|
|
from app.services.policy_service import (
|
|
assert_quota_available,
|
|
build_task_tracking_kwargs,
|
|
PolicyError,
|
|
record_accepted_usage,
|
|
resolve_web_actor,
|
|
validate_actor_file,
|
|
)
|
|
from app.utils.file_validator import FileValidationError
|
|
from app.utils.sanitizer import generate_safe_path
|
|
from app.tasks.removebg_tasks import remove_bg_task
|
|
|
|
removebg_bp = Blueprint("removebg", __name__)
|
|
|
|
ALLOWED_IMAGE_TYPES = ["png", "jpg", "jpeg", "webp"]
|
|
|
|
|
|
@removebg_bp.route("", methods=["POST"])
|
|
@limiter.limit("5/minute")
|
|
def remove_bg_route():
|
|
"""Remove the background from an image.
|
|
|
|
Accepts: multipart/form-data with:
|
|
- 'file': Image file (PNG, JPG, JPEG, WebP)
|
|
Returns: JSON with task_id for polling
|
|
"""
|
|
if not current_app.config.get("FEATURE_EDITOR", False):
|
|
return jsonify({"error": "This feature is not enabled."}), 403
|
|
|
|
if "file" not in request.files:
|
|
return jsonify({"error": "No file provided."}), 400
|
|
|
|
file = request.files["file"]
|
|
|
|
actor = resolve_web_actor()
|
|
try:
|
|
assert_quota_available(actor)
|
|
except PolicyError as e:
|
|
return jsonify({"error": e.message}), e.status_code
|
|
|
|
try:
|
|
original_filename, ext = validate_actor_file(
|
|
file, allowed_types=ALLOWED_IMAGE_TYPES, actor=actor
|
|
)
|
|
except FileValidationError as e:
|
|
return jsonify({"error": e.message}), e.code
|
|
|
|
task_id, input_path = generate_safe_path(ext, folder_type="upload")
|
|
file.save(input_path)
|
|
|
|
task = remove_bg_task.delay(
|
|
input_path, task_id, original_filename,
|
|
**build_task_tracking_kwargs(actor),
|
|
)
|
|
record_accepted_usage(actor, "remove-bg", task.id)
|
|
|
|
return jsonify({
|
|
"task_id": task.id,
|
|
"message": "Background removal started. Poll /api/tasks/{task_id}/status for progress.",
|
|
}), 202
|