Update app.py

app.py

@@ -353,7 +353,7 @@ def enhance_prompt_with_proofs(system_prompt, subject, topic):
     # Add specific instructions for using the examples
     enhanced_prompt = f"""{system_prompt}
 ADDITIONAL PROOF GUIDELINES:
-1. Consider the following proof examples from established textbooks
+1. Consider the following proof examples from an established textbook
 2. Maintain similar level of rigor and detail
 3. Use similar proof techniques where applicable
 4. Follow similar notation and presentation style
@@ -508,7 +508,7 @@ The computation question MAY NOT:
     }
     return problem_type_additions.get(question_type, "")
 
-def generate_question(subject, difficulty, question_type):
+def generate_question(subject, difficulty, question_type, use_enhancement=False):
     """Generate a single math question with additional verification"""
     try:
         if not os.environ.get('ANTHROPIC_API_KEY'):
@@ -516,6 +516,7 @@ def generate_question(subject, difficulty, question_type):
             return "Error: Anthropic API key not configured", None, None
         
         logger.debug(f"Generating {question_type} question for subject: {subject} at difficulty level: {difficulty}")
+        logger.debug(f"Textbook enhancement: {'enabled' if use_enhancement else 'disabled'}")
         
         # Check rate limit
         now = datetime.now()
@@ -589,8 +590,7 @@ STRICT REQUIREMENTS:
     - Connect each step with a rationale
     - Always justify the existence of bounds
     - If you conclude certain terms vanish in a limit, clearly justify why
-    - Label subsections of the proof. Don't just say 'step 1', 'step 2', etc but rather the labels of the subsections
-       themselves should provide a clear outline of the main aspects of the proof
+    - If you state that a function has a certain characteristic, such as being Riemann integrable for example, clearly explain why
     - In notes after the proof, if you observe aspects of the problem that might confuse students, address them."""
 
 #Consider
@@ -599,9 +599,12 @@ STRICT REQUIREMENTS:
    #- Import specific functions instead of using 'from sympy import *'
    #- Print results of each calculation step
 
-   # Enhance the prompt with proof examples if applicable
-        if subject == "Real Analysis" and question_type == "proof":
+    # Only enhance the prompt if explicitly requested AND it's a Real Analysis proof
+        if use_enhancement and subject == "Real Analysis" and question_type == "proof":
+            logger.debug("Applying textbook enhancement to prompt")
             system_prompt = enhance_prompt_with_proofs(system_prompt, subject, selected_topic)
+        else:
+            logger.debug("Skipping textbook enhancement")
     
         logger.debug("Sending request to Anthropic API")
         message = anthropic.messages.create(
@@ -934,6 +937,14 @@ with gr.Blocks() as interface:
                 info="Select the type of question you want",
                 value="computation"
             )
+            
+            # Add the new enhancement checkbox
+            use_enhancement = gr.Radio(
+                choices=["yes", "no"],
+                label="Enhance with Textbook Material",
+                info="Include relevant textbook examples to guide question generation",
+                value="no"
+            )
     
     generate_btn = gr.Button("Generate Question")
     
@@ -949,12 +960,14 @@ with gr.Blocks() as interface:
         questions_file = gr.File(label="Question Only (LaTeX)")
         full_file = gr.File(label="Question with Solution (LaTeX)")
     
+    # Update the click event to include the new parameter
     generate_btn.click(
         generate_question,
         inputs=[
             subject_dropdown,
             difficulty_slider,
-            question_type
+            question_type,
+            use_enhancement
         ],
         outputs=[output_text, questions_file, full_file]
     )