Update app.py

app.py

@@ -43,6 +43,7 @@ def process_images(front_img, side_img, real_height_cm):
     frontposes = frontresult.get_poses()
     sideposes = sideresult.get_poses()
     
+    # Calculate body sizes
     body_sizes = measure_body_sizes(side_colored_mask, front_colored_mask, sideposes, frontposes, real_height_cm, rainbow)
     measurements_df = pd.DataFrame([body_sizes]) if isinstance(body_sizes, dict) else pd.DataFrame(body_sizes)
     
@@ -53,15 +54,15 @@ def process_images(front_img, side_img, real_height_cm):
     else:
         measurements_df.to_csv(csv_file, mode='a', header=False, index=False)
     
-    # Format measurements as HTML
-    measurements_html = measurements_df.to_html(index=False, justify="center", border=1)
+    # Prepare measurements for display
+    measurement_display = measurements_df.to_html(index=False, justify="center", border=1)
     
     return f"""
     <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;"/>
     </div>
-    {measurements_html}
+    {measurement_display}
     """
 
 # Create the Gradio interface
@@ -72,7 +73,7 @@ interface = gr.Interface(
         gr.Image(sources="webcam", type="numpy", label="Side Pose"),
         gr.Number(label="Enter Your Height (cm)")
     ],
-    outputs=gr.HTML(label="Measurement Results"),  # Use HTML output to display the logo and measurements
+    outputs=gr.HTML(label="Measurement Results"),  # Use HTML output to display the measurements
     title="Body Sizing System Demo",
     description="Capture two webcam images: Front View and Side View, and input your height in cm."
 )