Update agent.py

agent.py

@@ -247,7 +247,7 @@ class ExcelReaderTool(Tool):
 
 
 
-import pytesseract
+
 
 @tool
 def PNG2FENTool(png_file: str) -> str:
@@ -265,22 +265,37 @@ def PNG2FENTool(png_file: str) -> str:
 #        If the PNG file cannot be processed or does not contain a valid chess board.
 
     try:
-        # Open the PNG file using PIL
-        image = Image.open(png_file)
-
-        # Use pytesseract to extract text from the image
-        text = pytesseract.image_to_string(image)
-
-        # Process the extracted text to get the FEN position string
-        fen_position = process_text_to_fen(text)
-
-        return fen_position
-
+        import easyocr
+        reader = easyocr.Reader(['en'])
+        result = reader.readtext(png_file, detail=0)
+        fen_candidates = [text for text in result if validate_fen_format(text)]
+        
+        if not fen_candidates:
+            raise ValueError("No valid FEN found in image")
+            
+        return fen_candidates[0]
+        
+    except Exception as e:
+        raise ValueError(f"OCR processing failed: {str(e)}")
+    
+    
+#    try:
+#        # Open the PNG file using PIL
+#        image = Image.open(png_file)
+#
+#        # Use pytesseract to extract text from the image
+#        text = pytesseract.image_to_string(image)
+#
+#        # Process the extracted text to get the FEN position string
+#        fen_position = process_text_to_fen(text)
+#
+#        return fen_position
+#
     except FileNotFoundError:
         raise FileNotFoundError("PNG file not found.")
-
-    except Exception as e:
-        raise ValueError("Error processing PNG file: " + str(e))
+#
+#    except Exception as e:
+#        raise ValueError("Error processing PNG file: " + str(e))
 
 def process_text_to_fen(text):
     """