random pick an incomplete task instead of go with sequential order

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 from chatbot_simulator import ChatbotSimulation
-from huggingface_hub import HfApi, create_repo
 from datasets import load_dataset
 import json_repair
 import random
@@ -18,23 +17,25 @@ db = firestore.client()
 openai_api_key = os.getenv("OPENAI_API_KEY")
 
 
-def find_smallest_incomplete_task(app_name):
+def find_random_incomplete_task(app_name):
 
     collection_ref = db.collection(app_name)
-    docs = collection_ref.stream()
-
-    smallest_incomplete_idx = None
-    for doc in docs:
-        doc_id = doc.id
-        _, idx = doc_id.split('_')
-        idx = int(idx)
+    incomplete_indices = []
 
+    for doc in collection_ref.stream():
         task_data = doc.to_dict()
         if task_data['task_completed'] is None and task_data['task_completed_steps'] is None:
-            if smallest_incomplete_idx is None or idx < smallest_incomplete_idx:
-                smallest_incomplete_idx = idx
+            _, idx = doc.id.split('_')
+            idx = int(idx)
+            incomplete_indices.append(idx)
+
+    # Check if any incomplete tasks found
+    if not incomplete_indices:
+        return None
 
-    return smallest_incomplete_idx
+    # Pick a random incomplete index from the list
+    random_idx = random.choice(incomplete_indices)
+    return random_idx
 
 
 def write_task_data(app_name, idx, task, task_complete, task_completed_step):