Update app.py

app.py

@@ -130,40 +130,48 @@ with gr.Blocks() as demo:
         sim_button = gr.Button("📌 Solve System")
 
         def solve_system_of_equations(image):
-            try:
-                img = preprocess_handwritten_image(image)
-                raw_latex = model(img)
-                if not raw_latex or len(raw_latex.strip()) < 3:
-                    return "❌ Could not extract valid LaTeX."
-
-                raw_latex = raw_latex.replace("\n", " ")
-                equations = re.split(r'[;,]', raw_latex)
-                cleaned_equations = [clean_latex2(eq) for eq in equations if '=' in eq]
-
-                sympy_eqs = []
-                symbols = set()
-                for eq in cleaned_equations:
-                    parsed = parse_latex(eq)
-                    if isinstance(parsed, sp.Equality):
-                        sympy_eqs.append(parsed)
-                        symbols.update(parsed.free_symbols)
-
-                if len(sympy_eqs) < 2:
-                    return f"❌ Not enough valid equations detected.\n\n```latex\n{raw_latex}\n```"
-
-                sol = sp.solve(sympy_eqs, list(symbols), dict=True)
-                output = f"## 🧾 Detected Equations:\n```latex\n{raw_latex}\n```\n"
-                output += "---\n## ✏️ Parsed & Cleaned:\n"
-                for eq in sympy_eqs:
-                    output += f"$$ {sp.latex(eq)} $$\n"
-                output += "---\n## ✅ Solution:\n"
-                if sol:
-                    output += "$$" + sp.latex(sol[0]) + "$$"
-                else:
-                    output += "❌ No solution found or system may be inconsistent."
-                return output
-            except Exception as e:
-                return f"❌ Error: {str(e)}"
+    try:
+        img = preprocess_handwritten_image(image)
+        raw_latex = model(img)
+        if not raw_latex or len(raw_latex.strip()) < 3:
+            return "❌ Could not extract valid LaTeX."
+
+        # Clean up array environments if present
+        if r"\begin{array" in raw_latex:
+            lines = re.findall(r'\\begin{array}{[^}]+}(.+?)\\end{array}', raw_latex, re.DOTALL)
+            if lines:
+                raw_latex = lines[0]
+            raw_latex = raw_latex.replace(r'\\', '\n')  # line separator
+            raw_latex = re.sub(r'&', '', raw_latex)     # remove LaTeX column markers
+
+        raw_latex = raw_latex.replace("\n", " ")
+        equations = re.split(r'[;,]', raw_latex)
+        cleaned_equations = [clean_latex2(eq) for eq in equations if '=' in eq]
+
+        sympy_eqs = []
+        symbols = set()
+        for eq in cleaned_equations:
+            parsed = parse_latex(eq)
+            if isinstance(parsed, sp.Equality):
+                sympy_eqs.append(parsed)
+                symbols.update(parsed.free_symbols)
+
+        if len(sympy_eqs) < 2:
+            return f"❌ Not enough valid equations detected.\n\n```latex\n{raw_latex}\n```"
+
+        sol = sp.solve(sympy_eqs, list(symbols), dict=True)
+        output = f"## 🧾 Detected Equations:\n```latex\n{raw_latex}\n```\n"
+        output += "---\n## ✏️ Parsed & Cleaned:\n"
+        for eq in sympy_eqs:
+            output += f"$$ {sp.latex(eq)} $$\n"
+        output += "---\n## ✅ Solution:\n"
+        if sol:
+            output += "$$" + sp.latex(sol[0]) + "$$"
+        else:
+            output += "❌ No solution found or system may be inconsistent."
+        return output
+    except Exception as e:
+        return f"❌ Error: {str(e)}"
 
         sim_button.click(fn=solve_system_of_equations, inputs=[sim_image_input], outputs=[sim_output])