Update main.py

main.py

@@ -231,6 +231,8 @@ async def predict_injections(file: UploadFile = File(...)):
         }
         optimal_list.append(pt_info)
 
+    all_coords_list = [[float(f"{p.x:.4f}"), float(f"{p.y:.4f}")] for p in lms[:468]]
+
     response = {
         "status": "success",
         "summary": {
@@ -238,8 +240,10 @@ async def predict_injections(file: UploadFile = File(...)):
             "max_confidence": float(probs_loc.max())
         },
         "optimal_injections": optimal_list,
-        # IMPROVEMENT: Return full heatmap probabilities
-        "all_probabilities": [float(f"{p:.4f}") for p in probs_loc.tolist()]
+        
+        # HEATMAP DATA
+        "all_probabilities": [float(f"{p:.4f}") for p in probs_loc.tolist()],
+        "all_coordinates": all_coords_list # New Field
     }
         
     return response