Update generator

app.py

@@ -147,6 +147,7 @@ async def generate_saved_endpoint(
     }
     
     all_mcqs = {}
+    counter = 1
 
     for difficulty, n_questions in difficulty_counts.items():
         try:
@@ -161,14 +162,27 @@ async def generate_saved_endpoint(
                 enable_fiddler=enable_fiddler,
                 target_difficulty=difficulty,
             )
-            for qid, qobj in mcqs.items():
-                if qid not in all_mcqs:
-                    all_mcqs[qid] = qobj
-                else:
-                    if isinstance(all_mcqs[qid], list):
-                        all_mcqs[qid].append(qobj)
+            questions_list = []
+            if isinstance(mcqs, dict):
+                # if dict values are lists or single objects, collect them
+                for v in mcqs.values():
+                    if isinstance(v, list):
+                        questions_list.extend(v)
                     else:
-                        all_mcqs[qid] = [all_mcqs[qid], qobj]
+                        questions_list.append(v)
+            elif isinstance(mcqs, list):
+                questions_list = mcqs
+            else:
+                # unexpected shape — skip or handle as needed
+                continue
+
+            # assign sequential ids (as strings). use `counter` for numeric keys if preferred.
+            for qobj in questions_list:
+                if isinstance(qobj, dict):
+                    # optional: keep which difficulty the question came from
+                    qobj["_difficulty"] = difficulty
+                all_mcqs[str(counter)] = qobj
+                counter += 1
 
         except Exception as e:
             raise HTTPException(status_code=500, detail=f"Generation from saved file failed: {e}")