Update image.py

image.py

@@ -8,6 +8,8 @@ import re
 import io
 import logging
 from llm_utils import explain_with_llm  # ✅ Added for LLM explanation
+from solver import solve_polynomial as solve_poly_from_solver, solve_linear_system as solve_sys_from_solver
+
 
 # Configure logging for debugging
 logging.basicConfig(level=logging.INFO)
@@ -40,6 +42,13 @@ def clean_latex_expression(latex_str):
     if '=' in latex_str:
         left, right = latex_str.split('=')
         latex_str = f"{left} - ({right})"  # Move right-hand side to left
+
+        # Add missing multiplication: (expr)x → (expr)*x
+    latex_str = re.sub(r'(\))([a-zA-Z])', r'\1*\2', latex_str)
+    
+    # Add * between number/symbol and parenthesis: e.g., 3( → 3*(...
+    latex_str = re.sub(r'(\d|\w)\(', r'\1*(', latex_str)
+
     return latex_str.strip()
 
 def parse_equation_type(latex_str):
@@ -295,14 +304,15 @@ def solve_linear_system_from_coeffs(eq1_str, eq2_str):
 
 def solve_extracted_equation(eq_data, real_only):
     if eq_data["type"] == "polynomial":
-        return solve_polynomial(eq_data["degree"], eq_data["coeffs"], real_only)
+        return solve_poly_from_solver(eq_data["degree"], eq_data["coeffs"])
     elif eq_data["type"] == "linear":
         return "❌ Single linear equation not supported. Please upload a system of equations.", None, ""
     elif eq_data["type"] == "linear_system":
-        return solve_linear_system_from_coeffs(eq_data["eq1_coeffs"], eq_data["eq2_coeffs"])
+        return solve_sys_from_solver(eq_data["eq1_coeffs"], eq_data["eq2_coeffs"])
     else:
         return "❌ Unknown equation type", None, ""
 
+
 def image_tab():
     """Create the Image Upload Solver tab"""
     with gr.Tab("Image Upload Solver"):