Update app.py

app.py

@@ -47,28 +47,29 @@ 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]
+    else:
+       #return None  # No good match found
+        return "https://editor.p5js.org/kfahn/full/2XD5Y8MiV"
 
-def get_image_url(image_type: str):
-    """Finds an approximate match using simple substring matching."""
-    image_type = image_type.lower()
+# 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]
+#     # 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):