Update app.py

app.py

@@ -334,9 +334,8 @@ def extract_and_run_sympy_code(response_text):
         
         # Import necessary modules
         import io
-        import sys
-        from contextlib import redirect_stdout
         import sympy
+        from contextlib import redirect_stdout
 
         # Create a string buffer to capture print output
         output_buffer = io.StringIO()
@@ -352,8 +351,8 @@ def extract_and_run_sympy_code(response_text):
         with redirect_stdout(output_buffer):
             exec(sympy_code, sympy_globals, local_vars)
         
-        # Append the calculated variables
-        output_buffer.write("SymPy Calculation Results:\n")
+        # Append the calculated variables to the output
+        output_buffer.write("\nSymPy Calculation Results:\n")
         output_buffer.write("-" * 25 + "\n")
         for var_name, value in local_vars.items():
             if not var_name.startswith('__') and not hasattr(value, '__module__'):