Looking for math functionality

app.py

@@ -8,18 +8,8 @@ import json
 import numpy as np
 from sklearn.preprocessing import LabelEncoder
 from sklearn.feature_extraction.text import CountVectorizer
-import os
-#from flask_wtf.csrf import CSRFProtect
-
-# Load secret key from environment variable
-SECRET_KEY = os.getenv('SECRET_KEY', 'pumpkin')
 
 app = Flask(__name__)
-app.secret_key = SECRET_KEY
-
-# Enable CSRF protection
-#csrf = CSRFProtect(app)
-
 x, y, z = sm.symbols('x y z')
 
 @app.route('/', methods=['GET', 'POST'])
@@ -30,15 +20,55 @@ def index():
     latex_result = ''
 
     if request.method == 'POST':
-        expression = re.sub(r'[^0-9a-zA-Z+\-*/^() ]', '', request.form['expression'])  # Sanitize input
+        expression = request.form['expression']
         operation = request.form['operation']
         
+        # Replace e**x with exp(x) to make sure SymPy interprets it correctly
+        expression = re.sub(r'e\*\*([a-zA-Z0-9_]+)', r'exp(\1)', expression)
+
+        # Insert an asterisk between numbers and letters using regex
+        expression = re.sub(r'(\d)([a-zA-Z])', r'\1*\2', expression)
+
         try:
             expr = sm.sympify(expression)
-            if operation in ['diff', 'expand', 'solve', 'simplify', 'factorize', 'rationalize', 'diffx', 'diffy', 'diffz', 'integratex', 'integratey', 'integratez']:
-                result = getattr(sm, operation)(expr)
-            else:
-                result = "Invalid operation"
+            if operation == 'diff':
+                result = sm.diff(expr, x)
+            elif operation == 'expand':
+                result = sm.expand(expr)
+            elif operation == 'solve':
+                result = sm.solve(expr, x)
+            elif operation == 'simplify':
+                result = sm.simplify(expr)
+            elif operation == 'factorize':
+                result = sm.factor(expr)
+            elif operation == 'rationalize':
+                result = sm.rationalize(expr)
+            elif operation == 'diffx':
+                result = sm.diff(expr, x)
+            elif operation == 'diffy':
+                result = sm.diff(expr, y)
+            elif operation == 'diffz':
+                result = sm.diff(expr, z)
+            elif operation == 'integratex':
+                # Perform the integration
+                integral_result = sm.integrate(expr, x)
+                # Format the result with C after the variables (if not empty)
+                if integral_result != 0:
+                    result = sm.sympify(f"{integral_result} + C")
+                else:
+                    result = "C"  # If the integral is zero, just return C
+            elif operation == 'integratey':
+                integral_result = sm.integrate(expr, y)
+                if integral_result != 0:
+                    result = sm.sympify(f"{integral_result} + C")
+                else:
+                    result = "C"
+            elif operation == 'integratez':
+                integral_result = sm.integrate(expr, z)
+                if integral_result != 0:
+                    result = sm.sympify(f"{integral_result} + C")
+                else:
+                    result = "C"
             
             latex_result = sm.latex(result)
 
@@ -106,12 +136,10 @@ def get_response(message):
             return response
 
 @app.route('/chat', methods=['POST'])
-#@csrf.exempt  # Temporarily disable CSRF for API endpoint (ensure API is secure)
 def chat():
     user_message = request.json['message']
     response = get_response(user_message)
     return jsonify({'response': response})
 
 if __name__ == "__main__":
-    # Use HTTPS in production
-    app.run(port=7860, ssl_context=('cert.pem', 'key.pem'), debug=True)
+    app.run(port=7860, debug=True)