Update app.py

app.py

@@ -375,6 +375,10 @@ STRICT REQUIREMENTS:
    - Use Abs() for absolute value (not abs())
    - Define functions using Function() class when needed
    - Include print statements for ALL calculations and results
+   - Use simple print statements instead of f-strings for SymPy expressions
+   - Print expressions with labels on separate lines:
+     print("Expression label:")
+     print(expression)
    - Print intermediate steps and final answers
    - Print variable values after they are computed
    - For numeric calculations:
@@ -433,6 +437,9 @@ STRICT REQUIREMENTS:
    - Include print statements for ALL calculations and results
    - Print intermediate steps and final answers
    - Print variable values after they are computed
+   - Print expressions with labels on separate lines:
+     print("Expression label:")
+     print(expression)
    - For numeric calculations:
      * Use Float() for decimal numbers
      * Use float() for final printing of results
@@ -500,6 +507,7 @@ STRICT REQUIREMENTS:
 def extract_and_run_sympy_code_simple(response_text):
     """
     Extract SymPy code from the response and execute it.
+    Handles SymPy expression printing safely without f-strings.
     Returns the output exactly as SymPy would produce it.
     """
     try:
@@ -515,21 +523,49 @@ def extract_and_run_sympy_code_simple(response_text):
         
         sympy_code = response_text[code_start:code_end].strip()
         
+        # Replace problematic f-strings with safe print statements
+        sympy_code = sympy_code.replace('f"', '"')
+        sympy_code = sympy_code.replace('{', '{{')
+        sympy_code = sympy_code.replace('}', '}}')
+        sympy_code = sympy_code.replace('print(f"', 'print("')
+        
+        # Add safe printing function
+        safe_print_setup = """
+def safe_print_expr(label, expr):
+    print(label)
+    print(expr)
+"""
+        
+        # Replace f-string prints with safe prints
+        sympy_code = safe_print_setup + sympy_code
+        sympy_code = sympy_code.replace('print(f"', 'safe_print_expr("')
+        
         # Set up basic SymPy environment
         import io
         import sympy
         from contextlib import redirect_stdout
         
+        # Create a dictionary of allowed names
+        globals_dict = {
+            "sympy": sympy,
+            "Symbol": sympy.Symbol,
+            "solve": sympy.solve,
+            "diff": sympy.diff,
+            "simplify": sympy.simplify,
+            "print": print,
+            "float": float
+        }
+        
         # Capture output
         output_buffer = io.StringIO()
         with redirect_stdout(output_buffer):
-            exec(sympy_code, {"sympy": sympy, "print": print})
+            exec(sympy_code, globals_dict)
             
         return output_buffer.getvalue().strip() or "No output produced"
     
     except Exception as e:
         return f"Error executing SymPy code: {str(e)}"
-               
+        
 def check_and_resolve_discrepancy(initial_response, sympy_output):
     """
     Compare the SymPy output with the initial response and resolve any discrepancies