Update app.py

app.py

@@ -9,12 +9,14 @@ openai.api_key = st.secrets["OPENAI_API_KEY"]
 client = openai.OpenAI(api_key=openai.api_key)
 
 # Function to generate exam questions using OpenAI API with retry logic
-def generate_questions_with_retry(knowledge_material, question_type, cognitive_level, extra_instructions, case_based, max_retries=3):
+def generate_questions_with_retry(knowledge_material, question_type, cognitive_level, extra_instructions, case_based, num_choices=None, max_retries=3):
     # Adjust the number of questions based on the type
     if question_type == "Multiple Choice":
         num_questions = 3
     elif question_type == "Fill in the Blank":
         num_questions = 10
+    elif question_type == "True/False":
+        num_questions = 5
     else:  # Open-ended
         num_questions = 3
 
@@ -27,9 +29,11 @@ def generate_questions_with_retry(knowledge_material, question_type, cognitive_l
         if question_type != "Fill in the Blank":
             prompt += " Provide answers with short explanations."
 
-    else:
-        if question_type != "Fill in the Blank":
-            prompt += " Provide answers with short explanations."
+    if question_type == "Multiple Choice" and num_choices:
+        prompt += f" Each multiple choice question should have {num_choices} choices."
+
+    if question_type == "True/False":
+        prompt += " Generate True/False questions based on the provided material."
 
     retries = 0
     while retries < max_retries:
@@ -84,7 +88,12 @@ else:
 
     # Select question type
     question_type = st.selectbox("Select question type:", 
-                                 ["Multiple Choice", "Fill in the Blank", "Open-ended"])
+                                 ["Multiple Choice", "Fill in the Blank", "Open-ended", "True/False"])
+
+    # For multiple choice, let users select the number of choices
+    num_choices = None
+    if question_type == "Multiple Choice":
+        num_choices = st.selectbox("Select the number of choices for each question:", [3, 4, 5])
 
     # Select cognitive level
     cognitive_level = st.selectbox("Select cognitive level:", 
@@ -108,7 +117,8 @@ else:
                 question_type, 
                 cognitive_level, 
                 extra_instructions, 
-                case_based
+                case_based, 
+                num_choices
             )
             
             if questions:
@@ -137,7 +147,8 @@ else:
                 question_type, 
                 cognitive_level, 
                 extra_instructions, 
-                case_based
+                case_based, 
+                num_choices
             )
 
             # Check if the new set of questions is not the same as the previous set