Update app.py

app.py

@@ -170,6 +170,10 @@ def preprocess_image(image):
 #     except Exception as e:
 #         return f"An error occurred: {e}"
 
+def get_akc_link(breed):
+    formatted_breed = breed.lower().replace(" ", "-")
+    return f"https://www.akc.org/dog-breeds/{formatted_breed}/"
+
 def predict(image):
     try:
         image_tensor = preprocess_image(image)
@@ -185,8 +189,8 @@ def predict(image):
         # Retrieve breed description
         description = get_dog_description(breed)
         
-        # Generate Wikipedia link
-        wiki_link = f"https://en.wikipedia.org/wiki/Special:Search?search={quote(breed)}+dog"
+        # Generate AKC link
+        akc_link = get_akc_link(breed)
         
         # Formatting the description for better display
         if isinstance(description, dict):
@@ -202,8 +206,8 @@ def predict(image):
         else:
             description_str = description
         
-        # Add Wikipedia link
-        description_str += f"\n\n[Click here to search for {breed} on Wikipedia]({wiki_link})"
+        # Add AKC link
+        description_str += f"\n\n[Click here for more information about {breed} on AKC]({akc_link})"
         
         return description_str
     except Exception as e: