Update app.py

app.py

@@ -32,6 +32,8 @@ def clean_latex(latex):
     latex = latex.replace('−', '-')
     return latex.strip()
 
+# ------------------ Tab 1: Polynomial Solver ------------------
+
 def solve_polynomial(image):
     try:
         img = preprocess_handwritten_image(image)
@@ -76,7 +78,7 @@ def wrapped_solver(img):
     result, cleaned, prompt = solve_polynomial(img)
     return result, cleaned, prompt
 
-# ---------- Tab 2 Logic: Solve Simultaneous Linear Equations ----------
+# ------------------ Tab 2: Simultaneous Equations ------------------
 
 def solve_simultaneous_equations(img):
     try:
@@ -84,14 +86,14 @@ def solve_simultaneous_equations(img):
         latex_result = model(img)
         cleaned_latex = clean_latex(latex_result)
 
-        # Extract lines: fix missing `\\` from array or aligned environments
+        # Extract lines from environments OR fallback to splitting by '='
         if 'beginarray' in cleaned_latex or 'begin{aligned}' in cleaned_latex:
             body = re.sub(r'\\?begin\{?array\}?|\\?end\{?array\}?', '', cleaned_latex)
-            if '\\\\' not in cleaned_latex:
-                # Fallback: separate equations if `\\` missing
-                lines = re.findall(r'[^=]+=.*?(?=[a-zA-Z]|$)', body)
-            else:
+            if '\\\\' in cleaned_latex:
                 lines = re.findall(r'(.*?)\\\\', body + '\\\\')
+            else:
+                # fallback: split into equations using = and variable heuristics
+                lines = re.findall(r'[^\n=]+=.*?(?=\s*[a-zA-Z]|$)', body)
         else:
             lines = [line.strip() for line in cleaned_latex.split('\n') if line.strip()]