Update app.py

app.py

@@ -345,26 +345,36 @@ def extract_and_run_sympy_code(response_text):
         sympy_code = '\n'.join(filtered_lines)
         logger.debug(f"Code after removing imports:\n{sympy_code}")
         
-        # Create globals dict with all SymPy symbols
+        # Create output buffer
         import io
         import sys
         from contextlib import redirect_stdout
+        output_buffer = io.StringIO()
         
-        # Import SymPy first at the function level
+        # Import SymPy 
         import sympy
         
-        # Create output buffer
-        output_buffer = io.StringIO()
-        
         # Create globals dict with SymPy module
         globals_dict = {
             '__builtins__': __builtins__,
+            'Symbol': sympy.Symbol,
             'symbols': sympy.symbols,
-            'diff': sympy.diff,
             'solve': sympy.solve,
+            'diff': sympy.diff,
             'integrate': sympy.integrate,
+            'limit': sympy.limit,
             'oo': sympy.oo,
-            'Sum': sympy.Sum
+            'Sum': sympy.Sum,
+            'simplify': sympy.simplify,
+            'expand': sympy.expand,
+            'factor': sympy.factor,
+            'sin': sympy.sin,
+            'cos': sympy.cos,
+            'tan': sympy.tan,
+            'exp': sympy.exp,
+            'log': sympy.log,
+            'sqrt': sympy.sqrt,
+            'pi': sympy.pi
         }
         
         logger.debug(f"Globals dictionary contains: {list(globals_dict.keys())}")
@@ -387,11 +397,10 @@ def extract_and_run_sympy_code(response_text):
         
         # Add stdout if there was any
         if stdout_output.strip():
-            output_lines.append(f"Output:")
-            output_lines.append(stdout_output.strip())
+            output_lines.extend(["", "Output:", stdout_output.strip()])
         
         # Add variables
-        output_lines.append("\nVariables:")
+        output_lines.extend(["", "Variables:"])
         for name, value in local_vars.items():
             if not name.startswith('__') and not hasattr(value, '__module__'):
                 output_lines.append(f"{name} = {value}")
@@ -407,7 +416,7 @@ def extract_and_run_sympy_code(response_text):
         import traceback
         logger.error(f"Full traceback:\n{traceback.format_exc()}")
         return f"Error executing SymPy code: {str(e)}"
-
+        
         
 # Create Gradio interface
 with gr.Blocks() as interface: