Update app.py

app.py

@@ -359,7 +359,7 @@ async def predict(image):
         dogs = await detect_multiple_dogs(image)
         
         color_list = ['#FF0000', '#00FF00', '#0000FF', '#FFFF00', '#00FFFF', '#FF00FF', '#800080', '#FFA500']
-        buttons = []
+        all_buttons = []
         annotated_image = image.copy()
         draw = ImageDraw.Draw(annotated_image)
         font = ImageFont.load_default()
@@ -383,6 +383,7 @@ async def predict(image):
         
             elif combined_confidence >= 0.15:
                 dogs_info += f"<p>Top 3 possible breeds:</p><ul>"
+                dog_buttons = []  # Reset buttons for each dog
                 for j, (breed, prob) in enumerate(zip(topk_breeds[:3], topk_probs_percent[:3])):
                     prob = float(prob.replace('%', ''))
                     button_id = f"Dog {i+1}: More about {breed}"
@@ -392,8 +393,9 @@ async def predict(image):
                         <button class="breed-button" onclick="handle_button_click('{button_id}')">{breed}</button>
                     </li>
                     """
-                    buttons.append(button_id)
+                    dog_buttons.append(button_id)
                 dogs_info += "</ul>"
+                all_buttons.extend(dog_buttons)  # Add this dog's buttons to the overall list
         
             else:
                 dogs_info += "<p>The image is unclear or the breed is not in the dataset. Please upload a clearer image.</p>"
@@ -421,13 +423,13 @@ async def predict(image):
 
         initial_state = {
             "dogs_info": dogs_info,
-            "buttons": buttons,
+            "buttons": all_buttons,
             "show_back": True,
             "image": annotated_image,
             "is_multi_dog": len(dogs) > 1,
             "html_output": html_output
         }
-        return html_output, annotated_image, gr.update(visible=True, choices=buttons), initial_state
+        return html_output, annotated_image, gr.update(visible=True, choices=all_buttons), initial_state
 
     except Exception as e:
         error_msg = f"An error occurred: {str(e)}\n\nTraceback:\n{traceback.format_exc()}"
@@ -440,7 +442,6 @@ def show_details_html(choice, previous_output, initial_state):
         return previous_output, gr.update(visible=True), initial_state
 
     try:
-        # Extract the breed name from the choice string
         breed = choice.split("More about ")[-1]
         description = get_dog_description(breed)
         formatted_description = format_description_html(description, breed)
@@ -461,34 +462,6 @@ def show_details_html(choice, previous_output, initial_state):
         print(error_msg)
         return f"<p style='color: red;'>{error_msg}</p>", gr.update(visible=True), initial_state
 
-        
-
-
-# def show_details_html(choice, previous_output, initial_state):
-#     if not choice:
-#         return previous_output, gr.update(visible=True), initial_state
-
-#     try:
-#         breed = choice.split("More about ")[-1]
-#         description = get_dog_description(breed)
-#         formatted_description = format_description_html(description, breed)
-        
-#         html_output = f"""
-#         <div class="dog-info">
-#             <h2>{breed}</h2>
-#             {formatted_description}
-#         </div>
-#         """
-        
-#         initial_state["current_description"] = html_output
-#         initial_state["original_buttons"] = initial_state.get("buttons", [])
-        
-#         return html_output, gr.update(visible=True), initial_state
-#     except Exception as e:
-#         error_msg = f"An error occurred while showing details: {e}"
-#         print(error_msg)
-#         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;'>"