Update main.py

main.py

@@ -559,14 +559,14 @@ async def generate_questions(request):
             seg_index = 0
             seg_index_map = ['Q', 'A', 'B', 'C', 'D', '{']
             raw_segmented = []
-            print(o['content'])
+
             for seg in o['content'].split('\n'):
                 
                 if seg.strip().startswith(seg_index_map[seg_index]):
                     print(seg)
                     raw_segmented.append(seg)
                     seg_index+=1
-                if seg_index == 5:
+                if seg_index == 6:
                     __parsed_outputs.extend(await parse_wh_question(raw_segmented, pgph_i))
                     break
             seg_index = 0