Update app.py

app.py

@@ -84,7 +84,12 @@ def solve_simultaneous_equations(img):
         latex_result = model(img)
         cleaned_latex = clean_latex(latex_result)
 
-        lines = [line.strip() for line in cleaned_latex.splitlines() if line.strip()]
+        # Extract lines: handle environments or normal newlines
+        if 'begin{array}' in cleaned_latex or 'begin{aligned}' in cleaned_latex:
+            lines = re.findall(r'(.*?)\\\\', cleaned_latex + '\\\\')  # Append fake end to catch last line
+        else:
+            lines = [line.strip() for line in cleaned_latex.split('\n') if line.strip()]
+
         if len(lines) < 2:
             return f"❌ Expected at least 2 equations but found {len(lines)}.\n\nExtracted:\n{cleaned_latex}"
 
@@ -92,12 +97,15 @@ def solve_simultaneous_equations(img):
         symbols = set()
 
         for line in lines:
-            parsed = parse_latex(line)
-            if not isinstance(parsed, sp.Equality):
-                return f"❌ Failed to parse line as equation: {line}"
-            lhs = parsed.lhs - parsed.rhs
-            symbols.update(lhs.free_symbols)
-            equations.append(lhs)
+            try:
+                parsed = parse_latex(line)
+                if not isinstance(parsed, sp.Equality):
+                    return f"❌ Failed to parse line as equation: {line}"
+                lhs = parsed.lhs - parsed.rhs
+                symbols.update(lhs.free_symbols)
+                equations.append(lhs)
+            except Exception as e:
+                return f"❌ Error parsing line: `{line}`\n{str(e)}"
 
         if len(symbols) < 2:
             return f"❌ Need at least 2 variables to solve system. Found: {symbols}"