Update app.py

app.py

@@ -397,10 +397,41 @@ def process_image(image, user_height_cm):
     return measurements, cv2.cvtColor(image_cv, cv2.COLOR_BGR2RGB), keypoints.tolist()
 
 # === Save to DB ===
+# def save_to_database(measurements, image, user_height_cm, user_id):
+#     if not user_id:
+#         return "❌ user_id missing from URL."
+
+#     if measurements is None or image is None:
+#         return "⚠️ No data to save."
+
+#     buffered = BytesIO()
+#     pil_image = Image.fromarray(image)
+#     pil_image.save(buffered, format="JPEG")
+#     img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
+
+#     payload = {
+#         "imageBase64": img_str,
+#         "heightCm": user_height_cm,
+#         "measurements": measurements
+#     }
+
+    # try:
+    #     response = requests.post(
+    #         f"https://9d68-210-212-162-140.ngrok-free.app/upload/{user_id}",
+    #         json=payload
+    #     )
+    #     if response.status_code == 201:
+    #         return "✅ Measurements and image saved to database!"
+    #     else:
+    #         return f"❌ Failed: {response.status_code} - {response.text}"
+    # except Exception as e:
+    #     return f"⚠️ Error during save: {str(e)}"
+
+
+
 def save_to_database(measurements, image, user_height_cm, user_id):
     if not user_id:
         return "❌ user_id missing from URL."
-
     if measurements is None or image is None:
         return "⚠️ No data to save."
 
@@ -412,7 +443,11 @@ def save_to_database(measurements, image, user_height_cm, user_id):
     payload = {
         "imageBase64": img_str,
         "heightCm": user_height_cm,
-        "measurements": measurements
+        "waistCircumferenceCm": measurements.get("Waist Circumference (cm)"),
+        "hipcircumference": measurements.get("Hip Circumference (cm)"),
+        "shoulderwidth": measurements.get("Shoulder Width (cm)"),
+        "torsolength": measurements.get("Torso Length (Neck to Pelvis, cm)"),
+        "fullarmlength": measurements.get("Full Arm Length (Shoulder to Wrist, cm)"),
     }
 
     try:
@@ -427,6 +462,7 @@ def save_to_database(measurements, image, user_height_cm, user_id):
     except Exception as e:
         return f"⚠️ Error during save: {str(e)}"
 
+
 # === Gradio App ===
 with gr.Blocks() as demo:
     gr.Markdown("# 📏 AI-Powered Body Measurement Tool")