Update app.py

app.py

@@ -84,9 +84,14 @@ def solve_simultaneous_equations(img):
         latex_result = model(img)
         cleaned_latex = clean_latex(latex_result)
 
-        # 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
+        # Extract lines: fix missing `\\` from array or aligned environments
+        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:
+                lines = re.findall(r'(.*?)\\\\', body + '\\\\')
         else:
             lines = [line.strip() for line in cleaned_latex.split('\n') if line.strip()]
 
@@ -139,10 +144,6 @@ with gr.Blocks() as demo:
         output_box = gr.Markdown(label="📋 Step-by-step Solution")
         submit_btn = gr.Button("🔍 Solve")
         submit_btn.click(fn=wrapped_solver, inputs=[image_input], outputs=[output_box, hidden_latex, explanation_prompt])
-        # Human explanation (commented out)
-        # explain_box = gr.Markdown(label="🗣️ Human-style Explanation")
-        # explain_btn = gr.Button("🧠 Explain Human-Solution")
-        # explain_btn.click(fn=request_llm_explanation, inputs=[explanation_prompt, llm_url], outputs=explain_box)
 
     with gr.Tab("🧮 Simultaneous Equations"):
         sim_image_input = gr.Image(type="pil", label="📷 Upload Image with 2 or 3 Linear Equations")