Update app.py

app.py

@@ -550,7 +550,7 @@ async def predict(image):
                 for breed in topk_breeds[:3]:
                     button_id = f"Dog {i+1}: More about {breed}"
                     html_output += f'<button class="breed-button" onclick="handle_button_click(\'{button_id}\')">{breed}</button>'
-                    #buttons.append(button_id)
+                    buttons.append(button_id)
                 html_output += '</div>'
                 
             else:
@@ -577,7 +577,8 @@ async def predict(image):
                 "show_back": True,
                 "image": annotated_image,
                 "is_multi_dog": len(dogs) > 1,
-                "dogs_info": html_output
+                "dogs_info": html_output,
+                "original_html": html_output  # 新增這行
             }
             return html_output, annotated_image, gr.update(visible=True, choices=buttons), initial_state
         else:
@@ -642,7 +643,8 @@ def format_description_html(description, breed):
 def go_back(state):
     buttons = state.get("buttons", [])
     return (
-        state["dogs_info"],  # 這裡應該是完整的 HTML 內容
+        #state["dogs_info"],  # 這裡應該是完整的 HTML 內容
+        state["original_html"],
         state["image"],
         gr.update(visible=True, choices=buttons),
         gr.update(visible=False),