Update app.py

app.py

@@ -1,15 +1,45 @@
-import spacy
-import os
+import re
+import sympy as sp
+import gradio as gr
 
-# Ensure model is downloaded before usage
-def load_spacy_model():
-    if not os.path.exists("en_core_web_sm"):
-        os.system("python -m spacy download en_core_web_sm")
-    return spacy.load("en_core_web_sm")
+# Function to evaluate the mathematical question
+def evaluate_math_question(question):
+    try:
+        # Remove non-mathematical content and keep equations
+        equation = re.sub(r'[^\d\w\-\+\*/\=\^\.\s]', '', question)
+        
+        # Solve the equation using sympy
+        if "=" in equation:
+            lhs, rhs = equation.split("=")
+            equation = sp.Eq(sp.sympify(lhs), sp.sympify(rhs))
+            solution = sp.solve(equation)
+        else:
+            # Just evaluate the expression
+            solution = sp.sympify(equation)
+        
+        return f"The solution is: {solution}"
 
-# Cache the model to avoid reloading on every request
-@st.cache_resource
-def get_nlp_model():
-    return load_spacy_model()
+    except Exception as e:
+        return f"Error solving the equation: {str(e)}"
 
-nlp = get_nlp_model()
+# Function to restrict non-mathematical questions
+def is_math_question(question):
+    if any(op in question for op in "+-*/="):
+        return True
+    return False
+
+def math_chatbot(question):
+    if is_math_question(question):
+        return evaluate_math_question(question)
+    else:
+        return "This chatbot only answers questions related to mathematics. Please ask a mathematical question."
+
+# Create the Gradio interface
+iface = gr.Interface(fn=math_chatbot, 
+                     inputs=gr.Textbox(label="Enter your mathematical question"), 
+                     outputs="text", 
+                     title="Math Chatbot (Open Source)",
+                     description="Ask any mathematical question and get an answer. Non-mathematical questions will be restricted.")
+
+# Launch the Gradio app
+iface.launch()