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)