Update app.py

app.py

@@ -65,7 +65,8 @@ def play_countdown_audio():
         while pygame.mixer.music.get_busy():
             time.sleep(0.1)
 
-def capture_image(label, camera, person_height_cm, pixels_per_cm):
+def capture_image(label, person_height_cm, pixels_per_cm):
+    camera = gr.Button("Capture Images")
     print(f"Please position yourself for the {label} pose.")
     instruction_text = f"Please stand straight in the box for the {label} pose."
     audio_file = f"input1/instruction_{label}.mp3"
@@ -125,16 +126,14 @@ def process_images(front_img, side_img, real_height_cm):
     return measurements_df
 
 def gradio_interface(person_height_cm):
-    cap = cv2.VideoCapture(0)
-    if not cap.isOpened():
-        return "Error: Unable to access the camera."
+    
+#    if not cap.isOpened():
+#        return "Error: Unable to access the camera."
 
     pixels_per_cm = 5
-    front_frame = capture_image("front", cap, person_height_cm, pixels_per_cm)
-    side_frame = capture_image("side", cap, person_height_cm, pixels_per_cm)
+    front_frame = capture_image("front", person_height_cm, pixels_per_cm)
+    side_frame = capture_image("side", person_height_cm, pixels_per_cm)
 
-    cap.release()
-    cv2.destroyAllWindows()
 
     # Load the captured images to process for body measurement
     front_img = cv2.imread(os.path.join(output_dir, "output/front_image.jpg"))
@@ -148,7 +147,6 @@ def main():
     # Updated Gradio interface with separate button trigger
     with gr.Blocks() as demo:
         height_input = gr.Number(label="Enter your height (in cm)")
-        capture_button = gr.Button("Capture Images")
         output_df = gr.DataFrame()
 
         capture_button.click(fn=gradio_interface, inputs=height_input, outputs=output_df)