Added LS ID gen

.backup/working code backup/LS-AI-img-gen

@@ -0,0 +1 @@
+Subproject commit 90888e775b9ca58b4833b1dcef77fcee1c75429e

.gitignore

@@ -1,2 +1,2 @@
 images/
-.venv/
+.venv/

adventurers_data.json

@@ -0,0 +1,34 @@
+{
+    "data": {
+        "adventurers": [
+            {
+                "owner": "0x62f0ca21835f5fb7470f80eaee41dcf29f7989e14607f023f390fcfb864bc7c",
+                "id": 450,
+                "name": "Realms #5656",
+                "strength": 1,
+                "vitality": 6,
+                "dexterity": 4,
+                "intelligence": 1,
+                "wisdom": 2,
+                "charisma": 7,
+                "level": 13,
+                "xp": 175,
+                "health": 0,
+                "beastHealth": 0,
+                "head": "Leather Cap",
+                "hand": "Hard Leather Gloves",
+                "chest": "Hard Leather Armor",
+                "waist": "Hard Leather Belt",
+                "foot": "Linen Shoes",
+                "weapon": "Quarterstaff",
+                "gold": 0,
+                "neck": null,
+                "ring": null,
+                "luck": 2,
+                "battleActionCount": 0,
+                "customRenderer": null,
+                "statUpgrades": 0
+            }
+        ]
+    }
+}

adventurers_data2.json

@@ -0,0 +1,186 @@
+{
+    "data": {
+        "battles": [
+            {
+                "adventurerId": 450,
+                "adventurerHealth": 49,
+                "beast": "Wyvern",
+                "beastHealth": 10,
+                "beastLevel": 3,
+                "seed": "0x3f6f4350",
+                "attacker": "Adventurer",
+                "fled": false,
+                "damageDealt": 4,
+                "criticalHit": false,
+                "damageTaken": 0,
+                "damageLocation": null,
+                "xpEarnedAdventurer": 0,
+                "xpEarnedItems": 0,
+                "goldEarned": 0,
+                "discoveryTime": "2024-09-11T21:56:12.363Z"
+            },
+            {
+                "adventurerId": 450,
+                "adventurerHealth": 49,
+                "beast": "Wyvern",
+                "beastHealth": 10,
+                "beastLevel": 3,
+                "seed": "0x3f6f4350",
+                "attacker": "Beast",
+                "fled": false,
+                "damageDealt": 0,
+                "criticalHit": false,
+                "damageTaken": 13,
+                "damageLocation": "Foot",
+                "xpEarnedAdventurer": 0,
+                "xpEarnedItems": 0,
+                "goldEarned": 0,
+                "discoveryTime": "2024-09-11T21:56:12.363Z"
+            },
+            {
+                "adventurerId": 450,
+                "adventurerHealth": 75,
+                "beast": "Wyvern",
+                "beastHealth": 18,
+                "beastLevel": 3,
+                "seed": "0x3f6f4350",
+                "attacker": "Adventurer",
+                "fled": false,
+                "damageDealt": 4,
+                "criticalHit": false,
+                "damageTaken": 0,
+                "damageLocation": null,
+                "xpEarnedAdventurer": 0,
+                "xpEarnedItems": 0,
+                "goldEarned": 0,
+                "discoveryTime": "2024-09-11T21:56:12.361Z"
+            },
+            {
+                "adventurerId": 450,
+                "adventurerHealth": 62,
+                "beast": "Wyvern",
+                "beastHealth": 14,
+                "beastLevel": 3,
+                "seed": "0x3f6f4350",
+                "attacker": "Beast",
+                "fled": false,
+                "damageDealt": 0,
+                "criticalHit": false,
+                "damageTaken": 13,
+                "damageLocation": "Waist",
+                "xpEarnedAdventurer": 0,
+                "xpEarnedItems": 0,
+                "goldEarned": 0,
+                "discoveryTime": "2024-09-11T21:56:12.363Z"
+            },
+            {
+                "adventurerId": 450,
+                "adventurerHealth": 75,
+                "beast": "Wyvern",
+                "beastHealth": 18,
+                "beastLevel": 3,
+                "seed": "0x3f6f4350",
+                "attacker": "Beast",
+                "fled": false,
+                "damageDealt": 0,
+                "criticalHit": false,
+                "damageTaken": 4,
+                "damageLocation": "Chest",
+                "xpEarnedAdventurer": 0,
+                "xpEarnedItems": 0,
+                "goldEarned": 0,
+                "discoveryTime": "2024-09-11T21:56:12.362Z"
+            },
+            {
+                "adventurerId": 450,
+                "adventurerHealth": 79,
+                "beast": "Wyvern",
+                "beastHealth": 22,
+                "beastLevel": 3,
+                "seed": "0x3f6f4350",
+                "attacker": "Beast",
+                "fled": false,
+                "damageDealt": 0,
+                "criticalHit": false,
+                "damageTaken": 13,
+                "damageLocation": "Head",
+                "xpEarnedAdventurer": 0,
+                "xpEarnedItems": 0,
+                "goldEarned": 0,
+                "discoveryTime": "2024-09-11T21:56:12.361Z"
+            },
+            {
+                "adventurerId": 450,
+                "adventurerHealth": 90,
+                "beast": "Pixie",
+                "beastHealth": 0,
+                "beastLevel": 1,
+                "seed": "0x5a60e008",
+                "attacker": "Adventurer",
+                "fled": false,
+                "damageDealt": 4,
+                "criticalHit": false,
+                "damageTaken": 0,
+                "damageLocation": null,
+                "xpEarnedAdventurer": 4,
+                "xpEarnedItems": 8,
+                "goldEarned": 0,
+                "discoveryTime": "2024-09-11T21:53:25.168Z"
+            },
+            {
+                "adventurerId": 450,
+                "adventurerHealth": 90,
+                "beast": "Fairy",
+                "beastHealth": 3,
+                "beastLevel": 1,
+                "seed": "0x1c2",
+                "attacker": "Beast",
+                "fled": false,
+                "damageDealt": 0,
+                "criticalHit": false,
+                "damageTaken": 10,
+                "damageLocation": "Chest",
+                "xpEarnedAdventurer": 0,
+                "xpEarnedItems": 0,
+                "goldEarned": 0,
+                "discoveryTime": "2024-09-10T18:35:14.760Z"
+            },
+            {
+                "adventurerId": 450,
+                "adventurerHealth": 79,
+                "beast": "Wyvern",
+                "beastHealth": 22,
+                "beastLevel": 3,
+                "seed": "0x3f6f4350",
+                "attacker": "Adventurer",
+                "fled": false,
+                "damageDealt": 4,
+                "criticalHit": false,
+                "damageTaken": 0,
+                "damageLocation": null,
+                "xpEarnedAdventurer": 0,
+                "xpEarnedItems": 0,
+                "goldEarned": 0,
+                "discoveryTime": "2024-09-11T21:56:12.361Z"
+            },
+            {
+                "adventurerId": 450,
+                "adventurerHealth": 62,
+                "beast": "Wyvern",
+                "beastHealth": 14,
+                "beastLevel": 3,
+                "seed": "0x3f6f4350",
+                "attacker": "Adventurer",
+                "fled": false,
+                "damageDealt": 4,
+                "criticalHit": false,
+                "damageTaken": 0,
+                "damageLocation": null,
+                "xpEarnedAdventurer": 0,
+                "xpEarnedItems": 0,
+                "goldEarned": 0,
+                "discoveryTime": "2024-09-11T21:56:12.362Z"
+            }
+        ]
+    }
+}

app.py

@@ -1,7 +1,9 @@
 # app.py 
-from config.config import models, prompts, api_token  # Direct import
+from config.config import prompts, api_token  # Direct import
+from config.models import models
 import gradio as gr
 from src.img_gen import generate_image
+from metadata.metadata import fetch_metadata
 
 # Gradio Interface
 def gradio_interface():
@@ -20,22 +22,24 @@ def gradio_interface():
         with gr.Row():
             # Set default values for dropdowns
             prompt_dropdown = gr.Dropdown(choices=[p["alias"] for p in prompts], label="Select Beast", value=prompts[0]["alias"])
-            character_dropdown = gr.Dropdown(choices=["Beast only", "Wizard", "Warrior"], label="Select Character", value="Beast only")
+            adventurer_id = gr.Number(label="Adventurer ID:")
+            character_dropdown = gr.Dropdown(choices=["Portait", "Last battle", "Loot bag"], label="Select Scene", value="Portait")
             model_dropdown = gr.Dropdown(choices=[m["alias"] for m in models], label="Select Model", value=models[0]["alias"])
         with gr.Row():
             # Add a text box for custom user input (max 200 characters)
             custom_prompt_input = gr.Textbox(label="Custom Prompt (Optional)", placeholder="Enter additional details (max 200 chars)...", max_lines=1, max_length=200)
+            #custom_prompt_input = f""
         with gr.Row():
             generate_button = gr.Button("Generate Image")
         with gr.Row():
             output_image = gr.Image(elem_classes="output-image", label="Generated Image", show_label=False, scale=1, width="100%")
         with gr.Row():
             status_text = gr.Textbox(label="Status", placeholder="Waiting for input...", interactive=False)
-
         # Connect the button to the function
         generate_button.click(
             generate_image,
-            inputs=[prompt_dropdown, 
+            inputs=[adventurer_id,
+                    prompt_dropdown, 
                     custom_prompt_input,
                     character_dropdown,
                     model_dropdown

app_modal.py

@@ -0,0 +1,83 @@
+# gradio_interface.py
+import gradio as gr
+import modal
+from config.config import prompts, models_modal  # Indirect import
+#from img_gen import generate_image
+
+print("Hello from gradio_interface_head!")
+
+# Modal remote function synchronously
+def generate(prompt_dropdown, team_dropdown, model_dropdown, custom_prompt_input, cpu_gpu ="GPU"):
+    # Debug: 
+    debug_message = f"Debug: Button clicked! Inputs - Prompt: {prompt_dropdown}, Team: {team_dropdown}, Model: {model_dropdown}, Custom Prompt: {custom_prompt_input}"
+    print(debug_message)  # Print to console for debugging
+    try:
+        # Check for CPU/GPU dropdown option
+        if cpu_gpu == "GPU":
+            f = modal.Function.from_name("LS-img-gen-modal", "generate_image_gpu")
+        else:
+            f = modal.Function.from_name("LS-img-gen-modal", "generate_image_cpu")
+
+       # Import the remote function
+        image_path, message = f.remote(
+                            prompt_dropdown, 
+                            team_dropdown, 
+                            model_dropdown, 
+                            custom_prompt_input,
+                            )
+        return image_path, message
+    except Exception as e:
+        return None, f"Error calling generate_image function: {e}"
+    
+
+# Gradio Interface
+def gradio_interface():
+    with gr.Blocks(css="""
+                    .gradio-container {
+                        background-image: url(''); 
+                        background-size: cover;
+                        background-position: center;
+                    }
+                    .output-image img {
+                     width: 2500px; /* Force image to fill container width */
+                     object-fit: cover; /* ACTIVATE FOR IMAGE-FIT CONTAINER */
+                    }
+                    """) as demo:
+        gr.Markdown("# ========== Loot Survivor - AI Image Generator ==========")
+        with gr.Row():
+            # Set default values for dropdowns
+            prompt_dropdown = gr.Dropdown(choices=[p["alias"] for p in prompts], label="Select Beast", value=prompts[0]["alias"])
+            character_dropdown = gr.Dropdown(choices=["Beast only", "Wizard", "Warrior"], label="Select Character", value="Beast only")
+            model_dropdown = gr.Dropdown(choices=[m["alias"] for m in models_modal], label="Select Model", value=models_modal[0]["alias"])
+        with gr.Row():
+            # Add a text box for custom user input (max 200 characters)
+            custom_prompt_input = gr.Textbox(label="Custom Prompt (Optional)", placeholder="Enter additional details (max 200 chars)...", max_lines=1, max_length=200)
+        with gr.Row():
+            generate_button = gr.Button("Generate Image")
+        with gr.Row():
+            output_image = gr.Image(elem_classes="output-image", label="Generated Image", show_label=False, scale=1, width="100%")
+        with gr.Row():
+            status_text = gr.Textbox(label="Status", placeholder="Waiting for input...", interactive=False)
+
+
+        # Import the remote function
+        f = modal.Function.from_name("img-gen-modal-gpu", "generate_image")
+        
+        # Connect the button to the function
+        generate_button.click(
+            generate,
+            inputs=[prompt_dropdown, 
+                    custom_prompt_input,
+                    character_dropdown,
+                    model_dropdown
+                    ],
+            outputs=[output_image, status_text]
+        )
+    return demo
+
+# Create the demo instance
+demo = gradio_interface()
+
+# Only launch if running directly
+if __name__ == "__main__":
+        demo.queue().launch()

config/config.py

@@ -1,7 +1,7 @@
 # config.py
 import os
 from config.prompts import prompts 
-from config.models import models
+from config.models import models_modal
 
 # Retrieve the Hugging Face token
 api_token = os.getenv("HF_TOKEN")
@@ -9,4 +9,4 @@ api_token = os.getenv("HF_TOKEN")
 # Debugging: Print prompt and model options
 print("##### IMPORTING CONFIG #####")
 print("Prompt Options:", [p["alias"] for p in prompts])
-print("Model Options:", [m["alias"] for m in models])
+print("Model Options:", [m["alias"] for m in models_modal])

config/models.py

@@ -3,3 +3,11 @@ models = [
     {"alias": "FLUX.1-dev", "name": "black-forest-labs/FLUX.1-dev"},
     {"alias": "Midjourney", "name": "strangerzonehf/Flux-Midjourney-Mix2-LoRA"},
 ]
+
+models_modal = [
+    {"alias": "FLUX.1-dev_modal_local", "name": "FLUX.1-dev"},    
+    #{"alias": "FLUX.1-schnell_modal_local", "name": "FLUX.1-schnell"},
+    #{"alias": "FLUX.1-dev", "name": "black-forest-labs/FLUX.1-dev"},
+    #{"alias": "Midjourney", "name": "strangerzonehf/Flux-Midjourney-Mix2-LoRA"},
+    #{"alias": "FLUX.1-schnell", "name": "black-forest-labs/FLUX.1-schnell"},   
+]

metadata/metadata copy 2.py

@@ -0,0 +1,146 @@
+import requests
+import json  # Import the json module for saving data
+
+# Define the GraphQL endpoint
+url = "https://ls-indexer-sepolia.provable.games/graphql"
+
+# Define the GraphQL queries
+query = """
+query MyQuery {
+  adventurers(limit: 10, where: {id: {eq: 555}}) {
+    owner
+    id
+    name
+    strength
+    vitality
+    dexterity
+    intelligence
+    wisdom
+    charisma
+    level
+    xp
+    health
+    beastHealth
+    head
+    hand
+    chest
+    waist
+    foot
+    weapon
+    gold
+    neck
+    ring
+    luck
+    battleActionCount
+    customRenderer
+    statUpgrades
+  }
+}
+"""
+
+query2 = """
+query MyQuery {
+  battles(where: {adventurerId: {eq: 555}}) {
+    adventurerId
+    adventurerHealth
+    beast
+    beastHealth
+    beastLevel
+    seed
+    attacker
+    fled
+    damageDealt
+    criticalHit
+    damageTaken
+    damageLocation
+    xpEarnedAdventurer
+    xpEarnedItems
+    goldEarned
+    discoveryTime
+  }
+}
+"""
+
+# Define the request payloads
+payload = {
+    "query": query
+}
+payload2 = {
+    "query": query2
+}
+
+# Send the POST requests to the GraphQL API
+response = requests.post(url, json=payload)
+response2 = requests.post(url, json=payload2)
+
+# Check if the requests were successful
+if response.status_code == 200 and response2.status_code == 200:
+    # Parse the JSON responses
+    data = response.json()
+    data2 = response2.json()
+    print("Data fetched successfully.")
+
+    # Save the data to files
+    with open("adventurers_data.json", "w") as file:
+        json.dump(data, file, indent=4)  # Save with pretty-printing (indent=4)
+    with open("adventurers_data2.json", "w") as file:
+        json.dump(data2, file, indent=4)  # Save with pretty-printing (indent=4)
+    print("Data saved to 'adventurers_data.json' and 'adventurers_data2.json'.")
+
+    # Extract the list of adventurers from both queries
+    adventurers = data.get("data", {}).get("adventurers", [])
+    battles = data2.get("data", {}).get("battles", [])
+
+    # Create a dictionary to map adventurers by their ID for quick lookup
+    adventurers_dict = {adv["id"]: adv for adv in adventurers}
+
+    # Add fields from the second query to the corresponding adventurer
+    for adv2 in battles:
+        adventurer_id = adv2["adventurerId"]
+        if adventurer_id in adventurers_dict:
+            print("here")
+            # Add new fields to the existing adventurer
+            adventurers_dict[adventurer_id].update(adv2)
+        else:
+            print("else")
+
+    # Print each adventurer's details dynamically
+    for adventurer in adventurers_dict.values():
+        # Create a dictionary to store field values
+        adventurer_data = adventurer  # Use the updated dictionary
+
+        print("\n=====Adventurer Details=====")
+        for key, value in adventurer_data.items():
+            print(f"{key.capitalize()}: {value}")
+
+        if adventurer_data['health'] != 0:
+            print("\n=====THE ADVENTURER IS STILL ALIVE=====")
+        else:
+            print("\n=====THE ADVENTURER IS DEAD=====")
+
+        # Example: Access specific fields
+        print(f"\nEquipment list of {adventurer_data['name']}:")
+        print(f"\nAdventurer Head: {adventurer_data.get('head', 'None')}")
+        print(f"Hand: {adventurer_data.get('hand', 'None')}")
+        print(f"Chest: {adventurer_data.get('chest', 'None')}")
+        print(f"Waist: {adventurer_data.get('waist', 'None')}")
+        print(f"Foot: {adventurer_data.get('foot', 'None')}")
+        print(f"Weapon: {adventurer_data.get('weapon', 'None')}")
+        print(f"Last combat: {adventurer_data.get('beast', 'Unknown')}")
+        print(f"Beast Level: {adventurer_data.get('beastLevel', 'Unknown')}")
+        print(f"Attacker: {adventurer_data.get('attacker', 'Unknown')}")
+        print(f"Fled: {adventurer_data.get('fled', 'Unknown')}")
+        print(f"Damage Dealt: {adventurer_data.get('damageDealt', 'Unknown')}")
+        print(f"Damage Taken: {adventurer_data.get('damageTaken', 'Unknown')}")
+        print(f"Crticial Hit: {adventurer_data.get('criticalHit', 'Unknown')}")
+        print(f"Damage Location: {adventurer_data.get('damageLocation', 'Unknown')}")
+        print(f"Beast Health: {adventurer_data.get('beastHealth', 'Unknown')}")
+        print(f"Adventure Health: {adventurer_data.get('adventurerHealth', 'Unknown')}")
+
+else:
+    # Print detailed error information
+    print(f"Failed to fetch data. Status codes: {response.status_code}, {response2.status_code}")
+    print("Response 1 Headers:", response.headers)
+    print("Response 1 Body:", response.text)
+    print("Response 2 Headers:", response2.headers)
+    print("Response 2 Body:", response2.text)

metadata/metadata copy.py

@@ -0,0 +1,99 @@
+import requests
+import json  # Import the json module for saving data
+
+# Define the GraphQL endpoint
+url = "https://ls-indexer-sepolia.provable.games/graphql"
+
+# Define the GraphQL query
+query = """
+query MyQuery {
+  adventurers(limit: 10, where: {id: {eq: 555}}) {
+    owner
+    id
+    name
+    strength
+    vitality
+    dexterity
+    intelligence
+    wisdom
+    charisma
+    level
+    xp
+    health
+    beastHealth
+    head
+    hand
+    chest
+    waist
+    foot
+    weapon
+    gold
+    neck
+    ring
+    luck
+    battleActionCount
+    customRenderer
+    statUpgrades
+  }
+}
+"""
+
+# Define the request payload
+payload = {
+    "query": query
+}
+
+# Send the POST request to the GraphQL API
+response = requests.post(url, json=payload)
+
+# Check if the request was successful
+if response.status_code == 200:
+    # Parse the JSON response
+    data = response.json()
+    print("Data fetched successfully:")
+    print(data)
+    # Save the data to a file
+    with open("adventurers_data.json", "w") as file:
+        json.dump(data, file, indent=4)  # Save with pretty-printing (indent=4)
+    print("Data saved to 'adventurers_data.json'.")
+
+    # Extract the list of adventurers
+    adventurers = data.get("data", {}).get("adventurers", [])
+
+    # Print each adventurer's details dynamically
+    for adventurer in adventurers:
+        # Assign the 'hand' and 'head' fields to variables
+        hand_var = adventurer.get("hand")
+        head_var = adventurer.get("head")
+
+        # Print the variables (for debugging)
+        print(f"Hand: {hand_var}")
+        print(f"Head: {head_var}")
+
+        # Check if 'hand_var' is not None and call action()
+        if hand_var is not None:
+            print(hand_var)
+
+        print("\n=====Adventurer Details=====")
+        for key, value in adventurer.items():
+            print(f"{key.capitalize()}: {value}")
+            adventurer_data[key] = value  # Assign to dictionary
+
+        if heatlh is not 0:
+            print("\n=====THE ADVENTURER IS STILL ALIVE=====")
+        else:
+            print("\n=====THE ADVENTURER IS DEAD=====")
+        # Example: Access specific fields
+        print(f"\nEquipment list of {adventurer_data['name']}:")
+        print(f"\nAdventurer Head: {adventurer_data['head']}")
+        print(f"Hand: {adventurer_data['hand']}")
+        print(f"Chest: {adventurer_data['chest']}")
+        print(f"Waist: {adventurer_data['waist']}")
+        print(f"Foot: {adventurer_data['foot']}")
+        print(f"Weapon: {adventurer_data['weapon']}")
+        print(f"Last combat: {adventurer_data['weapon']}")
+
+
+else:
+    print(f"Failed to fetch data. Status code: {response.status_code}")
+    print(response.text)

metadata/metadata.py

@@ -0,0 +1,161 @@
+import requests
+import json  # Import the json module for saving data
+
+# Define the GraphQL endpoint
+url = "https://ls-indexer-sepolia.provable.games/graphql"
+
+def fetch_metadata(adv_id):
+    if adv_id:
+        print(f"Adventure ID: {adv_id}")
+
+    # Define the GraphQL queries
+    query = """
+    query MyQuery($id: FeltValue!) {
+        adventurers(limit: 10, where: {id: {eq: $id}}) {
+            owner
+            id
+            name
+            strength
+            vitality
+            dexterity
+            intelligence
+            wisdom
+            charisma
+            level
+            xp
+            health
+            beastHealth
+            head
+            hand
+            chest
+            waist
+            foot
+            weapon
+            gold
+            neck
+            ring
+            luck
+            battleActionCount
+            customRenderer
+            statUpgrades
+        }
+    }
+    """
+
+    query2 = """
+    query MyQuery($id: FeltValue!) {
+    battles(where: {adventurerId: {eq: $id}}) {
+        adventurerId
+        adventurerHealth
+        beast
+        beastHealth
+        beastLevel
+        seed
+        attacker
+        fled
+        damageDealt
+        criticalHit
+        damageTaken
+        damageLocation
+        xpEarnedAdventurer
+        xpEarnedItems
+        goldEarned
+        discoveryTime
+    }
+    }
+    """
+
+    variables = {"id": adv_id}
+
+    # Define the request payloads
+    payload = {
+        "query": query,
+        "variables": variables
+    }
+    payload2 = {
+        "query": query2,
+        "variables": variables
+    }
+
+    # Send the POST requests to the GraphQL API
+    response = requests.post(url, json=payload)
+    response2 = requests.post(url, json=payload2)
+
+    # Check if the requests were successful
+    if response.status_code == 200 and response2.status_code == 200:
+        # Parse the JSON responses
+        data = response.json()
+        data2 = response2.json()
+        print("Data fetched successfully.")
+
+        # Save the data to files
+        with open("adventurers_data.json", "w") as file:
+            json.dump(data, file, indent=4)  # Save with pretty-printing (indent=4)
+        with open("adventurers_data2.json", "w") as file:
+            json.dump(data2, file, indent=4)  # Save with pretty-printing (indent=4)
+        print("Data saved to 'adventurers_data.json' and 'adventurers_data2.json'.")
+
+        # Extract the list of adventurers from both queries
+        adventurers = data.get("data", {}).get("adventurers", [])
+        battles = data2.get("data", {}).get("battles", [])
+
+        # Create a dictionary to map adventurers by their ID for quick lookup
+        adventurers_dict = {adv["id"]: adv for adv in adventurers}
+
+        # Add fields from the second query to the corresponding adventurer
+        for adv2 in battles:
+            adventurer_id = adv2["adventurerId"]
+            if adventurer_id in adventurers_dict:
+                print("battles loop")
+                # Add new fields to the existing adventurer
+                adventurers_dict[adventurer_id].update(adv2)
+            else:
+                print("else")
+
+        # Print each adventurer's details dynamically
+        for adventurer in adventurers_dict.values():
+            # Create a dictionary to store field values
+            adventurer_data = adventurer  # Use the updated dictionary
+
+            print("\n=====Adventurer Details=====")
+            for key, value in adventurer_data.items():
+                print(f"{key.capitalize()}: {value}")
+
+            if adventurer_data['health'] != 0:
+                print("\n=====THE ADVENTURER IS STILL ALIVE=====")
+            else:
+                print("\n=====THE ADVENTURER IS DEAD=====")
+
+            # Example: Access specific fields
+            print(f"\nEquipment list of {adventurer_data['name']}:")
+            print(f"\nAdventurer Head: {adventurer_data.get('head', 'None')}")
+            print(f"Hand: {adventurer_data.get('hand', 'None')}")
+            print(f"Chest: {adventurer_data.get('chest', 'None')}")
+            print(f"Waist: {adventurer_data.get('waist', 'None')}")
+            print(f"Foot: {adventurer_data.get('foot', 'None')}")
+            print(f"Weapon: {(weapon := adventurer_data.get('weapon', 'None'))}")
+            print(f"Beast: {adventurer_data.get('beast', 'Unknown')}")
+            print(f"Beast Level: {adventurer_data.get('beastLevel', 'Unknown')}")
+            print(f"Attacker: {adventurer_data.get('attacker', 'Unknown')}")
+            print(f"Fled: {adventurer_data.get('fled', 'Unknown')}")
+            print(f"Damage Dealt: {adventurer_data.get('damageDealt', 'Unknown')}")
+            print(f"Damage Taken: {adventurer_data.get('damageTaken', 'Unknown')}")
+            print(f"Crticial Hit: {adventurer_data.get('criticalHit', 'Unknown')}")
+            print(f"Damage Location: {adventurer_data.get('damageLocation', 'Unknown')}")
+            print(f"Beast Health: {adventurer_data.get('beastHealth', 'Unknown')}")
+            print(f"Adventurer Health: {adventurer_data.get('adventurerHealth', 'Unknown')}")
+
+            if adventurer_data.get('weapon') == "Club":
+                print("HELLO!")
+
+            if weapon == "Club":
+                print("HELLO AGAIN!")
+    else:
+        # Print detailed error information
+        print(f"Failed to fetch data. Status codes: {response.status_code}, {response2.status_code}")
+        print("Response 1 Headers:", response.headers)
+        print("Response 1 Body:", response.text)
+        print("Response 2 Headers:", response2.headers)
+        print("Response 2 Body:", response2.text)
+
+    return adventurer_data

requirements.txt

@@ -10,4 +10,5 @@ invisible_watermark
 huggingface_hub[hf_transfer]
 sentencepiece
 opencv-python==4.5.5.64
-gguf
+gguf
+tensorflow

src/check_dependecies.py

@@ -0,0 +1,22 @@
+print("Running debug check...")
+# Debug function to check installed packages
+def check_dependencies():
+    packages = [
+        "diffusers",  # For Stable Diffusion
+        "transformers",  # For Hugging Face models
+        "torch",  # PyTorch
+        "accelerate",  # For distributed training/inference
+        "gradio",  # For the Gradio interface (updated to latest version)
+        "safetensors",  # For safe model loading
+        "pillow",  # For image processing
+        "sentencepiece",
+        "gguf",
+    ]
+
+    for package in packages:
+        try:
+            import importlib
+            module = importlib.import_module(package)
+            print(f" {package} is installed. Version:")
+        except ImportError:
+            print(f" {package} is NOT installed.")

src/img_gen copy.py

@@ -0,0 +1,104 @@
+# img_gen.py
+import sys
+import os
+import random
+from huggingface_hub import InferenceClient, login
+from datetime import datetime
+from config.config import prompts, api_token  
+from config.models import models
+from metadata.metadata import fetch_metadata
+
+def generate_image(
+        adventurer_id,
+        prompt_alias, 
+        custom_prompt, 
+        characer_dropdown,
+        model_alias, 
+        height=360, 
+        width=640, 
+        num_inference_steps=20, 
+        guidance_scale=2.0, 
+        seed=-1):
+    
+    adventurer = fetch_metadata(adventurer_id)
+    print(f"ANDRE {adventurer['name']}")
+
+    prompt = f"A portait of a medieval, fantasy adventurer, equiped with a {adventurer['weapon']} (depending on his weapon make the characer a warrior, or a hunter or a wizard). He is also equiped in the head with a {adventurer['head']}, hands with {adventurer['hand']}, the chest with a {adventurer['chest']}, and the waist with a {adventurer['waist']}. Unreal Engine render style, photorealistic, realistic fantasy style."
+    # Find the selected prompt and model
+    try:
+        #prompt = next(p for p in prompts if p["alias"] == prompt_alias)["text"]
+        model_name = next(m for m in models if m["alias"] == model_alias)["name"]
+
+    except StopIteration:
+        return None, "ERROR: Invalid prompt or model selected."
+
+    # Print the original prompt and dynamic values for debugging
+    print("Original Prompt:")
+    print(prompt)
+
+    # Append the custom character (if provided)
+    if characer_dropdown == "Wizard":
+        prompt += f" A wizard combats using powerful magic against the {prompt_alias}"
+    elif characer_dropdown == "Warrior":
+        prompt += f" A warrior combats using his weapons against the {prompt_alias}"
+    else:
+        pass
+
+   # Append the custom prompt (if provided)
+    if custom_prompt and len(custom_prompt.strip()) > 0:
+        prompt += " " + custom_prompt.strip()
+
+    # Print the formatted prompt for debugging
+    print("\nFormatted Prompt:")
+    print(prompt)
+
+    # Randomize the seed if needed
+    if seed == -1:
+        seed = random.randint(0, 1000000)
+
+    # HF LOGIN 
+    print("Initializing HF TOKEN")
+    print (api_token)
+    # login(token=api_token)
+    # print("model_name:")
+    # print(model_name)
+
+
+    # Initialize the InferenceClient
+    try:
+        print("-----INITIALIZING INFERENCE-----")
+        client = InferenceClient(model_name, token=api_token)
+        print("Inference activated")
+    except Exception as e:
+        return None, f"ERROR: Failed to initialize InferenceClient. Details: {e}"
+
+     #Generate the image
+    try:
+        print("-----GENERATING IMAGE-----")
+        print("-----HOLD ON-----")   
+        image = client.text_to_image(
+            prompt,
+            guidance_scale=guidance_scale,
+            num_inference_steps=num_inference_steps,
+            width=width,
+            height=height,
+            seed=seed
+        )
+        print("-----IMAGE GENERATED SUCCESSFULLY!-----")
+    except Exception as e:
+        return None, f"ERROR: Failed to generate image. Details: {e}"
+
+    # Save the image with a timestamped filename
+    print("-----SAVING-----", image)
+    path = "images"
+    
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    output_filename = f"{path}/{timestamp}_{seed}_{model_alias.replace(' ', '_').lower()}_{prompt_alias.replace(' ', '_').lower()}_{characer_dropdown.replace(' ', '_').lower()}.png"
+    try:
+        image.save(output_filename)
+    except Exception as e:
+        return None, f"ERROR: Failed to save image. Details: {e}"
+    print("-----DONE!-----")
+    print("-----CALL THE BANNERS!-----")
+
+    return output_filename, "Image generated successfully!"

src/img_gen.py

@@ -4,9 +4,12 @@ import os
 import random
 from huggingface_hub import InferenceClient, login
 from datetime import datetime
-from config.config import models, prompts, api_token  # Direct import
+from config.config import prompts, api_token  
+from config.models import models
+from metadata.metadata import fetch_metadata
 
 def generate_image(
+        adventurer_id,
         prompt_alias, 
         custom_prompt, 
         characer_dropdown,
@@ -16,9 +19,27 @@ def generate_image(
         num_inference_steps=20, 
         guidance_scale=2.0, 
         seed=-1):
+    
+    adventurer = fetch_metadata(adventurer_id)
+    print(f"ANDRE {adventurer['name']}")
+
+    # Set the custom prompt variables
+    if characer_dropdown == "Portait":
+        prompt = f"A portait of a medieval, fantasy adventurer, equiped with a weapon: a {adventurer['weapon']} (depending on his weapon, make the characer dressed as a warrior, or as a hunter or as a wizard). He is also equiped in the head with a {adventurer['head']}, the hands with {adventurer['hand']}, the chest with a {adventurer['chest']}, and the waist with a {adventurer['waist']}. Please be sure to use only medieval items that were possble to be made in that period. Unreal Engine render style, photorealistic, atmospheric light, realistic fantasy style."
+    
+    if characer_dropdown == "Last battle":
+        prompt = f"A battle between a medieval fantasy adventurer, and a big {adventurer['beast']} monster. The adventurer is combating with {adventurer['weapon']} (depending on his equipment, make the characer dressed as a warrior, or as a hunter or as a wizard). He is also equiped in the head with {adventurer['head']}, the hands with {adventurer['hand']}, the chest with {adventurer['chest']}, the waist with {adventurer['waist']}, the fet with {adventurer['foot']}. Please sure to use only medieval items that were possble to be made in that period. Add details for the monster as well. is Unreal Engine render style, photorealistic, realistic fantasy style."
+
+    elif characer_dropdown == "Loot bag":
+        prompt = f"A loot bag from a medieval fantasy adventurer and his equipments. On the floor also a {adventurer['weapon']} a {adventurer['head']}, a {adventurer['hand']}, a {adventurer['chest']}, a {adventurer['waist']}, and a {adventurer['foot']}. Please sure to use only medieval items that were possble to be made in that period. Inside the bag also gold coins. Atmospheric light, cavern, dungeon context. Unreal Engine render style, photorealistic, realistic fantasy style."
+    else:       
+        pass
+
+
+
     # Find the selected prompt and model
     try:
-        prompt = next(p for p in prompts if p["alias"] == prompt_alias)["text"]
+        #prompt = next(p for p in prompts if p["alias"] == prompt_alias)["text"]
         model_name = next(m for m in models if m["alias"] == model_alias)["name"]
 
     except StopIteration:
@@ -28,13 +49,6 @@ def generate_image(
     print("Original Prompt:")
     print(prompt)
 
-    # Append the custom character (if provided)
-    if characer_dropdown == "Wizard":
-        prompt += f" A wizard combats using powerful magic against the {prompt_alias}"
-    elif characer_dropdown == "Warrior":
-        prompt += f" A warrior combats using his weapons against the {prompt_alias}"
-    else:
-        pass
 
    # Append the custom prompt (if provided)
     if custom_prompt and len(custom_prompt.strip()) > 0:
@@ -84,8 +98,11 @@ def generate_image(
     print("-----SAVING-----", image)
     path = "images"
     
+    message = f"Image generated successfully! Call the banners! \nID: {adventurer['id']}, NAME: {adventurer['name']}, WEAPON: {adventurer['weapon']}, HEAD: {adventurer['head']}, HAND: {adventurer['hand']}, CHEST: {adventurer['chest']}, WAIST: {adventurer['waist']}, BEAST: {adventurer['beast']}"
+    file_name_extension = f"ID: {adventurer['id']}, NAME: {adventurer['name']}, WEAPON: {adventurer['weapon']}, HEAD: {adventurer['head']}, HAND: {adventurer['hand']}, CHEST: {adventurer['chest']}, WAIST: {adventurer['waist']}, BEAST: {adventurer['beast']}"
+
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-    output_filename = f"{path}/{timestamp}_{seed}_{model_alias.replace(' ', '_').lower()}_{prompt_alias.replace(' ', '_').lower()}_{characer_dropdown.replace(' ', '_').lower()}.png"
+    output_filename = f"{path}/{timestamp}_{seed}_{model_alias.replace(' ', '_').lower()}_{prompt_alias.replace(' ', '_').lower()}_{characer_dropdown.replace(' ', '_').lower()}_{file_name_extension.replace(' ', '_').lower()}.png"
     try:
         image.save(output_filename)
     except Exception as e:
@@ -93,4 +110,4 @@ def generate_image(
     print("-----DONE!-----")
     print("-----CALL THE BANNERS!-----")
 
-    return output_filename, "Image generated successfully!"
+    return output_filename, message

src/img_gen_modal.py

@@ -0,0 +1,205 @@
+#img_gen_modal.py
+import modal
+import random
+import io
+from config.config import prompts, api_token 
+from config.models import models_modal
+import os
+import gradio as gr
+import torch
+import sentencepiece
+import torch
+from huggingface_hub import login
+from transformers import AutoTokenizer
+import random
+from datetime import datetime
+from diffusers.callbacks import SDXLCFGCutoffCallback
+from diffusers import FluxPipeline
+from diffusers import DPMSolverMultistepScheduler, StableDiffusionPipeline, AutoencoderTiny, AutoencoderKL, DiffusionPipeline, FluxTransformer2DModel, GGUFQuantizationConfig
+from PIL import Image
+from src.check_dependecies import check_dependencies
+import numpy as np
+
+MAX_SEED = np.iinfo(np.int32).max
+MAX_IMAGE_SIZE = 2048
+
+CACHE_DIR = "/model_cache"
+
+# Define the Modal image
+image = (
+    modal.Image.from_registry("nvidia/cuda:12.2.0-devel-ubuntu22.04", add_python="3.9")
+                .pip_install_from_requirements("requirements.txt")
+    #modal.Image.debian_slim(python_version="3.9")  # Base image
+    # .apt_install(
+    #     "git",
+    # )
+    # .pip_install(
+    #     "diffusers",
+    #     f"git+https://github.com/huggingface/transformers.git"
+    # )
+    .env(
+        {
+            "HF_HUB_ENABLE_HF_TRANSFER": "1", "HF_HOME": "HF_HOME", "HF_HUB_CACHE": CACHE_DIR
+        }
+    )
+)
+
+# Create a Modal app
+app = modal.App("LS-img-gen-modal", image=image)
+with image.imports():
+    import os
+
+flux_model_vol = modal.Volume.from_name("flux-model-vol", create_if_missing=True)  # Reference your volume
+
+# GPU FUNCTION
+@app.function(volumes={"/data": flux_model_vol},
+              secrets=[modal.Secret.from_name("huggingface-token")],
+              gpu="L40S",
+              timeout = 300
+              )
+# MAIN GENERATE IMAGE FUNCTION
+def generate_image_gpu(
+        prompt_alias, 
+        custom_prompt, 
+        characer_dropdown,
+        model_alias, 
+        height=360, 
+        width=640, 
+        num_inference_steps=20, 
+        guidance_scale=2.0, 
+        seed=-1):
+    # Find the selected prompt and model
+    print("Hello from LS_img_gen!")
+
+    check_dependencies()
+
+    try:
+        prompt = next(p for p in prompts if p["alias"] == prompt_alias)["text"]
+        model_name = next(m for m in models_modal if m["alias"] == model_alias)["name"]
+
+    except StopIteration:
+        return None, "ERROR: Invalid prompt or model selected."
+
+    # Print the original prompt and dynamic values for debugging
+    print("Original Prompt:")
+    print(prompt)
+
+    # Append the custom character (if provided)
+    if characer_dropdown == "Wizard":
+        prompt += f" A wizard combats using powerful magic against the {prompt_alias}"
+    elif characer_dropdown == "Warrior":
+        prompt += f" A warrior combats using his weapons against the {prompt_alias}"
+    else:
+        pass
+
+   # Append the custom prompt (if provided)
+    if custom_prompt and len(custom_prompt.strip()) > 0:
+        prompt += " " + custom_prompt.strip()
+
+    # Print the formatted prompt for debugging
+    print("\nFormatted Prompt:")
+    print(prompt)
+
+    # Randomize the seed if needed
+    if seed == -1:
+        seed = random.randint(0, 1000000)
+
+    # HF LOGIN 
+    print("Initializing HF TOKEN")
+    print (api_token)
+    # login(token=api_token)
+    # print("model_name:")
+    # print(model_name)
+
+
+    # Use absolute path with leading slash
+    model_path = f"/data/{model_name}"  # Changed from "data/" to "/data/"
+    print(f"Loading model from local path: {model_path}")
+    
+    # Debug: Check if the directory exists and list its contents
+    if os.path.exists(model_path):
+        print("Directory exists. Contents:")
+        for item in os.listdir(model_path):
+            print(f" - {item}")
+    else:
+        print(f"Directory does not exist: {model_path}")
+        print("Contents of /data:")
+        print(os.listdir("/data"))
+    # CHECK FOR TORCH USING CUDA
+    print("CHECK FOR TORCH USING CUDA")
+    print(f"CUDA available: {torch.cuda.is_available()}")
+    if torch.cuda.is_available():
+        print("inside if")
+        print(f"CUDA device count: {torch.cuda.device_count()}")
+        print(f"Current device: {torch.cuda.current_device()}")
+        print(f"Device name: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+
+    try:
+        print("-----INITIALIZING PIPE-----")
+        pipe = FluxPipeline.from_pretrained(
+            model_path,
+            torch_dtype=torch.bfloat16,
+            #torch_dtype=torch.float16,
+            #torch_dtype=torch.float32,
+            #vae=taef1,
+            local_files_only=True,
+        )
+        #torch.cuda.empty_cache()
+
+        if torch.cuda.is_available():
+            print("CUDA available")
+            print("using gpu")
+            pipe = pipe.to("cuda")
+            pipe_message = "CUDA"
+            #pipe.enable_model_cpu_offload()  # official recommended method but is running slower w it
+        else:
+            print("CUDA not available")
+            print("using cpu")
+            pipe = pipe.to("cpu")
+            pipe_message = "CPU"
+        print(f"-----{pipe_message} PIPE INITIALIZED-----")
+        print(f"Using device: {pipe.device}")
+    except Exception as e:
+        print(f"Detailed error: {str(e)}")
+        return None, f"ERROR: Failed to initialize PIPE2. Details: {e}"
+
+    ########## SENDING IMG GEN TO PIPE - WORKING CODE ##########
+    try:
+        print("-----SENDING IMG GEN TO PIPE-----")
+        print("-----HOLD ON-----")   
+        image = pipe(
+            prompt,
+            guidance_scale=guidance_scale,
+            num_inference_steps=num_inference_steps,
+            width=width,
+            height=height,
+            max_sequence_length=512,
+            #callback_on_step_end=decode_tensors,
+            #callback_on_step_end_tensor_inputs=["latents"],
+            # seed=seed
+        ).images[0]
+        #############################################################
+
+        print("-----IMAGE GENERATED SUCCESSFULLY!-----")
+        print(image)  
+                
+    except Exception as e:
+        return f"ERROR: Failed to initialize InferenceClient. Details: {e}"
+        
+    try:
+        # Save the image with a timestamped filename
+        print("-----SAVING-----", image)
+
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        output_filename = f"/data/LS_images/{timestamp}_{seed}_{model_alias.replace(' ', '_').lower()}_{prompt_alias.replace(' ', '_').lower()}_{characer_dropdown.replace(' ', '_').lower()}.png"
+        try:
+            image.save(output_filename)
+        except Exception as e:
+            return None, f"ERROR: Failed to save image. Details: {e}"
+        print("-----DONE!-----")
+        print("-----CALL THE BANNERS!-----")
+
+    except Exception as e:
+        print(f"ERROR: Failed to save image. Details: {e}")
+    # Return the filename and success message
+    return image, "Image generated successfully!"