Update app.py

add fuzzy matching

app.py

@@ -10,6 +10,7 @@ from playwright.async_api import async_playwright
 from PIL import Image
 import subprocess
 import json
+from rapidfuzz import process
 import gradio as gr
 
 from tools.final_answer import FinalAnswerTool
@@ -43,8 +44,18 @@ def load_image_sources():
 
 image_sources = load_image_sources()
 
+# def get_image_url(image_type: str):
+#     return image_sources.get(image_type, None)  
+
 def get_image_url(image_type: str):
-    return image_sources.get(image_type, None)  
+    """Finds the best match for the given image type using fuzzy matching."""
+    choices = list(image_sources.keys())  # Get all available keys
+    best_match, score = process.extractOne(image_type, choices)  # Find closest match
+    
+    if score > 70:  # Set a threshold to ensure a reasonable match
+        return image_sources[best_match].get(image_type, None)
+    else:
+        return None  # No good match found
 
 async def capture_screenshot(image_type: str):
     """Launches Playwright and uses user input,  if any, to captures a screenshot of an image from p5.js."""