fix(eta): Base ETA on original place in queue

app.py

@@ -16,6 +16,29 @@ def index():
 tasks = {}
 
 
+def place_in_queue(task_id):
+
+    pending_tasks = list(task for task in tasks.values()
+                         if task["status"] == "pending")
+
+    try:
+        return pending_tasks.index(task_id) + 1
+    except:
+        return 0
+
+
+def calculate_eta(task_id):
+    total_durations = list(task["completed_at"] - task["created_at"]
+                           for task in tasks.values() if "completed_at" in task)
+
+    place = tasks[task_id]["initial_place_in_queue"] or 1
+
+    if len(total_durations):
+        return sum(total_durations) / len(total_durations) * place
+    else:
+        return 40 * place
+
+
 @app.route('/task/create')
 def create_task():
     prompt = request.args.get('prompt') or "покемон"
@@ -28,10 +51,14 @@ def create_task():
         "task_id": task_id,
         "created_at": created_at,
         "prompt": prompt,
+        "initial_place_in_queue": place_in_queue(task_id),
         "status": "pending",
         "poll_count": 0,
     }
 
+    print("Place in queue: ", place_in_queue(task_id))
+    print("ETA: ", calculate_eta(task_id))
+
     return jsonify(tasks[task_id])