Update breed_detection.py

breed_detection.py

@@ -35,8 +35,9 @@ def create_detection_tab(predict_fn, example_images):
             </div>
         """)
         
+        # 將輸入方法放在標籤頁中
         with gr.Tabs():
-            # tab1: upload images
+            # 標籤頁 1: 上傳圖片 (保留原有功能)
             with gr.TabItem("Upload Image"):
                 input_image = gr.Image(label="Upload a dog image", type="pil")
                 gr.Examples(
@@ -44,47 +45,48 @@ def create_detection_tab(predict_fn, example_images):
                     inputs=input_image
                 )
             
-            # tab2: camera feature
+            # 標籤頁 2: 拍照功能 (使用 gr.Webcam 而非 gr.Image(source="webcam"))
             with gr.TabItem("Take Photo"):
-                camera_input = gr.Image(
-                    source="webcam",
-                    label="Take a photo of a dog",
-                    type="pil",
-                    tool="editor"  # 允許用戶編輯拍攝的照片
-                )
+                camera_input = gr.Webcam(label="Take a photo of a dog")
         
-        # output area
+        # 輸出區域
         with gr.Row():
             output_image = gr.Image(label="Annotated Image")
             output = gr.HTML(label="Prediction Results")
         
-        # utilize state to store results
+        # 使用 State 保存預測結果
         initial_state = gr.State()
         
+        # 輔助函數，在函數內部定義避免循環導入
         def detect_from_inputs(upload_image, camera_image):
+            # 使用最後修改的圖片（優先相機拍攝的圖片）
             image_to_use = camera_image if camera_image is not None else upload_image
             
             if image_to_use is None:
                 return "Please upload an image or take a photo first.", None, None
             
+            # 使用作為參數傳入的 predict_fn
             return predict_fn(image_to_use)
         
+        # 修改輸入圖片事件處理
         input_image.change(
             predict_fn,
             inputs=input_image,
             outputs=[output, output_image, initial_state]
         )
         
-        # add camera input
+        # 添加相機拍攝事件處理 (針對 gr.Webcam)
         camera_input.change(
             predict_fn,
             inputs=camera_input,
             outputs=[output, output_image, initial_state]
         )
         
+        # 添加按鈕以便使用者可以主動觸發分析
         with gr.Row():
             detect_btn = gr.Button("Detect Breed", variant="primary")
         
+        # 為按鈕設置事件處理
         detect_btn.click(
             detect_from_inputs,
             inputs=[input_image, camera_input],