Update app.py

app.py

@@ -40,9 +40,13 @@ def generate_question_answer_pairs(input_file):
         if type(i) == dict:
             continue
         pair = i[0]['generated_text']
-        question_part = re.search(r'^(.*?\?)',pair).group(1).strip()
-        answer_part = re.search(r'\?(.*)', pair).group(1).strip()
-        result += f"Question: {question_part}\nAnswer: {answer_part}\n\n"
+        question = re.search(r'^(.*?\?)', pair)
+        answer = re.search(r'\?(.*)', pair)
+
+        if question and answer:
+            question_part = question.group(1).strip()
+            answer_part = answer.group(1).strip()
+            result += f"Question: {question_part}\nAnswer: {answer_part}\n\n"
             
     return result