Update app.py

app.py

@@ -136,13 +136,11 @@ 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
 2. Maintain similar level of rigor and detail
 3. Use similar proof techniques where applicable
 4. Follow similar notation and presentation style
-
 {proof_examples}"""
     
     return enhanced_prompt
@@ -264,7 +262,6 @@ The application question MUST:
     2. Translate the practical problem into mathematical terms
     3. Solve using appropriate mathematical techniques
     4. Interpret the results in the context of the original problem
-
 Example contexts might include:
 - Physics applications (motion, forces, work)
 - Engineering scenarios (optimization, rates of change)
@@ -280,7 +277,6 @@ The proof question MUST:
 - Focus on demonstrating logical reasoning
 - Require justification for each step
 - Emphasize theoretical understanding
-
 The proof question MAY NOT:
 - Include Real-world applications or scenarios
 - Include Pure computation problems
@@ -292,7 +288,6 @@ The computation question MUST:
 - Focus on mathematical techniques
 - Have concrete answers in the form of algebraic expressions (about half of questions) or numbers (about half of questions)
 - Test procedural knowledge
-
 The computation question MAY NOT:
 - Include extended real-world applications or scenarios
 - Ask for a proof
@@ -354,7 +349,6 @@ STRICT REQUIREMENTS:
    {chr(10).join(f'   - {c}' for c in difficulty_params['constraints'])}
    
    {problem_type_addition}
-
 3. Style Reference:
    Question should be {difficulty_params['example_style']}
 4. The question MUST:
@@ -400,7 +394,6 @@ STRICT REQUIREMENTS:
    {chr(10).join(f'   - {c}' for c in difficulty_params['constraints'])}
    
    {problem_type_addition}
-
 3. Style Reference:
    Question should be {difficulty_params['example_style']}
 4. For LaTeX formatting:
@@ -422,7 +415,8 @@ STRICT REQUIREMENTS:
    - Print variable values after they are computed
    - Format numerical results using evalf() for decimal
 9. When writing SymPy code to verify solutions:
-   - Start with "from sympy import" and include eall necessary imports
+   - Start with essential imports: from sympy import * 
+   - Or use specific imports as needed: from sympy import Symbol, solve, limit, Abs, integrate, Function, diff, sin, cos
    - DO NOT use NumPy, SciPy, or other libraries
    - For numerical integration, use integrate() instead of scipy.integrate.quad()
    - For any numerical computations, use native SymPy functions
@@ -518,37 +512,11 @@ def extract_and_run_sympy_code_simple(response_text):
         import io
         import sympy
         from contextlib import redirect_stdout
-        from sympy import (Symbol, symbols, solve, limit, Abs, integrate,
-                         Function, diff, sin, cos, exp, log, sqrt,
-                         Matrix, Derivative, Integral, series)
-        
-        # Create globals dict with all necessary imports
-        globals_dict = {
-            'sympy': sympy,
-            'Symbol': Symbol,
-            'symbols': symbols,
-            'solve': solve,
-            'limit': limit,
-            'Abs': Abs,
-            'integrate': integrate,
-            'Function': Function,
-            'diff': diff,
-            'sin': sin,
-            'cos': cos,
-            'exp': exp,
-            'log': log,
-            'sqrt': sqrt,
-            'Matrix': Matrix,
-            'Derivative': Derivative,
-            'Integral': Integral,
-            'series': series,
-            'print': print
-        }
         
         # Capture output
         output_buffer = io.StringIO()
         with redirect_stdout(output_buffer):
-            exec(sympy_code, globals_dict)
+            exec(sympy_code, {"sympy": sympy, "print": print})
             
         return output_buffer.getvalue().strip() or "No output produced"
     
@@ -569,17 +537,13 @@ def check_and_resolve_discrepancy(initial_response, sympy_output):
         then please say that they are consistent and briefly explain why. 
         
         If the two answers are inconsistent with each other then please:
-
         1. Identify which solution is correct
         2. Explain the error in the incorrect solution
         3. Provide a revised complete solution that fixes any errors
-
 Original solution:
 {initial_response}
-
 SymPy Verification Results:
 {sympy_output}
-
 Please maintain the same LaTeX formatting as the original solution."""
 
         # Make API call for resolution