Update app.py

app.py

@@ -13,6 +13,8 @@ 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))
@@ -54,34 +56,33 @@ 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 display
-    measurement_display = measurements_df.to_html(
+    # 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="styled-table"
+        classes="vertical-table"
     )
     
-    # Add CSS styling for the table
+    # Add CSS styling for the vertical table
     css_styles = """
     <style>
-    .styled-table {
-        width: 80%;
+    .vertical-table {
+        width: 60%;
         margin: 0 auto;
         border-collapse: collapse;
         font-size: 16px;
-        text-align: center;
+        text-align: left;
     }
-    .styled-table th, .styled-table td {
+    .vertical-table th, .vertical-table td {
         border: 1px solid #dddddd;
         padding: 8px;
     }
-    .styled-table th {
+    .vertical-table th {
         background-color: #f4f4f4;
     }
-    .styled-table tr:nth-child(even) {
-        background-color: #f9f9f9;
-    }
     </style>
     """
     
@@ -89,22 +90,27 @@ def process_images(front_img, side_img, real_height_cm):
     {css_styles}
     <h3 style="text-align: center;">Body Measurements</h3>
     <div style="text-align: center;">
-        <img src="https://huggingface.co/front-logo.png" alt="Hugging Face Logo" style="width: 100px; margin-bottom: 20px;"/>
+        {vertical_table_html}
     </div>
-    {measurement_display}
     """
 
-# Update the Gradio output
+# 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!")
+
 interface = gr.Interface(
     fn=process_images,
     inputs=[
-        gr.Image(sources="webcam", type="numpy", label="Front Pose"),
-        gr.Image(sources="webcam", type="numpy", label="Side Pose"),
+        gr.Image(sources=["webcam", "upload"], type="numpy", label="Front Pose"),
+        gr.Image(sources=["webcam", "upload"], type="numpy", label="Side Pose"),
         gr.Number(label="Enter Your Height (cm)")
     ],
-    outputs=gr.HTML(label="Measurement Results"),  # Use HTML output to display the measurements
+    outputs=gr.HTML(label="Measurement Results"),
     title="Body Sizing System Demo",
-    description="Capture two webcam images: Front View and Side View, and input your height in cm."
+    description="Capture images from webcam or upload, and input your height in cm. 5-second countdown included for webcam captures."
 )
 
 # Launch the app