Update app.py

app.py

@@ -436,21 +436,73 @@ async def process_single_dog(image):
         }
         return explanation, image, buttons[0], buttons[1], buttons[2], gr.update(visible=True), initial_state
 
-def show_details(choice, previous_output, initial_state):
-    if not choice:
-        return previous_output, gr.update(visible=True), initial_state
+# def show_details(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(description, breed)
-        return formatted_description, gr.update(visible=True), initial_state
-    except Exception as e:
-        error_msg = f"An error occurred while showing details: {e}"
-        print(error_msg)  # 添加日誌輸出
-        return error_msg, gr.update(visible=True), initial_state
+#     try:
+#         breed = choice.split("More about ")[-1]
+#         description = get_dog_description(breed)
+#         formatted_description = format_description(description, breed)
+#         return formatted_description, gr.update(visible=True), initial_state
+#     except Exception as e:
+#         error_msg = f"An error occurred while showing details: {e}"
+#         print(error_msg)  # 添加日誌輸出
+#         return error_msg, gr.update(visible=True), initial_state
+
+# # 介面部分
+# with gr.Blocks() as iface:
+#     gr.HTML("<h1 style='text-align: center;'>🐶 Dog Breed Classifier 🔍</h1>")
+#     gr.HTML("<p style='text-align: center;'>Upload a picture of a dog, and the model will predict its breed, provide detailed information, and include an extra information link!</p>")
+    
+#     with gr.Row():
+#         input_image = gr.Image(label="Upload a dog image", type="pil")
+#         output_image = gr.Image(label="Annotated Image")
+    
+#     output = gr.Markdown(label="Prediction Results")
+    
+#     with gr.Row():
+#         btn1 = gr.Button("View More 1", visible=False)
+#         btn2 = gr.Button("View More 2", visible=False)
+#         btn3 = gr.Button("View More 3", visible=False)
+    
+#     back_button = gr.Button("Back", visible=False)
+    
+#     initial_state = gr.State()
+    
+#     input_image.change(
+#         predict,
+#         inputs=input_image,
+#         outputs=[output, output_image, btn1, btn2, btn3, back_button, initial_state]
+#     )
+
+#     for btn in [btn1, btn2, btn3]:
+#         btn.click(
+#             show_details,
+#             inputs=[btn, output, initial_state],
+#             outputs=[output, back_button, initial_state]
+#         )
+
+#     back_button.click(
+#         lambda state: (state["explanation"], 
+#                        state["buttons"][0] if len(state["buttons"]) > 0 else gr.update(visible=False),
+#                        state["buttons"][1] if len(state["buttons"]) > 1 else gr.update(visible=False),
+#                        state["buttons"][2] if len(state["buttons"]) > 2 else gr.update(visible=False),
+#                        gr.update(visible=state["show_back"])),
+#         inputs=[initial_state],
+#         outputs=[output, btn1, btn2, btn3, back_button]
+#     )
+
+#     gr.Examples(
+#         examples=['Border_Collie.jpg', 'Golden_Retriever.jpeg', 'Saint_Bernard.jpeg', 'French_Bulldog.jpeg', 'Samoyed.jpg'],
+#         inputs=input_image
+#     )
+
+#     gr.HTML('For more details on this project and other work, feel free to visit my GitHub <a href="https://github.com/Eric-Chung-0511/Learning-Record/tree/main/Data%20Science%20Projects/Dog_Breed_Classifier">Dog Breed Classifier</a>')
+
+#     if __name__ == "__main__":
+#         iface.launch()
 
-# 介面部分
 with gr.Blocks() as iface:
     gr.HTML("<h1 style='text-align: center;'>🐶 Dog Breed Classifier 🔍</h1>")
     gr.HTML("<p style='text-align: center;'>Upload a picture of a dog, and the model will predict its breed, provide detailed information, and include an extra information link!</p>")
@@ -462,43 +514,50 @@ with gr.Blocks() as iface:
     output = gr.Markdown(label="Prediction Results")
     
     with gr.Row():
-        btn1 = gr.Button("View More 1", visible=False)
-        btn2 = gr.Button("View More 2", visible=False)
-        btn3 = gr.Button("View More 3", visible=False)
+        buttons = [gr.Button(f"View More {i+1}", visible=False) for i in range(9)]  # 增加到9個按鈕
     
     back_button = gr.Button("Back", visible=False)
     
     initial_state = gr.State()
     
+    def show_details(button_text, current_output, state):
+        breed = button_text.split("More about ")[-1]
+        description = get_dog_description(breed)
+        formatted_description = format_description(description, breed)
+        state["current_breed"] = breed
+        return formatted_description, gr.update(visible=True), state
+
+    def go_back(state):
+        return (state["explanation"], 
+                *[btn if i < len(state["buttons"]) else gr.update(visible=False) 
+                  for i, btn in enumerate(state["buttons"])],
+                gr.update(visible=state["show_back"]))
+
     input_image.change(
         predict,
         inputs=input_image,
-        outputs=[output, output_image, btn1, btn2, btn3, back_button, initial_state]
+        outputs=[output, output_image] + buttons + [back_button, initial_state]
     )
-
-    for btn in [btn1, btn2, btn3]:
+    
+    for btn in buttons:
         btn.click(
             show_details,
             inputs=[btn, output, initial_state],
             outputs=[output, back_button, initial_state]
         )
-
+    
     back_button.click(
-        lambda state: (state["explanation"], 
-                       state["buttons"][0] if len(state["buttons"]) > 0 else gr.update(visible=False),
-                       state["buttons"][1] if len(state["buttons"]) > 1 else gr.update(visible=False),
-                       state["buttons"][2] if len(state["buttons"]) > 2 else gr.update(visible=False),
-                       gr.update(visible=state["show_back"])),
+        go_back,
         inputs=[initial_state],
-        outputs=[output, btn1, btn2, btn3, back_button]
+        outputs=[output] + buttons + [back_button]
     )
-
+    
     gr.Examples(
         examples=['Border_Collie.jpg', 'Golden_Retriever.jpeg', 'Saint_Bernard.jpeg', 'French_Bulldog.jpeg', 'Samoyed.jpg'],
         inputs=input_image
     )
-
+    
     gr.HTML('For more details on this project and other work, feel free to visit my GitHub <a href="https://github.com/Eric-Chung-0511/Learning-Record/tree/main/Data%20Science%20Projects/Dog_Breed_Classifier">Dog Breed Classifier</a>')
 
-    if __name__ == "__main__":
-        iface.launch()
+if __name__ == "__main__":
+    iface.launch()