Update app.py

app.py

@@ -125,7 +125,6 @@
 
 
 
-
 import torch
 import numpy as np
 import cv2
@@ -156,7 +155,7 @@ cfg.MODEL.DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 predictor = DefaultPredictor(cfg)
 
 # Replace with your actual Ngrok or backend URL
-NGROK_URL = " https://9dbc-210-212-162-140.ngrok-free.app/upload"  # Change this
+NGROK_URL = "https://your-ngrok-url.ngrok.io/measurements"  # Change this
 
 def process_image(image, user_height_cm):
     # Convert Gradio image input to OpenCV format
@@ -260,7 +259,7 @@ def save_to_database(measurements, image, user_height_cm):
 
     # Send POST request
     response = requests.post(
-        "https://your-ngrok-url.ngrok.io/measurements",  # Change this
+        "https://9dbc-210-212-162-140.ngrok-free.app/upload",  # Change this
         json={
             "imageBase64": img_str,
             "heightCm": user_height_cm,
@@ -275,22 +274,27 @@ def save_to_database(measurements, image, user_height_cm):
     return f"Status: {response.status_code}, Message: {response.text}"
 
 # Gradio Interface
-demo = gr.Interface(
-    fn=process_image,
-    inputs=[gr.Image(type="pil"), gr.Number(label="User Height (cm)")],
-    outputs=[gr.JSON(label="Measurements"), gr.Image(type="pil", label="Keypoint Overlay")],
-    title="Keypoint Measurement Extractor",
-    description="Upload an image, enter your height, and get body measurements based on keypoints.",
-)
-
-# Add Save to Database button functionality
-save_btn = gr.Button("Save to Database")
-save_status = gr.Textbox(label="Save Status", interactive=False)
-
-save_btn.click(
-    fn=save_to_database,
-    inputs=[demo.outputs[0], demo.outputs[1], gr.Number(label="User Height (cm)")],
-    outputs=save_status
-)
-
+def create_interface():
+    with gr.Blocks() as demo:
+        with gr.Row():
+            img_input = gr.Image(type="pil", label="Upload Image")
+            height_input = gr.Number(label="User Height (cm)")
+        
+        with gr.Row():
+            output_measurements = gr.JSON(label="Measurements")
+            output_image = gr.Image(type="pil", label="Keypoint Overlay")
+
+        save_btn = gr.Button("Save to Database")
+        save_status = gr.Textbox(label="Save Status", interactive=False)
+
+        img_input.change(fn=process_image, inputs=[img_input, height_input], outputs=[output_measurements, output_image])
+        save_btn.click(
+            fn=save_to_database,
+            inputs=[output_measurements, output_image, height_input],
+            outputs=save_status
+        )
+
+    return demo
+
+demo = create_interface()
 demo.launch()