Update app.py

app.py

@@ -578,7 +578,7 @@ async def predict(image):
                 "image": annotated_image,
                 "is_multi_dog": len(dogs) > 1,
                 "dogs_info": html_output,
-                "original_html": html_output  # 新增這行
+                #"original_html": html_output  # 新增這行
             }
             return html_output, annotated_image, gr.update(visible=True, choices=buttons), initial_state
         else:
@@ -616,7 +616,7 @@ def show_details_html(choice, previous_output, initial_state):
         """
         
         initial_state["current_description"] = html_output
-        initial_state["original_buttons"] = initial_state.get("buttons", [])
+        #initial_state["original_buttons"] = initial_state.get("buttons", [])
         
         return html_output, gr.update(visible=True), initial_state
     except Exception as e:
@@ -643,8 +643,8 @@ def format_description_html(description, breed):
 def go_back(state):
     buttons = state.get("buttons", [])
     return (
-        #state["dogs_info"],  # 這裡應該是完整的 HTML 內容
-        state["original_html"],
+        state["dogs_info"],  # 這裡應該是完整的 HTML 內容
+        #state["original_html"],
         state["image"],
         gr.update(visible=True, choices=buttons),
         gr.update(visible=False),