Update app.py

try another approach

app.py

@@ -47,16 +47,30 @@ 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):
-    """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
+# def get_image_url(image_type: str):
+#     """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
+#     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
 
+def get_image_url(image_type: str):
+    """Finds an approximate match using simple substring matching."""
+    image_type = image_type.lower()
+    
+    # Check for exact match first
+    if image_type in image_sources:
+        return image_sources[image_type]
+
+    # If no exact match, look for a partial match
+    for key in image_sources:
+        if image_type in key.lower():  # Check if input is a substring of a key
+            return image_sources[key]
+    
+    return None  # No 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."""
     print("Launching Playwright...")