Update app.py

app.py

@@ -10,6 +10,7 @@ model = LatexOCR()
 def preprocess_handwritten_image(pil_img):
     return pil_img.convert('RGB')
 
+# Tab 1 cleaning (KEEP UNCHANGED)
 def clean_latex(latex):
     latex = latex.replace('\\ ', '')
     latex = latex.replace('\\\\', '\\')
@@ -49,6 +50,27 @@ def clean_latex(latex):
     latex = latex.replace('e', '2.7183')
     return latex
 
+# ✅ Tab 2 specific cleaner (gentle)
+def clean_latex2(latex):
+    latex = latex.replace('\\ ', '')
+    latex = latex.replace('\\\\', '\\')
+    latex = re.sub(r'\\[ \t\n\r\f\v]*', '', latex)
+    latex = re.sub(r'\\([+\-=])', r'\1', latex)
+
+    replacements = {
+        r'\\chi': 'x', r'\\xi': 'x',
+        r'\\gamma': 'y', r'\\beta': 'b',
+        r'\\alpha': 'x', r'\\vartheta': '3',
+        r'\\cdot': '*', r'−': '-', r'–': '-',
+        r'\\mathcal': '', r'mathcal': '',
+    }
+    for wrong, right in replacements.items():
+        latex = re.sub(wrong, right, latex)
+
+    latex = latex.replace('{', '').replace('}', '')
+    latex = latex.strip().rstrip(',')
+    return latex
+
 def solve_polynomial(image):
     try:
         img = preprocess_handwritten_image(image)
@@ -93,12 +115,6 @@ def wrapped_solver(img):
     result, cleaned, prompt = solve_polynomial(img)
     return result, cleaned, prompt
 
-# def solve_from_coeffs(degree, coeff_str):
-#     ...
-
-# def request_llm_explanation(prompt, llm_url):
-#     ...
-
 with gr.Blocks() as demo:
     with gr.Tab("🖼️ Parse from Image"):
         image_input = gr.Image(type="pil", label="📷 Upload Image of Polynomial")
@@ -107,12 +123,6 @@ with gr.Blocks() as demo:
         output_box = gr.Markdown(label="📋 Step-by-step Solution")
         submit_btn = gr.Button("🔍 Solve")
         submit_btn.click(fn=wrapped_solver, inputs=[image_input], outputs=[output_box, hidden_latex, explanation_prompt])
-        # explain_box = gr.Markdown(label="🗣️ Human-style Explanation")
-        # explain_btn = gr.Button("🧠 Explain Human-Solution")
-        # explain_btn.click(fn=request_llm_explanation, inputs=[explanation_prompt, llm_url], outputs=explain_box)
-
-    # with gr.Tab("🧮 Solve by Coefficients"):
-    #     ...
 
     with gr.Tab("📐 Solve Simultaneous Equations"):
         sim_image_input = gr.Image(type="pil", label="📷 Upload Image with Simultaneous Equations")
@@ -128,7 +138,7 @@ with gr.Blocks() as demo:
 
                 raw_latex = raw_latex.replace("\n", " ")
                 equations = re.split(r'[;,]', raw_latex)
-                cleaned_equations = [clean_latex(eq) for eq in equations if '=' in eq]
+                cleaned_equations = [clean_latex2(eq) for eq in equations if '=' in eq]
 
                 sympy_eqs = []
                 symbols = set()