Update app.py

app.py

@@ -62,7 +62,7 @@ def clean_latex2(latex):
     latex = re.sub(r'\\text\s*\{.*?\}', '', latex)
     latex = re.sub(r'\\mathbf\s*\{.*?\}', '', latex)
     latex = re.sub(r'\\mathrm\s*\{.*?\}', '', latex)
-    latex = re.sub(r'\\[a-zA-Z]+\s*', '', latex)  # kill things like \mathbf, \times, \cdot, \nabla
+    latex = re.sub(r'\\[a-zA-Z]+\s*', '', latex)  # kills \cdot, \nabla, etc.
 
     latex = latex.replace('{', '').replace('}', '')
     latex = latex.replace('−', '-').replace('–', '-')
@@ -75,9 +75,9 @@ def clean_latex2(latex):
     for wrong, right in replacements.items():
         latex = re.sub(wrong, right, latex)
 
-    # Remove remaining junk
-    latex = re.sub(r'[^0-9a-zA-Z=+\-*/().,\n ]', '', latex)
     latex = latex.strip().rstrip(',')
+    latex = re.sub(r'(?<![=<>])=(?![=<>])', ' = ', latex)  # normalize =
+    latex = re.sub(r'[^0-9a-zA-Z=+\-*/().,\n ]', '', latex)
 
     return latex
 
@@ -143,7 +143,18 @@ def solve_system_of_equations(image):
 
         raw_latex = raw_latex.replace("\n", " ")
         equations = re.split(r'[;,]', raw_latex)
-        cleaned_equations = [clean_latex2(eq) for eq in equations if '=' in eq]
+
+        cleaned_equations = []
+        for eq in equations:
+            cleaned = clean_latex2(eq)
+            cleaned = re.sub(r'(?<![=<>])=(?![=<>])', ' = ', cleaned)
+            if '=' in cleaned:
+                try:
+                    parsed = parse_latex(cleaned)
+                    if isinstance(parsed, sp.Equality):
+                        cleaned_equations.append(cleaned)
+                except Exception:
+                    continue
 
         sympy_eqs = []
         symbols = set()
@@ -170,6 +181,7 @@ def solve_system_of_equations(image):
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
+
 # === UI ===
 with gr.Blocks() as demo:
     with gr.Tab("🖼️ Parse from Image"):