Update TextGen/gemini.py

TextGen/gemini.py

@@ -81,7 +81,7 @@ def get_items_string_representation(items_list):
         result += item_str
     return result
 
-def generate_map_markdown(data):
+def generate_map_markdown(rooms, room_of_interest,index_exit):
     import numpy as np
     
     # Define the room structure with walls and markers
@@ -98,8 +98,8 @@ def generate_map_markdown(data):
         ]
     
     # Extract rooms and rooms of interest
-    rooms = [eval(room) for room in data["rooms"]]
-    rooms_of_interest = [eval(room) for room in data["room_of_interest"]]
+    rooms = [eval(room) for room in rooms]
+    rooms_of_interest = [eval(room) for room in room_of_interest]
     
     # Determine grid size
     min_x = min(room[0] for room in rooms)
@@ -120,7 +120,7 @@ def generate_map_markdown(data):
         if room == (0, 0):
             room_char = "X"
         elif room in rooms_of_interest:
-            room_char = "P" if i == data["index_exit"] else "?"
+            room_char = "P" if i == index_exit else "?"
         else:
             room_char = " "
         room_structure = create_room(room_char, i)