Update app.py

app.py

@@ -32,23 +32,28 @@ def generate_mcqs(text_chunks, num_questions=5):
     mcqs = []
 
     for chunk in text_chunks:
-        input_text = f"Generate a multiple-choice question from the following text:\n\n{chunk}\n\nQuestion:"
+        input_text = f"Based on the following text, generate a multiple-choice question along with four plausible options and mark the correct answer:\n\n{chunk}\n\nQuestion:"
         generated = generator(input_text, max_length=400, num_return_sequences=1)
-        question_text = generated[0]["generated_text"].split("Question:")[1].strip()
-        
-        # Generate options for the question
-        options_text = f"Generate four plausible multiple-choice options for the following question:\n\n{question_text}\nOptions:"
-        options_generated = generator(options_text, max_length=200, num_return_sequences=1)
-        options_list = options_generated[0]["generated_text"].split("Options:")[1].strip().split("\n")
-        options = [f"Option {chr(65 + i)}: {option.strip()}" for i, option in enumerate(options_list[:4])]
-        
-        if len(options) < 4:
-            continue
+        generated_text = generated[0]["generated_text"]
+
+        # Extract question and options
+        try:
+            question_part = generated_text.split("Question:")[1].strip()
+            question = question_part.split("Options:")[0].strip()
+            options_part = question_part.split("Options:")[1].strip()
+            options = options_part.split("\n")
+            
+            # Ensure four options
+            if len(options) < 4:
+                continue
 
-        correct_answer = options[0]  # Placeholder for correct answer identification logic
-        
-        mcq_formatted = f"Q: {question_text}\n{options[0]}\n{options[1]}\n{options[2]}\n{options[3]}\nCorrect Answer: {correct_answer}"
-        mcqs.append(mcq_formatted)
+            options = [f"Option {chr(65 + i)}: {option.strip()}" for i, option in enumerate(options[:4])]
+            correct_answer = options[0]  # Placeholder for correct answer identification logic
+
+            mcq_formatted = f"Q: {question}\n{options[0]}\n{options[1]}\n{options[2]}\n{options[3]}\nCorrect Answer: {correct_answer}"
+            mcqs.append(mcq_formatted)
+        except:
+            continue
 
         if len(mcqs) >= num_questions:
             break