Update app.py

app.py

@@ -467,6 +467,7 @@ async def predict(image):
 
         dogs_info = ""
 
+        # 在每個品種旁邊生成 Gradio 按鈕
         for i, (cropped_image, detection_confidence, box) in enumerate(dogs):
             top1_prob, topk_breeds, topk_probs_percent = await predict_single_dog(cropped_image)
             color = color_list[i % len(color_list)]
@@ -486,14 +487,13 @@ async def predict(image):
                 dogs_info += f"<p>Top 3 possible breeds:</p><ul>"
                 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}"
         
-                    # 為每個品種生成按鈕，但不重複加入最下面的 buttons
-                    # 刪除 buttons.append(button_id)，因為這是多餘的部分
+                    # 使用 Gradio 內建按鈕，並與回調函數連結
+                    button = gr.Button(f"Learn More about {breed}")
+                    buttons.append(button)  # 增加到 buttons 列表中
+                    # 生成表單輸出
+                    dogs_info += f"<li><strong>{breed}</strong> ({prob:.2f}% confidence)</li>"
         
-                    # 加入 Learn More 按鈕，並綁定點擊事件
-                    dogs_info += f"<li><strong>{breed}</strong> ({prob:.2f}% confidence)"
-                    dogs_info += f'<button style="background-color: #4CAF50; color: white; border: none; padding: 5px 10px; border-radius: 3px; margin-left: 10px;" onclick="handle_button_click(\'{button_id}\')">Learn More</button></li>'
                 dogs_info += "</ul>"
         
             else:
@@ -501,25 +501,17 @@ async def predict(image):
         
             dogs_info += '</div>'
         
-        # 生成 Javascript 來處理按鈕點擊事件
+        # 更新 Gradio 前端
         html_output = f"""
         <style>
         .dog-info {{ border: 1px solid #ddd; margin-bottom: 20px; padding: 15px; border-radius: 5px; box-shadow: 0 2px 5px rgba(0,0,0,0.1); }}
         .dog-info h2 {{ background-color: #f0f0f0; padding: 10px; margin: -15px -15px 15px -15px; border-radius: 5px 5px 0 0; }}
         </style>
         {dogs_info}
-        <script>
-        function handle_button_click(button_id) {{
-            const radio = document.querySelector('input[type=radio][value="' + button_id + '"]');
-            if (radio) {{
-                radio.click();  // 點擊按鈕後觸發事件
-            }} else {{
-                console.error("Radio button not found:", button_id);
-            }}
-        }}
-        </script>
         """
 
+
+