Update app.py

app.py

@@ -455,8 +455,11 @@ def show_details(choice, previous_output, initial_state):
         breed = choice.split("More about ")[-1]
         description = get_dog_description(breed)
         formatted_description = format_description(description, breed)
-        initial_state["current_description"] = formatted_description  # 保存當前顯示的描述
-        initial_state["show_back"] = True  # 確保 back 按鈕可見
+        
+        # 保存當前描述和原始按鈕狀態
+        initial_state["current_description"] = formatted_description
+        initial_state["original_buttons"] = initial_state.get("buttons", [])
+        
         return formatted_description, gr.update(visible=True), initial_state
     except Exception as e:
         error_msg = f"An error occurred while showing details: {e}"
@@ -464,29 +467,16 @@ def show_details(choice, previous_output, initial_state):
         return error_msg, gr.update(visible=True), initial_state
 
 def go_back(state):
-    if state.get("is_multi_dog", False):
-        # 恢復到多狗情境的初始狀態
-        buttons = state.get("buttons", [])
-        return (
-            state["explanation"],
-            state["image"],
-            buttons[0] if len(buttons) > 0 else gr.update(visible=False),
-            buttons[1] if len(buttons) > 1 else gr.update(visible=False),
-            buttons[2] if len(buttons) > 2 else gr.update(visible=False),
-            gr.update(visible=False),  # 隱藏 back 按鈕
-            state
-        )
-    else:
-        # 單狗情境，不需要特殊處理
-        return (
-            state["explanation"],
-            state["image"],
-            gr.update(visible=False),
-            gr.update(visible=False),
-            gr.update(visible=False),
-            gr.update(visible=False),
-            state
-        )
+    buttons = state.get("buttons", [])
+    return (
+        state["explanation"],
+        state["image"],
+        buttons[0] if len(buttons) > 0 else gr.update(visible=False),
+        buttons[1] if len(buttons) > 1 else gr.update(visible=False),
+        buttons[2] if len(buttons) > 2 else gr.update(visible=False),
+        gr.update(visible=False),  # 隱藏 back 按鈕
+        state
+    )
 
 with gr.Blocks() as iface:
     gr.HTML("<h1 style='text-align: center;'>🐶 Dog Breed Classifier 🔍</h1>")