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, and the model will predict its breed and provide detailed information!
+                    Upload a picture of a dog or take a photo, and the model will predict its breed and provide detailed information!
                 </p>
                 <p style='
                     font-size: 0.9em;
@@ -35,27 +35,67 @@ def create_detection_tab(predict_fn, example_images):
             </div>
         """)
         
+        with gr.Tabs():
+            # tab1: upload images
+            with gr.TabItem("Upload Image"):
+                input_image = gr.Image(label="Upload a dog image", type="pil")
+                gr.Examples(
+                    examples=example_images,
+                    inputs=input_image
+                )
+            
+            # tab2: camera feature
+            with gr.TabItem("Take Photo"):
+                camera_input = gr.Image(
+                    source="webcam",
+                    label="Take a photo of a dog",
+                    type="pil",
+                    tool="editor"  # 允許用戶編輯拍攝的照片
+                )
+        
+        # output area
         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")
+            output = gr.HTML(label="Prediction Results")
+        
+        # utilize state to store 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
+            
+            return predict_fn(image_to_use)
+        
         input_image.change(
             predict_fn,
             inputs=input_image,
             outputs=[output, output_image, initial_state]
         )
-
-        gr.Examples(
-            examples=example_images,
-            inputs=input_image
+        
+        # add camera input
+        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]
         )
 
     return {
         'input_image': input_image,
+        'camera_input': camera_input,
         'output_image': output_image,
         'output': output,
-        'initial_state': initial_state
-    }
+        'initial_state': initial_state,
+        'detect_btn': detect_btn
+    }