added user choices of type and level

app.py

@@ -80,14 +80,36 @@ def save_to_temp_file(content, filename):
         logger.error(f"Error saving temporary file: {str(e)}")
         raise
 
-def generate_test(subject):
+def validate_question_counts(computation, proof, application):
+    """Validate that question counts sum to 3"""
+    try:
+        comp = int(computation)
+        prf = int(proof)
+        app = int(application)
+        
+        if comp + prf + app != 3:
+            return False, "Total number of questions must equal 3"
+        if any(x < 0 for x in [comp, prf, app]):
+            return False, "Question counts cannot be negative"
+        return True, ""
+    except ValueError:
+        return False, "Please enter valid numbers"
+
+def generate_test(subject, difficulty, computation_count, proof_count, application_count):
     """Generate a math test"""
     try:
+        # Validate inputs
+        is_valid, error_message = validate_question_counts(
+            computation_count, proof_count, application_count
+        )
+        if not is_valid:
+            return error_message, None, None
+            
         if not os.environ.get('ANTHROPIC_API_KEY'):
             logger.error("Anthropic API key not found")
             return "Error: Anthropic API key not configured", None, None
         
-        logger.debug(f"Generating test for subject: {subject}")
+        logger.debug(f"Generating test for subject: {subject} at difficulty level: {difficulty}")
         
         # Check rate limit
         now = datetime.now()
@@ -106,18 +128,35 @@ def generate_test(subject):
         
         selected_topics = random.sample(topics.get(subject, ["general"]), min(3, len(topics.get(subject, ["general"]))))
         logger.debug(f"Selected topics: {selected_topics}")
+
+        # Create question type list based on user input
+        question_types = (
+            ["computation"] * int(computation_count) +
+            ["proof"] * int(proof_count) +
+            ["application"] * int(application_count)
+        )
+        
+        difficulty_descriptions = {
+            1: "introductory undergraduate level",
+            2: "early undergraduate level",
+            3: "advanced undergraduate level",
+            4: "graduate level",
+            5: "advanced graduate level"
+        }
         
         system_prompt = f"""You will write math exam questions. Follow these requirements EXACTLY:
         1. Write exactly 3 university-level questions focusing on these specific topics: {', '.join(selected_topics)}
-        2. Include one of each of the following three types of question: computation, proof, application
-        3. For LaTeX math formatting:
+        2. Include the following question types in this exact order:
+           {', '.join(question_types)}
+        3. Make all questions {difficulty_descriptions[difficulty]} difficulty
+        4. For LaTeX math formatting:
            - Use $ for simple inline math
            - For equations and solution steps, use $$ on separate lines
            - For multi-step solutions, put each step on its own line in $$ $$
            - DO NOT use \\begin{{aligned}} or any other environments
-        4. Number each question as 1), 2), 3)
-        5. Include detailed solutions after each question
-        6. Keep formatting simple and clear"""
+        5. Number each question as 1), 2), 3)
+        6. Include detailed solutions after each question
+        7. Keep formatting simple and clear"""
         
         logger.debug("Sending request to Anthropic API")
         message = anthropic.messages.create(
@@ -160,23 +199,55 @@ with gr.Blocks() as interface:
     Each test features different topics and difficulty levels. Limited to 25 requests per day.""")
     
     with gr.Row():
-        subject_dropdown = gr.Dropdown(
-            choices=[
-                "Single Variable Calculus",
-                "Multivariable Calculus", 
-                "Linear Algebra",
-                "Differential Equations",
-                "Real Analysis",
-                "Complex Analysis",
-                "Abstract Algebra",
-                "Probability Theory",
-                "Numerical Analysis",
-                "Topology"
-            ],
-            label="Select Mathematics Subject",
-            info="Choose a subject for the exam questions"
-        )
-        generate_btn = gr.Button("Generate Test")
+        with gr.Column():
+            subject_dropdown = gr.Dropdown(
+                choices=[
+                    "Single Variable Calculus",
+                    "Multivariable Calculus", 
+                    "Linear Algebra",
+                    "Differential Equations",
+                    "Real Analysis",
+                    "Complex Analysis",
+                    "Abstract Algebra",
+                    "Probability Theory",
+                    "Numerical Analysis",
+                    "Topology"
+                ],
+                label="Select Mathematics Subject",
+                info="Choose a subject for the exam questions"
+            )
+            
+            difficulty_slider = gr.Slider(
+                minimum=1,
+                maximum=5,
+                step=1,
+                value=3,
+                label="Difficulty Level",
+                info="1: Introductory Undergraduate, 3: Advanced Undergraduate, 5: Advanced Graduate"
+            )
+    
+    with gr.Row():
+        with gr.Column():
+            computation_count = gr.Number(
+                value=1,
+                label="Number of Computation Questions",
+                info="Enter the number of computation questions (total must be 3)",
+                precision=0
+            )
+            proof_count = gr.Number(
+                value=1,
+                label="Number of Proof Questions",
+                info="Enter the number of proof questions (total must be 3)",
+                precision=0
+            )
+            application_count = gr.Number(
+                value=1,
+                label="Number of Application Questions",
+                info="Enter the number of application questions (total must be 3)",
+                precision=0
+            )
+    
+    generate_btn = gr.Button("Generate Test")
     
     output_text = gr.Markdown(
         label="Generated Test Preview",
@@ -192,7 +263,13 @@ with gr.Blocks() as interface:
     
     generate_btn.click(
         generate_test,
-        inputs=[subject_dropdown],
+        inputs=[
+            subject_dropdown,
+            difficulty_slider,
+            computation_count,
+            proof_count,
+            application_count
+        ],
         outputs=[output_text, questions_file, full_file]
     )