Update app.py

app.py

@@ -333,6 +333,19 @@ def extract_and_run_sympy_code(response_text):
             return "Malformed SymPy code block."
         
         sympy_code = response_text[code_start:code_end].strip()
+
+        if "print" not in sympy_code and any(x in sympy_code for x in ['diff', 'integrate', 'solve']):
+            lines = sympy_code.split('\n')
+            modified_lines = []
+            for line in lines:
+                modified_lines.append(line)
+                # If line assigns a value and looks like it creates an expression
+                if ('=' in line and '#' not in line and 'import' not in line 
+                    and any(x in line for x in ['diff', 'integrate', 'solve', 'sqrt', 'log'])):
+                    var_name = line.split('=')[0].strip()
+                    modified_lines.append(f"print('{var_name} = ')")
+                    modified_lines.append(f"print(simplify({var_name}))")
+            sympy_code = '\n'.join(modified_lines)
         
         # Import necessary modules
         import io
@@ -354,6 +367,7 @@ def extract_and_run_sympy_code(response_text):
             "exp": sympy.exp,
             "simplify": sympy.simplify,
             "expand": sympy.expand,
+            "log": sympy.log, 
             "print": print,  # Allow print statements
             "pi": sympy.pi,
             "N": sympy.N,