Update app.py

app.py

@@ -7,9 +7,10 @@ import re
 
 # Preprocessing for handwritten image
 def preprocess_handwritten_image(pil_img):
-    return pil_img.convert('RGB')
+    return pil_img.convert('RGB')  # Minimal processing, just convert to RGB
 
-# Load Pix2Tex model
+
+# Load Pix2Tex model (once)
 model = LatexOCR()
 
 # Clean LaTeX output
@@ -25,7 +26,7 @@ def clean_latex(latex):
     latex = re.sub(r'(\d+)\s*\\pi', r'(\1*3.1416)', latex)
     latex = latex.replace(r'\pi', '3.1416')
 
-    # Replace coefficients like 4e with (4*2.7183)
+    # Replace coefficients like 5e with (5*2.7183)
     latex = re.sub(r'(\d+)\s*e', r'(\1*2.7183)', latex)
     latex = re.sub(r'(?<![a-zA-Z0-9])e(?![a-zA-Z0-9])', '2.7183', latex)
 
@@ -41,22 +42,21 @@ def clean_latex(latex):
     # Wrap complex coefficients with variables: (a+bI)x^n → (a+b*I)*x^n
     latex = re.sub(r'\(([^()]+?)\)\s*([xX](\^\d+)?)', r'(\1)*\2', latex)
 
-    # Append =0 if not already present
+    # Append '=0' if not already present
     if '=' not in latex:
         latex += '=0'
 
     return latex
 
-# Main solving function
+
+# Main function
 def solve_polynomial(image):
     try:
         img = preprocess_handwritten_image(image)
         latex_result = model(img)
         cleaned_latex = clean_latex(latex_result)
 
-        # Make I symbolic imaginary unit
-        expr = parse_latex(cleaned_latex, local_dict={'I': sp.I})
-
+        expr = parse_latex(cleaned_latex)
         output = f"## 📄 Extracted LaTeX\n```\n{latex_result}\n```\n"
         output += "---\n"
         output += f"## 🧹 Cleaned LaTeX Used\n```\n{cleaned_latex}\n```\n"
@@ -96,6 +96,7 @@ def solve_polynomial(image):
     except Exception as e:
         return f"❌ **Error**: {str(e)}"
 
+
 # Gradio UI
 demo = gr.Interface(
     fn=solve_polynomial,