Update app.py

app.py

@@ -295,8 +295,17 @@ def main():
             if equation:
                 try:
                     # Solve the equation or expression
-                    solution = solve_equation(equation)
-                    steps = generate_steps(equation)
+                    if "laplace" in equation.lower():
+                        solution = laplace_transform_expression(equation)
+                        steps = solution  # Since Laplace transform doesn't have steps like solving an equation
+                    elif "integrate" in equation.lower():
+                        solution = integrate_expression(equation)
+                        steps = solution  # Integration steps are handled inside `integrate_expression`
+                    else:
+                        solution = solve_equation(equation)
+                        steps = solution  # If neither Laplace nor integration, solve normally
+
+
 
                     # Display solution
                     st.markdown("### Solution")