Update app.py

app.py

@@ -1,45 +1,50 @@
-import re
+import streamlit as st
+from transformers import pipeline
 import sympy as sp
-import gradio as gr
 
-# 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 so it's loaded only once
+@st.cache_resource
+def load_model():
+    # Load an open-source Hugging Face model for natural language processing
+    return pipeline('text-classification', model='mrm8488/t5-base-finetuned-summarize-news')
 
-    except Exception as e:
-        return f"Error solving the equation: {str(e)}"
+# Initialize the model
+nlp_model = load_model()
 
-# Function to restrict non-mathematical questions
+# Function to check if the question is mathematical
 def is_math_question(question):
-    if any(op in question for op in "+-*/="):
+    try:
+        parsed_expr = sp.sympify(question)
         return True
-    return False
+    except (sp.SympifyError, SyntaxError):
+        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."
+# Function to solve mathematical questions using SymPy
+def solve_math_question(question):
+    try:
+        # Parse and solve the mathematical expression
+        solution = sp.solve(sp.sympify(question))
+        return f"The solution is: {solution}"
+    except Exception as e:
+        return f"Error solving the equation: {e}"
+
+# Streamlit UI
+st.title("Math Chatbot (Open Source)")
+st.write("Ask any mathematical question and get an answer. Non-mathematical questions will be restricted.")
 
-# 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.")
+# User input
+question = st.text_input("Enter your mathematical question:")
 
-# Launch the Gradio app
-iface.launch()
+# Processing the input
+if st.button("Submit"):
+    if is_math_question(question):
+        # Solve the mathematical question
+        answer = solve_math_question(question)
+        st.write(f"Answer: {answer}")
+    else:
+        # Filter non-mathematical questions using NLP model
+        nlp_result = nlp_model(question)[0]
+        if nlp_result['label'] == 'Math':
+            st.write("Answer: Processing your math question...")
+        else:
+            st.write("This chatbot only answers questions related to mathematics. Please ask a mathematical question.")