Update app.py

app.py

@@ -368,14 +368,14 @@ def extract_and_run_sympy_code(response_text):
         try:
             # Execute the code and capture all output in one redirect_stdout block
             with redirect_stdout(output_buffer):
-                # First print a header
                 print("SymPy Calculation Results:")
                 print("-" * 25)
                 
                 # Execute the code
                 exec(sympy_code, globals_dict, local_vars)
                 
-                # Print all variables created
+                # Print all variables that were created
+                print("\nFinal Values:")
                 for var_name, value in local_vars.items():
                     if (not var_name.startswith('__') and 
                         not hasattr(value, '__module__')):
@@ -391,8 +391,7 @@ def extract_and_run_sympy_code(response_text):
             logger.error(f"Other error during execution: {str(e)}")
             return f"Error executing SymPy code: {str(e)}"
         
-        # Get the captured output
-        result = output_buffer.getvalue()
+        result = output_buffer.getvalue().strip()
         logger.debug(f"Final output:\n{result}")
         return result