Atualizar app/app.py

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/<task_id>")
+@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