Update app.py

app.py

@@ -13,8 +13,6 @@ import numpy as np
 from calculations import measure_body_sizes
 import gradio as gr
 import pandas as pd
-import time
-from threading import Timer
 
 # Load BodyPix model
 bodypix_model = load_model(download_model(BodyPixModelPaths.MOBILENET_FLOAT_50_STRIDE_16))
@@ -47,6 +45,7 @@ def process_images(front_img, side_img, real_height_cm):
     
     # Calculate body sizes
     body_sizes = measure_body_sizes(side_colored_mask, front_colored_mask, sideposes, frontposes, real_height_cm, rainbow)
+    print(body_sizes)
     measurements_df = pd.DataFrame([body_sizes]) if isinstance(body_sizes, dict) else pd.DataFrame(body_sizes)
     
     # Save measurements to CSV
@@ -56,61 +55,27 @@ def process_images(front_img, side_img, real_height_cm):
     else:
         measurements_df.to_csv(csv_file, mode='a', header=False, index=False)
     
-    # Prepare measurements for vertical display
-    measurement_display = measurements_df.T.reset_index()
-    measurement_display.columns = ["Measurement", "Value"]
-    vertical_table_html = measurement_display.to_html(
-        index=False,
-        justify="center",
-        border=1,
-        classes="vertical-table"
-    )
-    
-    # Add CSS styling for the vertical table
-    css_styles = """
-    <style>
-    .vertical-table {
-        width: 60%;
-        margin: 0 auto;
-        border-collapse: collapse;
-        font-size: 16px;
-        text-align: left;
-    }
-    .vertical-table th, .vertical-table td {
-        border: 1px solid #dddddd;
-        padding: 8px;
-    }
-    .vertical-table th {
-        background-color: #f4f4f4;
-    }
-    </style>
-    """
-    
+    # Prepare measurements for display
+    measurement_display = measurements_df.to_html(index=False, justify="center", border=1)
+    print(measurement_display)
     return f"""
-    {css_styles}
     <h3 style="text-align: center;">Body Measurements</h3>
     <div style="text-align: center;">
-        {vertical_table_html}
     </div>
+    {measurement_display}
     """
 
-# Gradio interface with countdown and upload options
-def start_countdown():
-    for i in range(5, 0, -1):
-        print(f"Capturing in {i} seconds...")
-        time.sleep(1)
-    print("Image captured!")
-
+# Create the Gradio interface
 interface = gr.Interface(
     fn=process_images,
     inputs=[
-        gr.Image(sources=["webcam", "upload"], type="numpy", label="Front Pose"),
-        gr.Image(sources=["webcam", "upload"], type="numpy", label="Side Pose"),
+        gr.Image(sources="webcam", type="numpy", label="Front Pose"),
+        gr.Image(sources="webcam", type="numpy", label="Side Pose"),
         gr.Number(label="Enter Your Height (cm)")
     ],
-    outputs=gr.HTML(label="Measurement Results"),
+    outputs=gr.HTML(label="Measurement Results"),  # Use HTML output to display the measurements
     title="Body Sizing System Demo",
-    description="Capture images from webcam or upload, and input your height in cm. 5-second countdown included for webcam captures."
+    description="Capture two webcam images: Front View and Side View, and input your height in cm."
 )
 
 # Launch the app