Update app.py

app.py

@@ -1,4 +1,4 @@
-import gradio as gr
+import streamlit as st
 from PIL import Image
 from pix2tex.cli import LatexOCR
 import sympy as sp
@@ -7,17 +7,20 @@ import re
 import cv2
 import numpy as np
 
-# Preprocessing for handwritten image
 def preprocess_handwritten_image(pil_img):
-    img = np.array(pil_img.convert('L'))  # Grayscale
+    img = np.array(pil_img.convert('L'))  # Convert to grayscale
     _, binary = cv2.threshold(img, 0, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)
     denoised = cv2.medianBlur(binary, 3)
-    return Image.fromarray(255 - denoised)  # Text black on white
+    return Image.fromarray(255 - denoised)  # Make text black on white background
 
-# Load Pix2Tex model (once)
-model = LatexOCR()
+# Load the model once using Streamlit's caching
+@st.cache_resource
+def load_model():
+    return LatexOCR()
 
-# Clean LaTeX output
+model = load_model()
+
+# Function to clean the LaTeX output
 def clean_latex(latex):
     latex = re.sub(r'\\(cal|mathcal)\s*X', 'x', latex)
     latex = latex.replace('{', '').replace('}', '')
@@ -26,51 +29,56 @@ def clean_latex(latex):
         latex += '=0'
     return latex
 
-# Main function
-def solve_polynomial(image):
-    try:
-        img = preprocess_handwritten_image(image)
+# Streamlit UI
+st.title("🧠 Polynomial Solver from Image")
+st.write("Upload an image with a polynomial (e.g., $x^3 + 4x^2 + x - 6$).")
+
+uploaded_file = st.file_uploader("Upload Image", type=["png", "jpg", "jpeg"])
+
+if uploaded_file:
+    # Display the uploaded image
+    raw_img = Image.open(uploaded_file)
+    img = preprocess_handwritten_image(raw_img)
+    st.image(img, caption="Preprocessed Image", use_column_width=True)
+
+    with st.spinner("🔍 Extracting LaTeX from image..."):
         latex_result = model(img)
         cleaned_latex = clean_latex(latex_result)
 
+    st.subheader("📄 Extracted LaTeX")
+    st.code(latex_result, language="latex")
+
+    st.subheader("🧹 Cleaned LaTeX Used")
+    st.code(cleaned_latex, language="latex")
+
+    # Try to parse and solve
+    try:
         expr = parse_latex(cleaned_latex)
-        output = f"### 📄 Extracted LaTeX\n`{latex_result}`\n\n"
-        output += f"### 🧹 Cleaned LaTeX Used\n`{cleaned_latex}`\n\n"
-        output += f"### 🧠 Parsed Expression\n${sp.latex(expr)}$\n\n"
+        st.subheader("🧠 Parsed Expression")
+        st.latex(sp.latex(expr))
 
         if isinstance(expr, sp.Equality):
-            lhs = expr.lhs - expr.rhs
-            output += "### ✏️ Step 1: Convert to Polynomial Equation\n"
-            output += f"${sp.latex(lhs)} = 0$\n\n"
+            lhs = expr.lhs - expr.rhs  # Convert equation to f(x) = 0
+
+            st.markdown("---")
+            st.markdown("### ✏️ Step 1: Convert to Polynomial Equation")
+            st.latex(f"{sp.latex(lhs)} = 0")
 
-            output += "### 🧩 Step 2: Factor the Polynomial\n"
+            st.markdown("### 🧩 Step 2: Factor the Polynomial")
             factored = sp.factor(lhs)
-            output += f"${sp.latex(factored)} = 0$\n\n"
+            st.latex(f"{sp.latex(factored)} = 0")
 
-            output += "### ✅ Step 3: Solve for Roots\n"
+            st.markdown("### ✅ Step 3: Solve for Roots")
             roots = sp.solve(sp.Eq(lhs, 0), dict=True)
-            for i, sol in enumerate(roots, 1):
+            clean_roots = [{str(k): v} for r in roots for k, v in r.items()]
+            for i, sol in enumerate(clean_roots, 1):
                 for var, val in sol.items():
-                    output += f"**Root {i}:** ${var} = {sp.latex(val)}$\n\n"
+                    st.latex(f"{var} = {sp.latex(val)}")
+
         else:
             simplified = sp.simplify(expr)
-            output += "### ➕ Simplified Expression\n"
-            output += f"${sp.latex(simplified)}$"
-
-        return output
+            st.subheader("🧮 Simplified Expression")
+            st.latex(sp.latex(simplified))
 
     except Exception as e:
-        return f"❌ Error: {str(e)}"
-
-# Gradio UI
-demo = gr.Interface(
-    fn=solve_polynomial,
-    inputs=gr.Image(type="pil", label="📷 Upload Image of Polynomial"),
-    outputs=gr.Markdown(label="📋 Step-by-step Solution"),
-    title="🧠 Polynomial Solver from Image",
-    description="Upload an image of a polynomial (typed or handwritten). The app will extract, solve, and explain it step-by-step.",
-    allow_flagging="never"
-)
-
-if __name__ == "__main__":
-    demo.launch()
+        st.error(f"❌ Error: {e}")