Create app.py

app.py

@@ -0,0 +1,92 @@
+import gradio as gr
+from PIL import Image
+from pix2tex.cli import LatexOCR
+import sympy as sp
+from sympy.parsing.latex import parse_latex
+import re
+
+model = LatexOCR()
+
+def preprocess_handwritten_image(pil_img):
+    return pil_img.convert('RGB')
+
+def clean_latex2(latex):
+    latex = latex.replace('\\ ', '')
+    latex = latex.replace('\\\\', '\\')
+    latex = re.sub(r'\\[ \t\n\r\f\v]*', '', latex)
+    latex = latex.replace(r'\times', 'x')
+    latex = latex.replace(r'\cdot', '*')
+    latex = latex.replace('−', '-').replace('–', '-')
+    latex = re.sub(r'\\(text|mathbf|mathrm|mathit|textbf|mathcal|cal)\s*\{([^{}]+)\}', r'\2', latex)
+    latex = re.sub(r'\\[a-zA-Z]+', '', latex)
+    latex = re.sub(r'\{+\}+', '', latex)
+    latex = re.sub(r'\{([^\{\}]*)\}', r'\1', latex)
+    latex = re.sub(r'[;,]', '\n', latex)
+    latex = re.sub(r'[^\w\s=+\-*/().]', '', latex)
+    latex = re.sub(r'\s+', ' ', latex)
+    latex = re.sub(r'(?<![=<>])=(?![=<>])', ' = ', latex)
+    replacements = {
+        'chi': 'x', 'xi': 'x', 'alpha': 'x', 'beta': 'b',
+        'gamma': 'y', 'vartheta': '3', 'mathcal': '',
+        'cal': '', 'mathrm': ''
+    }
+    for wrong, right in replacements.items():
+        latex = re.sub(wrong, right, latex)
+    return latex.strip()
+
+def solve_from_multiple_images(img1, img2, img3):
+    try:
+        images = [img for img in [img1, img2, img3] if img is not None]
+        if len(images) < 2:
+            return "❌ Please upload at least 2 images (each with one equation)."
+
+        raw_latexes = []
+        cleaned_eqs = []
+        for img in images:
+            img = preprocess_handwritten_image(img)
+            latex = model(img)
+            raw_latexes.append(latex)
+            cleaned = clean_latex2(latex)
+            if '=' in cleaned:
+                try:
+                    parsed = parse_latex(cleaned)
+                    if isinstance(parsed, sp.Equality):
+                        cleaned_eqs.append(parsed)
+                except:
+                    continue
+
+        if len(cleaned_eqs) < 2:
+            return "❌ Could not parse enough valid equations."
+
+        symbols = set()
+        for eq in cleaned_eqs:
+            symbols.update(eq.free_symbols)
+
+        solution = sp.solve(cleaned_eqs, list(symbols), dict=True)
+        output = "## 🧾 Extracted Equations\n"
+        for latex in raw_latexes:
+            output += f"```latex\n{latex}\n```\n"
+        output += "---\n## ✏️ Parsed & Cleaned:\n"
+        for eq in cleaned_eqs:
+            output += f"$$ {sp.latex(eq)} $$\n"
+        output += "---\n## ✅ Solution:\n"
+        if solution:
+            output += "$$" + sp.latex(solution[0]) + "$$"
+        else:
+            output += "❌ No solution found or system is inconsistent."
+        return output
+    except Exception as e:
+        return f"❌ Error: {str(e)}"
+
+with gr.Blocks() as demo:
+    gr.Markdown("## 📐 Solve System from Multiple Equation Images")
+    with gr.Row():
+        img1 = gr.Image(type="pil", label="📷 Equation 1")
+        img2 = gr.Image(type="pil", label="📷 Equation 2")
+        img3 = gr.Image(type="pil", label="📷 (Optional) Equation 3")
+    solve_btn = gr.Button("📌 Solve System")
+    output = gr.Markdown()
+    solve_btn.click(fn=solve_from_multiple_images, inputs=[img1, img2, img3], outputs=output)
+
+if __name__ == "__main__":
+    demo.launch()