return floor plan if requested (#20)

ai4ce · Dec 18, 2024 · 48e14c0 · 48e14c0
1 parent 1f7c506
commit 48e14c0
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/modal_functions/unav.py b/src/modal_functions/unav.py
@@ -41,6 +41,7 @@ def planner(
         floor: str = "",
         place: str = "",
         base_64_image: str = None,
+        get_floor_plan: bool = False,
     ):
 
         import json
@@ -103,6 +104,10 @@ def planner(
 
         scale = self.server.config["location"]["scale"]
 
+        if(get_floor_plan):
+            floorplan_base64 = pose["floorplan_base64"]
+            return json.dumps({"trajectory": trajectory, "scale": scale, "floorplan_base64" : floorplan_base64}) #return floor plan if requested 
+
         return json.dumps({"trajectory": trajectory, "scale": scale})
 
     @method()