Update app.py

app.py

@@ -54,35 +54,26 @@ def clean_latex(latex):
 
 def clean_latex2(latex):
     latex = latex.replace('\\ ', '')
-    latex = latex.replace('\\\\', '\n')
-    latex = latex.replace('~', '')
+    latex = latex.replace('\\\\', '\\')
     latex = latex.replace('−', '-').replace('–', '-')
-
-    # Replace math operators
     latex = re.sub(r'\\times', '*', latex)
     latex = re.sub(r'\\cdot', '*', latex)
 
-    # Strip unwanted styles
+    # Strip LaTeX wrappers
     latex = re.sub(r'\\(text|mathbf|mathrm|mathit|textbf|mathcal|cal)\s*\{([^{}]+)\}', r'\2', latex)
-    latex = re.sub(r'\\[a-zA-Z]+', '', latex)  # remove leftover LaTeX commands
-
-    # Remove symbols that interfere
+    latex = re.sub(r'\\[a-zA-Z]+\s*', '', latex)  # leftover LaTeX commands
     latex = latex.replace('{', '').replace('}', '').replace('&', '')
 
-    # Fix splitting
+    # Normalize spacing and equation delimiters
     latex = re.sub(r'[;,]', '\n', latex)
     latex = re.sub(r'=+', '=', latex)
-
-    # Clean extra space
     latex = re.sub(r'\s+', ' ', latex).strip()
 
-    # OCR character corrections
+    # Fix OCR misreads
     latex = re.sub(r'\bchi\b|\bxi\b|\balpha\b', 'x', latex)
     latex = re.sub(r'\bbeta\b', 'b', latex)
     latex = re.sub(r'\bgamma\b', 'y', latex)
 
-    # Insert * where missing
-    latex = re.sub(r'(\d)([a-zA-Z])', r'\1*\2', latex)
     return latex
 
 def solve_polynomial(image):
@@ -135,19 +126,17 @@ def wrapped_solver(img):
 
 def solve_system_of_equations(image):
     try:
-        if isinstance(image, np.ndarray):
-            image = Image.fromarray(image)
         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."
 
-        # Extract inside array block
+        # Handle \array environment
         if r"\begin{array" in raw_latex:
-            match = re.search(r'\\begin{array}{[^}]+}(.+?)\\end{array}', raw_latex, re.DOTALL)
-            if match:
-                raw_latex = match.group(1)
+            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')
             raw_latex = re.sub(r'&', '', raw_latex)
 
@@ -162,7 +151,7 @@ def solve_system_of_equations(image):
                     parsed = parse_latex(cleaned)
                     if isinstance(parsed, sp.Equality):
                         cleaned_equations.append(cleaned)
-                except Exception:
+                except:
                     continue
 
         sympy_eqs = []
@@ -190,6 +179,7 @@ def solve_system_of_equations(image):
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
+
         
 # === UI ===
 with gr.Blocks() as demo: