Update main.py

main.py

@@ -544,7 +544,9 @@ async def generate_questions(request):
             seg_index_map = ['Q.', 'A.', 'B.', 'C.', 'D.', '{']
             raw_segmented = []
             for seg in o['content'].split('\n'):
-                if seg.startswith(seg_index_map[seg_index]):
+                
+                if seg.strip().startswith(seg_index_map[seg_index]):
+                    print(seg)
                     raw_segmented.append(seg)
                     seg_index+=1
                 if seg_index == 5: