diff --git a/app/app.py b/app/app.py index ce64a77..4d8d48e 100644 --- a/app/app.py +++ b/app/app.py @@ -841,6 +841,35 @@ def vision(cid): return jsonify({"status": "ok", "description": description}) + +from tasks import generate_task + +@app.route("/api/queue/generate", methods=["POST"]) +@login_required +def queue_generate(): + data = request.json + cid = data["conversation_id"] + model = data["model"] + prompt = data["prompt"] + + task = generate_task.delay(cid, model, prompt) + + return jsonify({"task_id": task.id}) + +from celery_app import celery + +@app.route("/api/queue/status/") +@login_required +def queue_status(task_id): + result = celery.AsyncResult(task_id) + return jsonify({ + "task_id": task_id, + "state": result.state, + "result": result.result if result.ready() else None + }) + + + # *** import json