Update app.py

app.py

@@ -9,7 +9,7 @@ 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, max_retries=3):
+def generate_questions_with_retry(knowledge_material, question_type, cognitive_level, extra_instructions, case_based, max_retries=3):
     # Adjust the number of questions based on the type
     if question_type == "Multiple Choice":
         num_questions = 3
@@ -18,15 +18,24 @@ def generate_questions_with_retry(knowledge_material, question_type, cognitive_l
     else:  # Open-ended
         num_questions = 3
 
+    # Base prompt
     prompt = f"Generate {num_questions} {question_type.lower()} exam questions based on {cognitive_level.lower()} level from the following material: {knowledge_material}. {extra_instructions}"
-    if question_type != "Fill in the Blank":
-        prompt += " Provide answers with short explanations."
+
+    # If case-based medical situation is selected, modify the prompt
+    if case_based:
+        prompt = f"Generate {num_questions} {question_type.lower()} exam questions based on {cognitive_level.lower()} level from the following medical material: {knowledge_material}. The questions should be based on case-based medical situations, such as patient scenarios. {extra_instructions}"
+        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."
 
     retries = 0
     while retries < max_retries:
         try:
             response = client.chat.completions.create(
-                model="GPT-4o mini",
+                model="gpt-4o-mini",
                 messages=[
                     {"role": "system", "content": "You are a helpful assistant for generating exam questions."},
                     {"role": "user", "content": prompt}
@@ -61,13 +70,17 @@ else:
     # Word count check
     if len(knowledge_material.split()) > 3000:
         st.warning("Please limit the knowledge material to 3,000 words or fewer.")
-    
+
     # File uploader for PDFs (limited to 5 MB)
     uploaded_file = st.file_uploader("Upload a file (PDF)", type="pdf")
     
     if uploaded_file is not None:
         if uploaded_file.size > 5 * 1024 * 1024:  # 5 MB limit
             st.warning("File size exceeds 5 MB. Please upload a smaller file.")
+        else:
+            # Here you can add code to extract text from the PDF if needed
+            # For simplicity, we're focusing on the text input for now
+            st.success("File uploaded successfully! (Text extraction not implemented yet.)")
 
     # Select question type
     question_type = st.selectbox("Select question type:", 
@@ -77,6 +90,9 @@ else:
     cognitive_level = st.selectbox("Select cognitive level:", 
                                    ["Recall", "Understanding", "Application", "Analysis", "Synthesis", "Evaluation"])
 
+    # Checkbox for Case-Based Medical Situations
+    case_based = st.checkbox("Generate case-based medical exam questions")
+
     # Extra input field for additional instructions (placed below cognitive level)
     extra_instructions = st.text_area("Enter additional instructions (e.g., how you want the questions to be phrased):")
 
@@ -87,7 +103,13 @@ else:
     if st.button("Generate Questions"):
         if len(knowledge_material.split()) <= 3000:
             # Generate questions with retry logic
-            questions = generate_questions_with_retry(knowledge_material, question_type, cognitive_level, extra_instructions)
+            questions = generate_questions_with_retry(
+                knowledge_material, 
+                question_type, 
+                cognitive_level, 
+                extra_instructions, 
+                case_based
+            )
             
             if questions:
                 st.write("Generated Exam Questions:")
@@ -110,7 +132,13 @@ else:
     if st.button("Generate More Questions"):
         if len(knowledge_material.split()) <= 3000:
             # Regenerate new questions, trying to avoid repeated content
-            questions = generate_questions_with_retry(knowledge_material, question_type, cognitive_level, extra_instructions)
+            questions = generate_questions_with_retry(
+                knowledge_material, 
+                question_type, 
+                cognitive_level, 
+                extra_instructions, 
+                case_based
+            )
 
             # Check if the new set of questions is not the same as the previous set
             if questions and questions not in st.session_state['previous_questions']: