Create gemini.py

TextGen/gemini.py

@@ -0,0 +1,134 @@
+import google.generativeai as genai
+from google.api_core import retry
+
+import pathlib
+import textwrap
+import json
+
+GOOGLE_API_KEY = os.environ['GOOGLE_API_KEY']
+
+genai.configure(api_key=GOOGLE_API_KEY)
+
+model = genai.GenerativeModel(model_name='models/gemini-1.5-pro-latest')
+
+def generate_story(available_items):
+    response = model.generate_content(f"""
+      You are game master in a roguelike game. You should help me create an consise objective for the player to fulfill in the current level.
+      Our protagonist, A girl with a red hood is wandering inside a dungeon.
+      You are ONLY allowed to only use the following elements and not modify either the map or the enemies:
+      {available_items}
+      Example of possible Objectives:
+      Example 1 : Defeat N monster $monster_name and then report it to npc :$npc_name
+      Example 2 : Loot two treasure chest then defeat the boss $boss_name to exit.
+      A object of type NPC can be talked to but can't move and can't be fought, an ennemy type object can be fought but not talked to. A boss can be fought but not talked to
+      To escape, the player will have to pass by the portal.
+      
+      """, request_options={'retry': retry.Retry()})
+    story = response.text
+    return story
+
+def place_objects(available_items,story,myMap):
+    prompt = f"""
+    Please return JSON describing the objective of this level, the objects that should be placed as well as all enemies/npcs that shall be placed:
+    
+    {{"objective": str, "objects":list[OBJECT], "enemies":list[ENEMIES], "boss":list[BOSS],"npcs": list[NPC]}}
+    In all of the following example, placement is the id of the room to place the object in.
+    # The Objects that should be placed in the treasure chests
+    OBJECT = {{"name": str, "description": str, "stength": str, "speed": str}}
+    # The infos necessary about the enemies
+    ENEMIES = {{"name": str, "placement": int}}
+    # The NPC that should be added to the map
+    NPC = {{"name": str, "persona": str, "placement": int}}
+    # The Boss monster of this level.
+    BOSS = {{"name": str, "placement": int}}
+    
+    All fields are not necessarily required; if null, don't add it. it should make sense with the crafted story.
+    Items can only be weapons or objects that can be useful for a special quest. "stength" and "speed" should only be declared for weapons and can only range between 0 and 1. 0 being a rusty sword and 1 being the best sword in the game.
+    
+    Important: Only return a single piece of valid JSON text.
+    
+    Here is the map:
+    
+    {myMap}
+    
+    X: is the starting position of the player
+    ?: represent and interesting room where it could be interesting to place an object
+    numbers represent the index of the room
+    
+    and the objective of this level: {story}
+    The objective MUST be feasible, otherwise the player will be stuck.
+    
+    REMEMBER, You are ONLY allowed to only use the following elements and not modify either the map or the enemies:
+      {available_items}
+      
+    """
+
+    print(prompt)
+    response = model.generate_content(
+      prompt,
+      generation_config={'response_mime_type':'application/json'}
+    )
+    try:
+        map_placements=json.dumps(json.loads(response.text), indent=4)
+    except:
+        map_placements="error"
+    return map_placements
+
+def get_items_string_representation(items_list):
+    result = ""
+    for item in items_list:
+        item_str = "Name: " + item["name"] + ", Type: " + item["type"] + ", Description: " + item["description"] + "\n"
+        result += item_str
+    return result
+
+def generate_map_markdown(data):
+    import numpy as np
+    
+    # Define the room structure with walls and markers
+    def create_room(room_char, index):
+        index_str = f"{index}".zfill(2)  # Zero pad index to 2 digits if needed
+        if index_str[0] !="0":
+            first_number=index_str[0]
+        else:
+            first_number=" "
+        return [
+            f"╔═{room_char}═╗",
+            f"║ {first_number}{index_str[1]}║",
+            f"╚═══╝"
+        ]
+    
+    # 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"]]
+    
+    # Determine grid size
+    min_x = min(room[0] for room in rooms)
+    max_x = max(room[0] for room in rooms)
+    min_y = min(room[1] for room in rooms)
+    max_y = max(room[1] for room in rooms)
+    
+    # Create grid with empty spaces represented by a room-like structure
+    map_height = (max_y - min_y + 1) * 3
+    map_width = (max_x - min_x + 1) * 5
+    grid = np.full((map_height, map_width), " ")
+    
+    # Populate grid with rooms and their characteristics
+    for i, room in enumerate(rooms):
+        x, y = room
+        x_offset = (x - min_x) * 5
+        y_offset = (max_y - y) * 3
+        if room == (0, 0):
+            room_char = "X"
+        elif room in rooms_of_interest:
+            room_char = "P" if i == data["index_exit"] else "?"
+        else:
+            room_char = " "
+        room_structure = create_room(room_char, i)
+        for j, row in enumerate(room_structure):
+            grid[y_offset + j, x_offset:x_offset + 5] = list(row)
+    
+    # Convert grid to a string format suitable for display
+    markdown_map = "\n".join("".join(row) for row in grid)
+    
+    # Return the map wrapped in triple backticks for proper display in markdown
+    return f"```\n{markdown_map}\n```"