Update app.py

app.py

@@ -126,8 +126,15 @@ def preprocess_image(image):
     return transform(image).unsqueeze(0)
 
 
-def get_akc_breeds_link():
-    return "https://www.akc.org/dog-breeds/"
+# def get_akc_breeds_link():
+#     return "https://www.akc.org/dog-breeds/"
+
+
+def get_akc_breeds_link(breed):
+    base_url = "https://www.akc.org/dog-breeds/"
+    breed_url = breed.lower().replace('_', '-')
+    return f"{base_url}{breed_url}/"
+
 
 
 async def predict_single_dog(image):
@@ -373,6 +380,21 @@ def show_details_html(choice, previous_output, initial_state):
         return f"<p style='color: red;'>{error_msg}</p>", gr.update(visible=True), initial_state
 
 
+# def format_description_html(description, breed):
+#     html = "<ul style='list-style-type: none; padding-left: 0;'>"
+#     if isinstance(description, dict):
+#         for key, value in description.items():
+#             html += f"<li style='margin-bottom: 10px;'><strong>{key}:</strong> {value}</li>"
+#     elif isinstance(description, str):
+#         html += f"<li>{description}</li>"
+#     else:
+#         html += f"<li>No description available for {breed}</li>"
+#     html += "</ul>"
+#     akc_link = get_akc_breeds_link()
+#     html += f'<p><a href="{akc_link}" target="_blank">Learn more about {breed} on the AKC website</a></p>'
+#     return html
+
+
 def format_description_html(description, breed):
     html = "<ul style='list-style-type: none; padding-left: 0;'>"
     if isinstance(description, dict):
@@ -383,11 +405,12 @@ def format_description_html(description, breed):
     else:
         html += f"<li>No description available for {breed}</li>"
     html += "</ul>"
-    akc_link = get_akc_breeds_link()
+    akc_link = get_akc_breeds_link(breed)
     html += f'<p><a href="{akc_link}" target="_blank">Learn more about {breed} on the AKC website</a></p>'
     return html
 
 
+
 def go_back(state):
     buttons = state.get("buttons", [])
     return (