Update breed_detection.py

breed_detection.py

@@ -22,7 +22,7 @@ def create_detection_tab(predict_fn, example_images):
                     -webkit-text-fill-color: transparent;
                     font-weight: 600;
                 '>
-                    Upload a picture of a dog or take a photo, and the model will predict its breed and provide detailed information!
+                    Upload a picture of a dog, and the model will predict its breed and provide detailed information!
                 </p>
                 <p style='
                     font-size: 0.9em;
@@ -35,69 +35,27 @@ def create_detection_tab(predict_fn, example_images):
             </div>
         """)
         
-        # 將輸入方法放在標籤頁中
-        with gr.Tabs():
-            # 標籤頁 1: 上傳圖片 (保留原有功能)
-            with gr.TabItem("Upload Image"):
-                input_image = gr.Image(label="Upload a dog image", type="pil")
-                gr.Examples(
-                    examples=example_images,
-                    inputs=input_image
-                )
-            
-            # 標籤頁 2: 拍照功能 (使用 gr.Webcam 而非 gr.Image(source="webcam"))
-            with gr.TabItem("Take Photo"):
-                camera_input = gr.Webcam(label="Take a photo of a dog")
-        
-        # 輸出區域
         with gr.Row():
+            input_image = gr.Image(label="Upload a dog image", type="pil")
             output_image = gr.Image(label="Annotated Image")
-            output = gr.HTML(label="Prediction Results")
-        
-        # 使用 State 保存預測結果
+
+        output = gr.HTML(label="Prediction Results")
         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]
         )
-        
-        # 添加相機拍攝事件處理 (針對 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],
-            outputs=[output, output_image, initial_state]
+
+        gr.Examples(
+            examples=example_images,
+            inputs=input_image
         )
 
     return {
         'input_image': input_image,
-        'camera_input': camera_input,
         'output_image': output_image,
         'output': output,
-        'initial_state': initial_state,
-        'detect_btn': detect_btn
+        'initial_state': initial_state
     }