Update app.py

app.py

@@ -53,42 +53,38 @@ def clean_latex(latex):
     return latex
 
 def clean_latex2(latex):
-    # Step 1: Replace basic formatting issues
     latex = latex.replace('\\ ', '')
     latex = latex.replace('\\\\', '\n')
     latex = latex.replace('~', '')
     latex = latex.replace('−', '-').replace('–', '-')
 
-    # Step 2: Fix common LaTeX math ops
+    # Replace math operators
     latex = re.sub(r'\\times', '*', latex)
     latex = re.sub(r'\\cdot', '*', latex)
 
-    # Step 3: Strip away wrappers like \mathcal{}, \mathbf{}, etc.
-    latex = re.sub(r'\\(text|mathbf|mathrm|mathit|textbf|mathcal|cal|nabla)\s*\{([^{}]+)\}', r'\2', latex)
+    # Strip unwanted styles
+    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
 
-    # Step 4: Remove leftover backslash commands (\!, \,, etc.)
-    latex = re.sub(r'\\[a-zA-Z]+', '', latex)
-
-    # Step 5: Strip non-math braces, ampersands, etc.
+    # Remove symbols that interfere
     latex = latex.replace('{', '').replace('}', '').replace('&', '')
 
-    # Step 6: Break into lines if multiple equations
+    # Fix splitting
     latex = re.sub(r'[;,]', '\n', latex)
-
-    # Step 7: Fix repeated symbols
-    latex = re.sub(r'-{2,}', '-', latex)
     latex = re.sub(r'=+', '=', latex)
 
-    # Step 8: Symbol OCR replacements
+    # Clean extra space
+    latex = re.sub(r'\s+', ' ', latex).strip()
+
+    # OCR character corrections
     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)
 
-    # Step 9: Normalize spacing
-    latex = re.sub(r'\s+', ' ', latex).strip()
-
+    # Insert * where missing
+    latex = re.sub(r'(\d)([a-zA-Z])', r'\1*\2', latex)
     return latex
-    
+
 def solve_polynomial(image):
     try:
         if isinstance(image, np.ndarray):
@@ -147,15 +143,15 @@ def solve_system_of_equations(image):
         if not raw_latex or len(raw_latex.strip()) < 3:
             return "❌ Could not extract valid LaTeX."
 
-        # Extract from array if applicable
+        # Extract inside array block
         if r"\begin{array" in raw_latex:
-            array_match = re.search(r'\\begin{array}{[^}]*}(.+?)\\end{array}', raw_latex, re.DOTALL)
-            if array_match:
-                raw_latex = array_match.group(1)
-            raw_latex = raw_latex.replace('\\\\', '\n')
-            raw_latex = raw_latex.replace('&', '')
+            match = re.search(r'\\begin{array}{[^}]+}(.+?)\\end{array}', raw_latex, re.DOTALL)
+            if match:
+                raw_latex = match.group(1)
+            raw_latex = raw_latex.replace(r'\\', '\n')
+            raw_latex = re.sub(r'&', '', raw_latex)
 
-        raw_latex = raw_latex.replace('\n', ' ')
+        raw_latex = raw_latex.replace("\n", " ")
         equations = re.split(r'[;,]', raw_latex)
 
         cleaned_equations = []
@@ -181,7 +177,6 @@ def solve_system_of_equations(image):
             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:
@@ -194,6 +189,7 @@ def solve_system_of_equations(image):
         return output
     except Exception as e:
         return f"❌ Error: {str(e)}"
+
         
 # === UI ===
 with gr.Blocks() as demo: