Update app.py

app.py

@@ -1,4 +1,4 @@
-import streamlit as st
+import gradio as gr
 from PIL import Image
 from pix2tex.cli import LatexOCR
 import sympy as sp
@@ -7,20 +7,17 @@ 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'))  # Convert to grayscale
+    img = np.array(pil_img.convert('L'))  # Grayscale
     _, binary = cv2.threshold(img, 0, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)
     denoised = cv2.medianBlur(binary, 3)
-    return Image.fromarray(255 - denoised)  # Make text black on white background
+    return Image.fromarray(255 - denoised)  # Text black on white
 
-# Load the model once using Streamlit's caching
-@st.cache_resource
-def load_model():
-    return LatexOCR()
+# Load Pix2Tex model (once)
+model = LatexOCR()
 
-model = load_model()
-
-# Function to clean the LaTeX output
+# Clean LaTeX output
 def clean_latex(latex):
     latex = re.sub(r'\\(cal|mathcal)\s*X', 'x', latex)
     latex = latex.replace('{', '').replace('}', '')
@@ -29,56 +26,52 @@ def clean_latex(latex):
         latex += '=0'
     return latex
 
-# 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..."):
+# Main function
+def solve_polynomial(image):
+    try:
+        img = preprocess_handwritten_image(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)
-        st.subheader("🧠 Parsed Expression")
-        st.latex(sp.latex(expr))
+        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"
 
         if isinstance(expr, sp.Equality):
-            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")
+            lhs = expr.lhs - expr.rhs
+            output += "### ✏️ Step 1: Convert to Polynomial Equation\n"
+            output += f"${sp.latex(lhs)} = 0$\n\n"
 
-            st.markdown("### 🧩 Step 2: Factor the Polynomial")
+            output += "### 🧩 Step 2: Factor the Polynomial\n"
             factored = sp.factor(lhs)
-            st.latex(f"{sp.latex(factored)} = 0")
+            output += f"${sp.latex(factored)} = 0$\n\n"
 
-            st.markdown("### ✅ Step 3: Solve for Roots")
+            output += "### ✅ Step 3: Solve for Roots\n"
             roots = sp.solve(sp.Eq(lhs, 0), dict=True)
-            clean_roots = [{str(k): v} for r in roots for k, v in r.items()]
-            for i, sol in enumerate(clean_roots, 1):
+            for i, sol in enumerate(roots, 1):
                 for var, val in sol.items():
-                    st.latex(f"{var} = {sp.latex(val)}")
-
+                    output += f"**Root {i}:** ${var} = {sp.latex(val)}$\n\n"
         else:
             simplified = sp.simplify(expr)
-            st.subheader("🧮 Simplified Expression")
-            st.latex(sp.latex(simplified))
+            output += "### ➕ Simplified Expression\n"
+            output += f"${sp.latex(simplified)}$"
+
+        return output
 
     except Exception as e:
-        st.error(f"❌ Error: {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()
+