Update app.py

app.py

@@ -27,16 +27,24 @@ templates = {
         "easy": ["Differentiate the function: f(x) = {a}x^2 + {b}x + {c}", "Find the derivative of: f(x) = {a}x^3 - {b}x + {c}"],
         "medium": ["Integrate the function: f(x) = {a}x^2 + {b}x + {c}", "Find the integral of: f(x) = {a}x^3 - {b}x + {c}"],
         "hard": ["Solve the differential equation: {a}dy/dx + {b}y = {c}", "Find the solution to the differential equation: {a}d^2y/dx^2 - {b}dy/dx + {c}y = 0"]
+    },
+    "geometry": {
+        "easy": ["Find the area of a rectangle with length {a} and width {b}", "Calculate the perimeter of a rectangle with length {a} and width {b}"],
+        "medium": ["Find the area of a triangle with base {a} and height {b}", "Calculate the circumference of a circle with radius {a}"],
+        "hard": ["Find the volume of a cylinder with radius {a} and height {b}", "Calculate the surface area of a sphere with radius {a}"]
+    },
+    "trigonometry": {
+        "easy": ["Find sin({a})", "Find cos({a})"],
+        "medium": ["Calculate tan({a})", "Find the value of sin({a}) + cos({a})"],
+        "hard": ["Solve for θ in the equation sin(θ) = {a}", "Find the angle θ for which tan(θ) = {a}"]
     }
-    # Add more areas and difficulties as needed
 }
 
-def generate_synthetic_math_problems(num_problems):
+def generate_synthetic_math_problems(num_problems, selected_templates):
     problems = []
-
     for _ in range(num_problems):
-        # Randomly choose an area of mathematics
-        area = random.choice(list(templates.keys()))
+        # Randomly choose an area of mathematics from the selected templates
+        area = random.choice(selected_templates)
         
         # Randomly choose a difficulty level
         difficulty = random.choice(list(templates[area].keys()))
@@ -79,8 +87,8 @@ def solve_problem(problem, max_length):
     return answer
 
 @spaces.GPU(duration=120)
-def generate_and_solve_problems(num_problems, max_length):
-    problems = generate_synthetic_math_problems(num_problems)
+def generate_and_solve_problems(num_problems, max_length, selected_templates):
+    problems = generate_synthetic_math_problems(num_problems, selected_templates)
     solved_problems = []
 
     for problem in problems:
@@ -92,9 +100,22 @@ def generate_and_solve_problems(num_problems, max_length):
 
     return solved_problems
 
-def gradio_interface(num_problems, max_length):
+def gradio_interface(num_problems, max_length, algebra, calculus, geometry, trigonometry):
     print(f"Generating and solving {num_problems} problems with max length {max_length}...")
-    solved_problems = generate_and_solve_problems(num_problems, max_length)
+    selected_templates = []
+    if algebra:
+        selected_templates.append("algebra")
+    if calculus:
+        selected_templates.append("calculus")
+    if geometry:
+        selected_templates.append("geometry")
+    if trigonometry:
+        selected_templates.append("trigonometry")
+    
+    if not selected_templates:
+        return "Please select at least one math area."
+    
+    solved_problems = generate_and_solve_problems(num_problems, max_length, selected_templates)
     return json.dumps(solved_problems, indent=4)
 
 # Create a Gradio interface
@@ -102,7 +123,11 @@ iface = gr.Interface(
     fn=gradio_interface,
     inputs=[
         gr.Number(label="Number of Problems", value=10, precision=0),
-        gr.Slider(label="Max Output Length", minimum=10, maximum=200, value=50)
+        gr.Slider(label="Max Output Length", minimum=10, maximum=200, value=50),
+        gr.Checkbox(label="Algebra", value=True),
+        gr.Checkbox(label="Calculus", value=True),
+        gr.Checkbox(label="Geometry", value=True),
+        gr.Checkbox(label="Trigonometry", value=True)
     ],
     outputs=gr.Textbox(label="Generated and Solved Problems"),
     title="Synthetic Math Problem Generator and Solver",