Update main.py

main.py

@@ -335,34 +335,36 @@ async def parse_wh_question(raw_qa_list, pgph_i):
             segments = [r for r in x.split('\n') if r.__len__()]
             raw_key = segments[5].strip()
             raw_key = 'A' if 'A' in raw_key else 'B' if 'B' in raw_key else 'D' if 'D' in raw_key else 'C'
+
+            # print(segments) 
+            match randint(0, 3):
+                case 0 | 1:
+    
+                    __parsed_outputs.append(
+                        {
+                            "pgph_i": pgph_i,
+                            "prompt": segments[0],
+                            "type": "MCQ",
+                            "choices": segments[1:5],
+                            "keys": [segments[__ANS_KEY_MAPPING[raw_key]],],
+                        }
+                    )
+                
+                case 2 | 3:
+                    __parsed_outputs.append(
+                        {
+                            "pgph_i": pgph_i,
+                            "prompt": segments[0],
+                            "type": "OPEN",
+                            # Cleaning up ML output
+                            "keys": [segments[__ANS_KEY_MAPPING[raw_key]].split(' ',1)[1]],
+                            "choices": [segments[__ANS_KEY_MAPPING[raw_key]]]
+                        }
+                    )
+
         except:
             print("invalid: ", x)
             continue
-        # print(segments) 
-        match randint(0, 3):
-            case 0 | 1:
-
-                __parsed_outputs.append(
-                    {
-                        "pgph_i": pgph_i,
-                        "prompt": segments[0],
-                        "type": "MCQ",
-                        "choices": segments[1:5],
-                        "keys": [segments[__ANS_KEY_MAPPING[raw_key]],],
-                    }
-                )
-            
-            case 2 | 3:
-                __parsed_outputs.append(
-                    {
-                        "pgph_i": pgph_i,
-                        "prompt": segments[0],
-                        "type": "OPEN",
-                        # Cleaning up ML output
-                        "keys": [segments[__ANS_KEY_MAPPING[raw_key]].split(' ',1)[1]],
-                        "choices": [segments[__ANS_KEY_MAPPING[raw_key]]]
-                    }
-                )
 
     return __parsed_outputs