Update app.py

app.py

@@ -4,39 +4,48 @@ from pix2tex.cli import LatexOCR
 import sympy as sp
 from sympy.parsing.latex import parse_latex
 import re
-import cv2
-import numpy as np
 
 # Preprocessing for handwritten image
 def preprocess_handwritten_image(pil_img):
-    img = np.array(pil_img.convert('L'))  # Grayscale
-    _, binary = cv2.threshold(img, 0, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)
-    denoised = cv2.medianBlur(binary, 3)
-    return Image.fromarray(255 - denoised)  # Text black on white
+    return pil_img.convert('RGB')  # Minimal processing, just convert to RGB
+
 
 # Load Pix2Tex model (once)
 model = LatexOCR()
 
 # Clean LaTeX output
 def clean_latex(latex):
-    # Replace mathcal X with x
+    # Replace \mathcal{X} or \cal X with 'x'
     latex = re.sub(r'\\(cal|mathcal)\s*X', 'x', latex)
 
-    # Remove unwanted characters
+    # Remove curly braces
     latex = latex.replace('{', '').replace('}', '')
     latex = latex.strip().rstrip(',.')
     
-    # Replace \pi and the Unicode character π with 3.1416
+    # Replace coefficients like 5\pi with (5*3.1416)
+    latex = re.sub(r'(\d+)\s*\\pi', r'(\1*3.1416)', latex)
     latex = latex.replace(r'\pi', '3.1416')
-    latex = latex.replace('π', '3.1416')
 
-    # Replace standalone 'e' with 2.7183 (not part of other variable names)
+    # 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)
 
-    # Append =0 if not present
+    # Insert * between number and variable (e.g., 45x → 45*x)
+    latex = re.sub(r'(\d)([a-zA-Z])', r'\1*\2', latex)
+
+    # Replace number followed by i with number*I
+    latex = re.sub(r'(\d+)\s*i', r'\1*I', latex)
+
+    # Replace standalone i with I
+    latex = re.sub(r'(?<![a-zA-Z0-9])i(?![a-zA-Z0-9])', 'I', 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
     if '=' not in latex:
         latex += '=0'
-    
+
     return latex
 
 
@@ -77,8 +86,6 @@ def solve_polynomial(image):
                         output += f"\\text{{Root {i}}}:\\quad {var} &= {sp.latex(val)} \\\\\n"
                 output += "\\end{aligned}\n$$\n"
 
-
-
         else:
             simplified = sp.simplify(expr)
             output += "## ➕ Simplified Expression\n"
@@ -101,4 +108,4 @@ demo = gr.Interface(
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()