Update app.py

app.py

@@ -50,7 +50,7 @@ def clean_latex(latex):
     latex = latex.replace('e', '2.7183')
     return latex
 
-# ✅ Tab 2 specific cleaner (gentle)
+# ✅ Tab 2 specific cleaner
 def clean_latex2(latex):
     latex = latex.replace('\\ ', '')
     latex = latex.replace('\\\\', '\\')
@@ -115,34 +115,21 @@ def wrapped_solver(img):
     result, cleaned, prompt = solve_polynomial(img)
     return result, cleaned, prompt
 
-with gr.Blocks() as demo:
-    with gr.Tab("🖼️ Parse from Image"):
-        image_input = gr.Image(type="pil", label="📷 Upload Image of Polynomial")
-        hidden_latex = gr.Textbox(visible=False)
-        explanation_prompt = gr.Textbox(visible=False)
-        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])
-
-    with gr.Tab("📐 Solve Simultaneous Equations"):
-        sim_image_input = gr.Image(type="pil", label="📷 Upload Image with Simultaneous Equations")
-        sim_output = gr.Markdown(label="📋 Solved System Output")
-        sim_button = gr.Button("📌 Solve System")
-
-        def solve_system_of_equations(image):
+# ✅ FIXED: moved outside of gr.Blocks()
+def solve_system_of_equations(image):
     try:
         img = preprocess_handwritten_image(image)
         raw_latex = model(img)
         if not raw_latex or len(raw_latex.strip()) < 3:
             return "❌ Could not extract valid LaTeX."
 
-        # Clean up array environments if present
+        # Handle array environment
         if r"\begin{array" in raw_latex:
             lines = re.findall(r'\\begin{array}{[^}]+}(.+?)\\end{array}', raw_latex, re.DOTALL)
             if lines:
                 raw_latex = lines[0]
-            raw_latex = raw_latex.replace(r'\\', '\n')  # line separator
-            raw_latex = re.sub(r'&', '', raw_latex)     # remove LaTeX column markers
+            raw_latex = raw_latex.replace(r'\\', '\n')
+            raw_latex = re.sub(r'&', '', raw_latex)
 
         raw_latex = raw_latex.replace("\n", " ")
         equations = re.split(r'[;,]', raw_latex)
@@ -173,6 +160,20 @@ with gr.Blocks() as demo:
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
+# === UI ===
+with gr.Blocks() as demo:
+    with gr.Tab("🖼️ Parse from Image"):
+        image_input = gr.Image(type="pil", label="📷 Upload Image of Polynomial")
+        hidden_latex = gr.Textbox(visible=False)
+        explanation_prompt = gr.Textbox(visible=False)
+        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])
+
+    with gr.Tab("📐 Solve Simultaneous Equations"):
+        sim_image_input = gr.Image(type="pil", label="📷 Upload Image with Simultaneous Equations")
+        sim_output = gr.Markdown(label="📋 Solved System Output")
+        sim_button = gr.Button("📌 Solve System")
         sim_button.click(fn=solve_system_of_equations, inputs=[sim_image_input], outputs=[sim_output])
 
 if __name__ == "__main__":