Update main.py

main.py

@@ -557,18 +557,23 @@ async def generate_questions(request):
 
             # print(raw_segmented)
             seg_index = 0
-            seg_index_map = ['Q', 'A', 'B', 'C', 'D', '{']
+            seg_index_map = ['Q', 'A', 'B', 'C', 'D', '']
             raw_segmented = []
 
             for seg in o['content'].split('\n'):
                 
                 if seg.strip().startswith(seg_index_map[seg_index]):
+                    if seg_index == 5:
+                        if not ('A' in seg or 'B' in seg or 'C' in seg or 'D' in seg):
+                            continue
                     print(seg)
                     raw_segmented.append(seg)
                     seg_index+=1
                 if seg_index == 6:
-                    __parsed_outputs.extend(await parse_wh_question(raw_segmented, pgph_i))
-                    break
+                    __parsed_outputs.extend(await parse_wh_question(raw_segmented.copy(), pgph_i))
+                    raw_segmented = []
+                    seg_index = 0
+                    
             seg_index = 0
 
         elif (o['style'] == QType.STMT):