Update app.py

app.py

@@ -30,7 +30,7 @@ def process_images(front_img, side_img, real_height_cm):
     fimage_array = preprocessing.image.img_to_array(front_img)
     simage_array = preprocessing.image.img_to_array(side_img)
     
-    # bodypix prediction
+    # BodyPix prediction
     frontresult = bodypix_model.predict_single(fimage_array)
     sideresult = bodypix_model.predict_single(simage_array)
     
@@ -41,32 +41,28 @@ def process_images(front_img, side_img, real_height_cm):
     side_colored_mask = sideresult.get_colored_part_mask(side_mask, rainbow)
     
     frontposes = frontresult.get_poses()
-    front_image_with_poses = draw_poses(
-        fimage_array.copy(),
-        frontposes,
-        keypoints_color=(255, 100, 100),
-        skeleton_color=(100, 100, 255)
-    )
-    
     sideposes = sideresult.get_poses()
-    side_image_with_poses = draw_poses(
-        simage_array.copy(),
-        sideposes,
-        keypoints_color=(255, 100, 100),
-        skeleton_color=(100, 100, 255)
-    )
     
     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)
+    
+    # Save measurements to CSV
     csv_file = "Body-measurement.csv"
     if not os.path.exists(csv_file):
-        # Save as a new file if it doesn't exist
         measurements_df.to_csv(csv_file, index=False)
     else:
-        # Append to the existing file
         measurements_df.to_csv(csv_file, mode='a', header=False, index=False)
-        
-    return "Your body part measured and saved to the database."
+    
+    # Format measurements as HTML
+    measurements_html = 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}
+    """
 
 # Create the Gradio interface
 interface = gr.Interface(
@@ -76,7 +72,7 @@ interface = gr.Interface(
         gr.Image(sources="webcam", type="numpy", label="Side Pose"),
         gr.Number(label="Enter Your Height (cm)")
     ],
-    outputs=gr.Textbox(label="Confirmation Message"),  # Confirmation message output
+    outputs=gr.HTML(label="Measurement Results"),  # Use HTML output to display the logo and measurements
     title="Body Sizing System Demo",
     description="Capture two webcam images: Front View and Side View, and input your height in cm."
 )