chore: add @doist/todoist-ai
dependency to package.json اول دفعة من التطوير
This commit is contained in:
@@ -1,4 +1,6 @@
|
||||
"""Background removal route."""
|
||||
import os
|
||||
|
||||
from flask import Blueprint, request, jsonify, current_app
|
||||
|
||||
from app.extensions import limiter
|
||||
@@ -10,6 +12,7 @@ from app.services.policy_service import (
|
||||
resolve_web_actor,
|
||||
validate_actor_file,
|
||||
)
|
||||
from app.services.quote_service import create_quote, QuoteError
|
||||
from app.utils.file_validator import FileValidationError
|
||||
from app.utils.sanitizer import generate_safe_path
|
||||
from app.tasks.removebg_tasks import remove_bg_task
|
||||
@@ -52,13 +55,20 @@ def remove_bg_route():
|
||||
task_id, input_path = generate_safe_path(ext, folder_type="upload")
|
||||
file.save(input_path)
|
||||
|
||||
file_size_kb = os.path.getsize(input_path) / 1024
|
||||
try:
|
||||
quote = create_quote(actor.user_id, actor.plan, "remove-bg", file_size_kb=file_size_kb)
|
||||
except QuoteError as e:
|
||||
return jsonify({"error": e.message}), e.status_code
|
||||
|
||||
task = remove_bg_task.delay(
|
||||
input_path, task_id, original_filename,
|
||||
**build_task_tracking_kwargs(actor),
|
||||
)
|
||||
record_accepted_usage(actor, "remove-bg", task.id)
|
||||
record_accepted_usage(actor, "remove-bg", task.id, quote=quote)
|
||||
|
||||
return jsonify({
|
||||
"task_id": task.id,
|
||||
"message": "Background removal started. Poll /api/tasks/{task_id}/status for progress.",
|
||||
"quote": quote.to_dict(),
|
||||
}), 202
|
||||
|
||||
Reference in New Issue
Block a user