dynamic data

.backup/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()

.backup/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!"

.gitignore

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

adventurers_data.json

@@ -2,32 +2,32 @@
     "data": {
         "adventurers": [
             {
-                "owner": "0x5182d69e155054d4852c775f3da3df704fd12fd77ebcc97b0a4ec3d619f60e6",
-                "id": 463,
-                "name": "The Syndicate #140",
-                "strength": 0,
-                "vitality": 2,
-                "dexterity": 0,
-                "intelligence": 1,
-                "wisdom": 4,
-                "charisma": 2,
-                "level": 2,
-                "xp": 4,
+                "owner": "0x16c8b6168e8853b6bcc67f046f67d37abf214eb90172f1b184583d46541ed68",
+                "id": 750,
+                "name": "Blobert #736",
+                "strength": 3,
+                "vitality": 1,
+                "dexterity": 2,
+                "intelligence": 0,
+                "wisdom": 3,
+                "charisma": 4,
+                "level": 5,
+                "xp": 28,
                 "health": 0,
-                "beastHealth": 0,
-                "head": null,
-                "hand": null,
-                "chest": null,
-                "waist": null,
-                "foot": null,
-                "weapon": "Wand",
-                "gold": 25,
+                "beastHealth": 74,
+                "head": "Helm",
+                "hand": "Gloves",
+                "chest": "Shirt",
+                "waist": "Sash",
+                "foot": "Heavy Boots",
+                "weapon": "Grimoire",
+                "gold": 0,
                 "neck": null,
                 "ring": null,
-                "luck": 0,
+                "luck": 2,
                 "battleActionCount": 0,
                 "customRenderer": null,
-                "statUpgrades": 1
+                "statUpgrades": 0
             }
         ]
     }

adventurers_data2.json

@@ -2,12 +2,12 @@
     "data": {
         "battles": [
             {
-                "adventurerId": 463,
+                "adventurerId": 750,
                 "adventurerHealth": 90,
-                "beast": "Orc",
+                "beast": "Bear",
                 "beastHealth": 3,
                 "beastLevel": 1,
-                "seed": "0x1cf",
+                "seed": "0x2ee",
                 "attacker": "Beast",
                 "fled": false,
                 "damageDealt": 0,
@@ -17,12 +17,12 @@
                 "xpEarnedAdventurer": 0,
                 "xpEarnedItems": 0,
                 "goldEarned": 0,
-                "discoveryTime": "2024-09-10T18:35:14.764Z"
+                "discoveryTime": "2024-09-10T18:36:03.159Z"
             },
             {
-                "adventurerId": 463,
+                "adventurerId": 750,
                 "adventurerHealth": 90,
-                "beast": "Orc",
+                "beast": "Spider",
                 "beastHealth": 0,
                 "beastLevel": 1,
                 "seed": "0x5a60e008",
@@ -35,7 +35,133 @@
                 "xpEarnedAdventurer": 4,
                 "xpEarnedItems": 8,
                 "goldEarned": 0,
-                "discoveryTime": "2024-09-10T18:41:37.103Z"
+                "discoveryTime": "2024-09-17T00:38:39.843Z"
+            },
+            {
+                "adventurerId": 750,
+                "adventurerHealth": 107,
+                "beast": "Golem",
+                "beastHealth": 0,
+                "beastLevel": 6,
+                "seed": "0xc7bea6f8",
+                "attacker": "Adventurer",
+                "fled": true,
+                "damageDealt": 0,
+                "criticalHit": false,
+                "damageTaken": 0,
+                "damageLocation": null,
+                "xpEarnedAdventurer": 0,
+                "xpEarnedItems": 0,
+                "goldEarned": 0,
+                "discoveryTime": "2024-09-17T00:44:32.620Z"
+            },
+            {
+                "adventurerId": 750,
+                "adventurerHealth": 99,
+                "beast": "Leprechaun",
+                "beastHealth": 22,
+                "beastLevel": 9,
+                "seed": "0x46fe05ca",
+                "attacker": "Adventurer",
+                "fled": false,
+                "damageDealt": 0,
+                "criticalHit": false,
+                "damageTaken": 0,
+                "damageLocation": null,
+                "xpEarnedAdventurer": 0,
+                "xpEarnedItems": 0,
+                "goldEarned": 0,
+                "discoveryTime": "2024-09-17T00:47:42.533Z"
+            },
+            {
+                "adventurerId": 750,
+                "adventurerHealth": 99,
+                "beast": "Leprechaun",
+                "beastHealth": 22,
+                "beastLevel": 9,
+                "seed": "0x46fe05ca",
+                "attacker": "Beast",
+                "fled": false,
+                "damageDealt": 0,
+                "criticalHit": true,
+                "damageTaken": 16,
+                "damageLocation": "Waist",
+                "xpEarnedAdventurer": 0,
+                "xpEarnedItems": 0,
+                "goldEarned": 0,
+                "discoveryTime": "2024-09-17T00:47:42.533Z"
+            },
+            {
+                "adventurerId": 750,
+                "adventurerHealth": 92,
+                "beast": "Leprechaun",
+                "beastHealth": 22,
+                "beastLevel": 9,
+                "seed": "0x46fe05ca",
+                "attacker": "Adventurer",
+                "fled": false,
+                "damageDealt": 0,
+                "criticalHit": false,
+                "damageTaken": 0,
+                "damageLocation": null,
+                "xpEarnedAdventurer": 0,
+                "xpEarnedItems": 0,
+                "goldEarned": 0,
+                "discoveryTime": "2024-09-17T00:47:42.533Z"
+            },
+            {
+                "adventurerId": 750,
+                "adventurerHealth": 92,
+                "beast": "Leprechaun",
+                "beastHealth": 22,
+                "beastLevel": 9,
+                "seed": "0x46fe05ca",
+                "attacker": "Beast",
+                "fled": false,
+                "damageDealt": 0,
+                "criticalHit": false,
+                "damageTaken": 7,
+                "damageLocation": "Hand",
+                "xpEarnedAdventurer": 0,
+                "xpEarnedItems": 0,
+                "goldEarned": 0,
+                "discoveryTime": "2024-09-17T00:47:42.533Z"
+            },
+            {
+                "adventurerId": 750,
+                "adventurerHealth": 92,
+                "beast": "Leprechaun",
+                "beastHealth": 0,
+                "beastLevel": 9,
+                "seed": "0x46fe05ca",
+                "attacker": "Adventurer",
+                "fled": true,
+                "damageDealt": 0,
+                "criticalHit": false,
+                "damageTaken": 0,
+                "damageLocation": null,
+                "xpEarnedAdventurer": 0,
+                "xpEarnedItems": 0,
+                "goldEarned": 0,
+                "discoveryTime": "2024-09-17T00:47:42.533Z"
+            },
+            {
+                "adventurerId": 750,
+                "adventurerHealth": 0,
+                "beast": "Griffin",
+                "beastHealth": 74,
+                "beastLevel": 13,
+                "seed": "0xe1682360",
+                "attacker": "Beast",
+                "fled": false,
+                "damageDealt": 0,
+                "criticalHit": true,
+                "damageTaken": 190,
+                "damageLocation": "Chest",
+                "xpEarnedAdventurer": 0,
+                "xpEarnedItems": 0,
+                "goldEarned": 0,
+                "discoveryTime": "2024-09-17T00:53:20.627Z"
             }
         ]
     }

app.py

@@ -1,5 +1,6 @@
 # app.py 
-from config.config import prompts, api_token  # Direct import
+# venv w python3.11 
+from config.config import api_token  # Direct import
 from config.models import models
 import gradio as gr
 from src.img_gen import generate_image
@@ -23,12 +24,12 @@ def gradio_interface():
             # Set default values for dropdowns
             #prompt_dropdown = gr.Dropdown(choices=[p["alias"] for p in prompts], label="Select Beast", value=prompts[0]["alias"])
             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"])
+            #character_dropdown = gr.Dropdown(choices=["Wizard", "Hunter", "Warrior"], label="Select Character Type", value="Wizard")
+            scene_dropdown = gr.Dropdown(choices=["Adventurer Portait", "Beast Portait", "Last Battle", "Loot Bag"], label="Select Scene", value="Adventurer 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():
@@ -40,9 +41,10 @@ def gradio_interface():
             generate_image,
             inputs=[adventurer_id,
                     #prompt_dropdown, 
+                    #character_dropdown,
+                    scene_dropdown,
+                    #model_dropdown,
                     custom_prompt_input,
-                    character_dropdown,
-                    model_dropdown
                     ],
             outputs=[output_image, status_text]
         )

app_modal.py

@@ -1,15 +1,22 @@
 # gradio_interface.py
 import gradio as gr
 import modal
-from config.config import prompts, models_modal  # Indirect import
+from config.models import 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"):
+def generate(adventurer_id,
+            #prompt_dropdown,
+            #character_dropdown,
+            scene_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}"
+    debug_message = f"Debug: Button clicked! Inputs - ID: {adventurer_id}, Scene: {scene_dropdown}, Custom Prompt: {custom_prompt_input}"
     print(debug_message)  # Print to console for debugging
     try:
         # Check for CPU/GPU dropdown option
@@ -18,11 +25,14 @@ def generate(prompt_dropdown, team_dropdown, model_dropdown, custom_prompt_input
         else:
             f = modal.Function.from_name("LS-img-gen-modal", "generate_image_cpu")
 
+        print ("Sending to external function")
        # Import the remote function
         image_path, message = f.remote(
-                            prompt_dropdown, 
-                            team_dropdown, 
-                            model_dropdown, 
+                            adventurer_id,
+                            #prompt_dropdown, 
+                            #character_dropdown,
+                            scene_dropdown,
+                            #model_dropdown,
                             custom_prompt_input,
                             )
         return image_path, message
@@ -46,12 +56,14 @@ def gradio_interface():
         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"])
+            #prompt_dropdown = gr.Dropdown(choices=[p["alias"] for p in prompts], label="Select Beast", value=prompts[0]["alias"])
+            adventurer_id = gr.Number(label="Adventurer ID:")
+            #character_dropdown = gr.Dropdown(choices=["Wizard", "Hunter", "Warrior"], label="Select Character Type", value="Wizard")
+            scene_dropdown = gr.Dropdown(choices=["Adventurer Portait", "Beast Portait", "Last Battle", "Loot Bag"], label="Select Scene", value="Adventurer Portait")
+            #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)
+            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():
@@ -59,17 +71,15 @@ def gradio_interface():
         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, 
+            inputs=[adventurer_id,
+                    #prompt_dropdown, 
+                    #character_dropdown,
+                    scene_dropdown,
+                    #model_dropdown,
                     custom_prompt_input,
-                    character_dropdown,
-                    model_dropdown
                     ],
             outputs=[output_image, status_text]
         )

config/config.py

@@ -1,6 +1,6 @@
 # config.py
 import os
-from config.prompts import prompts 
+#from config.prompts import prompts 
 from config.models import models_modal
 
 # Retrieve the Hugging Face token
@@ -8,5 +8,5 @@ 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("Prompt Options:", [p["alias"] for p in prompts])
 print("Model Options:", [m["alias"] for m in models_modal])

config/models.py

@@ -6,8 +6,8 @@ models = [
 
 models_modal = [
     {"alias": "FLUX.1-dev_modal_local", "name": "FLUX.1-dev"},    
+    {"alias": "Midjourney", "name": "Flux-Midjourney-Mix2-LoRA"},
     #{"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"},   
 ]

config/prompts.py

@@ -41,3 +41,4 @@ prompts = [
     }
   ]
 
+

early_notes/ls prompts.txt

@@ -0,0 +1,2 @@
+
+help me create a prompt for img gen . it must be a good generic prompt because i will make it dynamicaly changing a variable. The imgs are about some beasts, or monsters, that are part of a fantasy / magical / medieval world. So lets say i will have a list of spider, wolf, bear, etc. And I need a generic prompt like: A massive VARIABLE monster. and then i will dynamicaly change VARIABLE to the monster i need. Make the promot 250 tokns length, and add light details for a realistic but also exciting scene

metadata/metadata copy.py

@@ -63,8 +63,9 @@ if response.status_code == 200:
     # 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")
+        
+        head = adventurer.get("head")
+        hand = adventurer.get("hand")
 
         # Print the variables (for debugging)
         print(f"Hand: {hand_var}")

src/img_gen.py

@@ -4,24 +4,27 @@ import os
 import random
 from huggingface_hub import InferenceClient, login
 from datetime import datetime
-from config.config import prompts, api_token  
+from config.config import api_token  
 from config.models import models
 from metadata.metadata import fetch_metadata
+from src.prompt_gen import prompt_gen
+
 
 def generate_image(
         adventurer_id,
         #prompt_alias, 
+        #character_dropdown,
+        scene_dropdown,
+        #model_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']}")
+    print ("Sending to prompt gen")
+    adventurer, prompt = prompt_gen(adventurer_id, scene_dropdown)
 
     id = adventurer['id']
     name = adventurer['name']
@@ -31,23 +34,11 @@ def generate_image(
     chest_equipment = adventurer['chest']
     waist_equipment = adventurer['waist']
     foot_equipment = adventurer['foot']
+    gold_equipment = adventurer['gold']
     beast_last_battle = adventurer['beast']
 
-    # Set the custom prompt variables
-    if characer_dropdown == "Portait":
-        prompt = f"A portait of a medieval, fantasy adventurer, holding a {weapon_equipment} (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 {head_equipment}, the hands with {hand_equipment}, the chest with a {chest_equipment}, and the waist with a {waist_equipment}. Please be sure to use only medieval items that were possble to be made in that period. Make the adventurer unshaved and dirty, he have been figthing for days down into the dungeons. 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 massive {beast_last_battle} monster. The adventurer is holding a weapon: a {weapon_equipment}. He is also equiped in the head with {head_equipment}, the hands with {hand_equipment}, the chest with {chest_equipment}, the waist with {waist_equipment}, the fet with {foot_equipment}. Use only medieval items that were possble to be made in that period. Add details for the monster as well. The scene location is the natural ambient of the {beast_last_battle}. 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 {weapon_equipment} a {head_equipment}, a {hand_equipment}, a {chest_equipment}, a {waist_equipment}, and a {foot_equipment}. 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
+    model_alias = "FLUX.1-dev"
     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"]
@@ -108,11 +99,13 @@ def generate_image(
     print("-----SAVING-----", image)
     path = "images"
     
-    message = f"Image generated successfully! Call the banners! \n\n=====ADVENTURER GENERATED===== \nID: {id}\nNAME: {name}\nWEAPON: {weapon_equipment}\nHEAD: {head_equipment}\nHAND: {hand_equipment}\nCHEST: {chest_equipment}\nWAIST: {waist_equipment}\nBEAST: {beast_last_battle}"
-    file_name_extension = f"ID: {id}\nNAME: {name}\nWEAPON: {weapon_equipment}\nHEAD: {head_equipment}\nHAND: {hand_equipment}\nCHEST: {chest_equipment}\nWAIST: {waist_equipment}\n LAST BATTLE BEAST: {beast_last_battle}"
+    message = f"Image generated successfully! Call the banners! \n\n=====ADVENTURER GENERATED===== \nID: {id}\nNAME: {name}\nWEAPON: {weapon_equipment}\nHEAD: {head_equipment}\nHAND: {hand_equipment}\nCHEST: {chest_equipment}\nWAIST: {waist_equipment}\nFOOT: {foot_equipment}\nGOLD: {gold_equipment}\nLAST BATTLE BEAST: {beast_last_battle}"
+
+    file_name_extension = f"{id}_{beast_last_battle}"
 
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-    output_filename = f"{path}/{timestamp}_{seed}_{model_alias.replace(' ', '_').lower()}_{characer_dropdown.replace(' ', '_').lower()}_{file_name_extension.replace(' ', '_').lower()}.png"
+
+    output_filename = f"{path}/{timestamp}_{seed}_{model_alias.replace(' ', '_').lower()}_{scene_dropdown.replace(' ', '_').lower()}_{file_name_extension.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"
 

src/img_gen_modal.py

@@ -2,7 +2,8 @@
 import modal
 import random
 import io
-from config.config import prompts, api_token 
+from config.config import api_token 
+from config.prompts import prompts
 from config.models import models_modal
 import os
 import gradio as gr
@@ -19,6 +20,8 @@ from diffusers import DPMSolverMultistepScheduler, StableDiffusionPipeline, Auto
 from PIL import Image
 from src.check_dependecies import check_dependencies
 import numpy as np
+from src.prompt_gen import prompt_gen
+
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
@@ -59,10 +62,12 @@ flux_model_vol = modal.Volume.from_name("flux-model-vol", create_if_missing=True
               )
 # MAIN GENERATE IMAGE FUNCTION
 def generate_image_gpu(
-        prompt_alias, 
+        adventurer_id,
+        #prompt_alias, 
+        #character_dropdown,
+        scene_dropdown,
+        #model_alias,
         custom_prompt, 
-        characer_dropdown,
-        model_alias, 
         height=360, 
         width=640, 
         num_inference_steps=20, 
@@ -71,26 +76,44 @@ def generate_image_gpu(
     # Find the selected prompt and model
     print("Hello from LS_img_gen!")
 
-    check_dependencies()
+    #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"]
+    print ("Sending to prompt gen")
+    adventurer, prompt = prompt_gen(adventurer_id, scene_dropdown)
+    
+    # Print the original prompt and dynamic values for debugging
+    print("Original Prompt:")
+    print(prompt)
 
+    print("Defining advanturer variables")
+    id = adventurer['id']
+    name = adventurer['name']
+    weapon_equipment = adventurer['weapon']
+    head_equipment = adventurer['head']
+    hand_equipment = adventurer['hand']
+    chest_equipment = adventurer['chest']
+    waist_equipment = adventurer['waist']
+    foot_equipment = adventurer['foot']
+    gold_equipment = adventurer['gold']
+    beast_last_battle = adventurer['beast']
+
+    print("Loading Model")
+    model_alias = "FLUX.1-dev"
+    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"]
+        model_name = model_alias
     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 character (if provided)
+    # if character_dropdown == "Wizard":
+    #     prompt += f" A wizard combats using powerful magic against the {beast_last_battle}"
+    # elif character_dropdown == "Warrior":
+    #     prompt += f" A warrior combats using his weapons against the {beast_last_battle}"
+    # else:
+    #     pass
 
    # Append the custom prompt (if provided)
     if custom_prompt and len(custom_prompt.strip()) > 0:
@@ -142,10 +165,13 @@ def generate_image_gpu(
             #torch_dtype=torch.float16,
             #torch_dtype=torch.float32,
             #vae=taef1,
-            local_files_only=True,
+            local_files_only=False,
         )
         #torch.cuda.empty_cache()
 
+        if model_alias == "Flux-Midjourney-Mix2-LoRA":
+            pipe.load_lora_weights("/data/Flux-Midjourney-Mix2-LoRA")
+
         if torch.cuda.is_available():
             print("CUDA available")
             print("using gpu")
@@ -190,8 +216,13 @@ def generate_image_gpu(
         # Save the image with a timestamped filename
         print("-----SAVING-----", image)
 
+        message = f"Image generated successfully! Call the banners! \n\n=====ADVENTURER GENERATED===== \nID: {id}\nNAME: {name}\nWEAPON: {weapon_equipment}\nHEAD: {head_equipment}\nHAND: {hand_equipment}\nCHEST: {chest_equipment}\nWAIST: {waist_equipment}\nFOOT: {foot_equipment}\nGOLD: {gold_equipment}\nLAST BATTLE BEAST: {beast_last_battle}"
+
+        file_name_extension = f"{id}_{beast_last_battle.replace(' ', '_').lower()}"
+
         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"
+        
+        output_filename = f"/data/LS_images/{timestamp}_{seed}_{model_alias.replace(' ', '_').lower()}_{scene_dropdown.replace(' ', '_').lower()}_{file_name_extension.replace(' ', '_').lower()}.png"
         try:
             image.save(output_filename)
         except Exception as e:
@@ -201,5 +232,6 @@ def generate_image_gpu(
 
     except Exception as e:
         print(f"ERROR: Failed to save image. Details: {e}")
+
     # Return the filename and success message
-    return image, "Image generated successfully!"
+    return image, message

src/old/img_gen copy 2.py

@@ -0,0 +1,131 @@
+# 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']}")
+
+    id = adventurer['id']
+    name = adventurer['name']
+    weapon_equipment = adventurer['weapon']
+    head_equipment = adventurer['head']
+    hand_equipment = adventurer['hand']
+    chest_equipment = adventurer['chest']
+    waist_equipment = adventurer['waist']
+    foot_equipment = adventurer['foot']
+    gold_equipment = adventurer['gold']
+    beast_last_battle = adventurer['beast']
+
+    # Set the custom prompt variables
+    if characer_dropdown == "Adventurer Portait":
+        prompt = f"A portait of a medieval, fantasy adventurer, holding only a {weapon_equipment} (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 {head_equipment}, the hands with {hand_equipment}, the chest with a {chest_equipment}, and the waist with a {waist_equipment}. Please be sure to use only medieval items that were possble to be made in that period. Make the adventurer unshaved and dirty, he have been figthing for days down into the dungeons. Unreal Engine render style, photorealistic, atmospheric light, realistic fantasy style."
+
+    if characer_dropdown == "Beast Portait":
+        prompt = f"A massive {beast_last_battle} stands in its natural domain—a place that reflects its primal essence. The beast's eyes burn with an eerie, intelligent light, and its powerful limbs are poised for action, as if ready to pounce or defend its territory. The air is thick with tension, and the ground is littered with signs of its dominance—broken terrain, scattered bones, or remnants of its prey. In the background, a dramatic landscape unfolds: dense forests, jagged mountains, dark caves, or misty swamps. The atmosphere is both terrifying and awe-inspiring, capturing the essence of a fantasy world where magic and monsters reign supreme. 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 massive {beast_last_battle}. The adventurer is holding only a {weapon_equipment}. He is also equiped in the head with {head_equipment}, the hands with {hand_equipment}, the chest with {chest_equipment}, the waist with {waist_equipment}, the feet with {foot_equipment}. Use only medieval items that were possble to be made in that period. The scene location is the natural ambient of the {beast_last_battle}. 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 {weapon_equipment} a {head_equipment}, a {hand_equipment}, a {chest_equipment}, a {waist_equipment}, and a {foot_equipment}. Please sure to use only medieval items that were possble to be made in that period. Inside the bag also {gold_equipment} 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"]
+        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 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"
+    
+    message = f"Image generated successfully! Call the banners! \n\n=====ADVENTURER GENERATED===== \nID: {id}\nNAME: {name}\nWEAPON: {weapon_equipment}\nHEAD: {head_equipment}\nHAND: {hand_equipment}\nCHEST: {chest_equipment}\nWAIST: {waist_equipment}\nLAST BATTLE BEAST: {beast_last_battle}"
+
+    file_name_extension = f"{id} {name} {weapon_equipment} {head_equipment} {hand_equipment} {chest_equipment} {waist_equipment} {beast_last_battle}"
+
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    output_filename = f"{path}/{timestamp}_{seed}_{model_alias.replace(' ', '_').lower()}_{characer_dropdown.replace(' ', '_').lower()}_{file_name_extension.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:
+        return None, f"ERROR: Failed to save image. Details: {e}"
+    print("-----DONE!-----")
+    print("-----CALL THE BANNERS!-----")
+
+    return output_filename, message

src/prompt_gen copy 2.py

@@ -0,0 +1,46 @@
+from metadata.metadata import fetch_metadata
+
+def prompt_gen(adventurer_id, character_dropdown, scene_dropdown):
+    adventurer = fetch_metadata(adventurer_id)
+    print(f"ANDRE {adventurer['name']}")
+
+    id = adventurer['id']
+    name = adventurer['name']
+    weapon_equipment = adventurer['weapon']
+    head_equipment = adventurer['head']
+    hand_equipment = adventurer['hand']
+    chest_equipment = adventurer['chest']
+    waist_equipment = adventurer['waist']
+    foot_equipment = adventurer['foot']
+    gold_equipment = adventurer['gold']
+    beast_last_battle = adventurer['beast']
+
+    # Check for weapon = None to fix eventual generator mistakes
+    if weapon_equipment == None:
+        weapon_sentence = "holding nothing in his hands"
+    else:   
+        weapon_sentence = f"holding only a {weapon_equipment}"
+    
+    if character_dropdown == "Wizard":
+        character_attack_sentence = f"The adventurer is attacking using powerful magic"
+    else:
+        character_attack_sentence = f"The adventurer is attacking using his {weapon_equipment}"
+
+
+    # Set the custom prompt variables
+    if scene_dropdown == "Adventurer Portait":
+        prompt = f"A portait of a medieval, fantasy adventurer, {weapon_sentence}. Make the adventurer dressed as a {character_dropdown}. He is also equiped in the head with a {head_equipment}, the hands with {hand_equipment}, the chest with a {chest_equipment}, and the waist with a {waist_equipment}. Please be sure to use only medieval items that were possble to be made in that period. Make the adventurer unshaved and dirty, he have been figthing for days down into the dungeons. Unreal Engine render style, photorealistic, atmospheric light, realistic fantasy style."
+
+    if scene_dropdown == "Beast Portait":
+        prompt = f"A massive {beast_last_battle} stands in its natural domain—a place that reflects its primal essence. The beast's eyes burn with an eerie, intelligent light, and its powerful limbs are poised for action, as if ready to pounce or defend its territory. It roars menacingly, surrounded by broken weapons and bones. In the background, a dramatic landscape unfolds: dense forests, underground dungeons, jagged mountains, dark caves, or misty swamps. The atmosphere is both terrifying and awe-inspiring, capturing the essence of a fantasy world where magic and monsters reign supreme. Unreal Engine render style, photorealistic, atmospheric light, realistic fantasy style."
+    
+    if scene_dropdown == "Last Battle":
+        prompt = f"A battle between a medieval fantasy adventurer, and a massive {beast_last_battle}. The adventurer is {weapon_sentence}. He is also equiped in the head with {head_equipment}, the hands with {hand_equipment}, the chest with {chest_equipment}, the waist with {waist_equipment}, the feet with {foot_equipment}. Use only medieval items that were possible to be made in that period. The {beast_last_battle} stands in its natural domain—a place that reflects its primal essence. The beast's eyes burn with an eerie, intelligent light, and its powerful limbs are poised for action, as if ready to pounce or defend its territory. It roars menacingly, surrounded by broken weapons and bones. {character_attack_sentence}. In the background, a dramatic landscape unfolds: dense forests, underground dungeons, jagged mountains, dark caves, or misty swamps. The atmosphere is both terrifying and awe-inspiring, capturing the essence of a fantasy world where magic and monsters reign supreme. Unreal Engine render style, photorealistic, realistic fantasy style."
+
+    elif scene_dropdown == "Loot Bag":
+        prompt = f"A loot bag from a medieval fantasy adventurer and his equipments. On the floor also a {weapon_equipment} a {head_equipment}, a {hand_equipment}, a {chest_equipment}, a {waist_equipment}, and a {foot_equipment}. Please sure to use only medieval items that were possble to be made in that period. Inside the bag {gold_equipment} gold coins. Atmospheric light, cavern, dungeon context. Unreal Engine render style, photorealistic, realistic fantasy style."
+    else:       
+        pass
+
+    return adventurer, prompt
+

src/prompt_gen copy.py

@@ -0,0 +1,34 @@
+from metadata.metadata import fetch_metadata
+
+def prompt_gen(adventurer_id, characer_dropdown):
+    adventurer = fetch_metadata(adventurer_id)
+    print(f"ANDRE {adventurer['name']}")
+
+    id = adventurer['id']
+    name = adventurer['name']
+    weapon_equipment = adventurer['weapon']
+    head_equipment = adventurer['head']
+    hand_equipment = adventurer['hand']
+    chest_equipment = adventurer['chest']
+    waist_equipment = adventurer['waist']
+    foot_equipment = adventurer['foot']
+    gold_equipment = adventurer['gold']
+    beast_last_battle = adventurer['beast']
+
+    # Set the custom prompt variables
+    if characer_dropdown == "Adventurer Portait":
+        prompt = f"A portait of a medieval, fantasy adventurer, holding only a {weapon_equipment} (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 {head_equipment}, the hands with {hand_equipment}, the chest with a {chest_equipment}, and the waist with a {waist_equipment}. Please be sure to use only medieval items that were possble to be made in that period. Make the adventurer unshaved and dirty, he have been figthing for days down into the dungeons. Unreal Engine render style, photorealistic, atmospheric light, realistic fantasy style."
+
+    if characer_dropdown == "Beast Portait":
+        prompt = f"A massive {beast_last_battle} stands in its natural domain—a place that reflects its primal essence. The beast's eyes burn with an eerie, intelligent light, and its powerful limbs are poised for action, as if ready to pounce or defend its territory. The air is thick with tension, and the ground is littered with signs of its dominance—broken terrain, scattered bones, or remnants of its prey. In the background, a dramatic landscape unfolds: dense forests, jagged mountains, dark caves, or misty swamps. The atmosphere is both terrifying and awe-inspiring, capturing the essence of a fantasy world where magic and monsters reign supreme. 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 massive {beast_last_battle}. The adventurer is holding only a {weapon_equipment}. He is also equiped in the head with {head_equipment}, the hands with {hand_equipment}, the chest with {chest_equipment}, the waist with {waist_equipment}, the feet with {foot_equipment}. Use only medieval items that were possible to be made in that period. The {beast_last_battle} stands in its natural domain—a place that reflects its primal essence. The beast's eyes burn with an eerie, intelligent light, and its powerful limbs are poised for action, as if ready to pounce or defend its territory. The the ground is littered with signs of its dominance—broken terrain, scattered bones, or remnants of its prey. The adventurer attacks using his {weapon_equipment} or powerful magic. In the background, a dramatic landscape unfolds: dense forests, jagged mountains, dark caves, or misty swamps. The atmosphere is both terrifying and awe-inspiring, capturing the essence of a fantasy world where magic and monsters reign supreme. 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 {weapon_equipment} a {head_equipment}, a {hand_equipment}, a {chest_equipment}, a {waist_equipment}, and a {foot_equipment}. Please sure to use only medieval items that were possble to be made in that period. Inside the bag also {gold_equipment} gold coins. Atmospheric light, cavern, dungeon context. Unreal Engine render style, photorealistic, realistic fantasy style."
+    else:       
+        pass
+
+    return adventurer, prompt
+

src/prompt_gen.py

@@ -0,0 +1,46 @@
+from metadata.metadata import fetch_metadata
+
+def prompt_gen(adventurer_id, scene_dropdown):
+    adventurer = fetch_metadata(adventurer_id)
+    print(f"ANDRE {adventurer['name']}")
+
+    id = adventurer['id']
+    name = adventurer['name']
+    weapon_equipment = adventurer['weapon']
+    head_equipment = adventurer['head']
+    hand_equipment = adventurer['hand']
+    chest_equipment = adventurer['chest']
+    waist_equipment = adventurer['waist']
+    foot_equipment = adventurer['foot']
+    gold_equipment = adventurer['gold']
+    beast_last_battle = adventurer['beast']
+
+    # Check for weapon = None to fix eventual generator mistakes
+    if weapon_equipment == None:
+        weapon_sentence = "holding nothing in his hands"
+    else:   
+        weapon_sentence = f"holding only a {weapon_equipment}"
+    
+    # if character_dropdown == "Wizard":
+    #     character_attack_sentence = f"The adventurer is attacking using powerful magic"
+    # else:
+    #     character_attack_sentence = f"The adventurer is attacking using his {weapon_equipment}"
+
+
+    # Set the custom prompt variables
+    if scene_dropdown == "Adventurer Portait":
+        prompt = f"A portait of a medieval, fantasy adventurer, {weapon_sentence} (depending on his weapon, make the character dressed as a warrior, or as a hunter or as a wizard). He is also equiped in the head with a {head_equipment}, the hands with {hand_equipment}, the chest with a {chest_equipment}, and the waist with a {waist_equipment}. Please be sure to use only medieval items that were possble to be made in that period. Make the adventurer unshaved and dirty, he have been figthing for days down into the dungeons. Low light environment. Unreal Engine render style, photorealistic, atmospheric dark light, realistic fantasy style."
+
+    if scene_dropdown == "Beast Portait":
+        prompt = f"A massive {beast_last_battle} stands in its natural domain—a place that reflects its primal essence. The beast's eyes burn with an eerie, intelligent light, and its powerful limbs are poised for action, as if ready to pounce or defend its territory. It roars menacingly surrounded by broken weapons and bones of his past preys. Torchlit environment, underground dungeons, or dense forests, or jagged mountains, or dark caves, or misty damp swamps. Unreal Engine render style, photorealistic, atmospheric dark lights, realistic fantasy style."
+    
+    if scene_dropdown == "Last Battle":
+        prompt = f"A battle between a medieval fantasy adventurer, and a massive {beast_last_battle}. The adventurer is {weapon_sentence}. He is also equiped in the head with {head_equipment}, the hands with {hand_equipment}, the chest with {chest_equipment}, the waist with {waist_equipment}, the feet with {foot_equipment}. Use only medieval items that were possible to be made in that period. The {beast_last_battle} stands in its natural domain—a place that reflects its primal essence. The beast's eyes burn with an eerie, intelligent light, and its powerful limbs are poised for action, as if ready to pounce or defend its territory. It roars menacingly, surrounded by broken weapons and bones. The adventurer attacks using his {weapon_equipment} or powerful magic. Torchlit environment, underground dungeons, or dense forests, or jagged mountains, or dark caves, or misty damp swamps. Unreal Engine render style, photorealistic, realistic fantasy style."
+
+    elif scene_dropdown == "Loot Bag":
+        prompt = f"A loot bag from a medieval fantasy adventurer and his equipments. On the floor also a {weapon_equipment} a {head_equipment}, a {hand_equipment}, a {chest_equipment}, a {waist_equipment}, and a {foot_equipment}. Please sure to use only medieval items that were possble to be made in that period. Inside the bag {gold_equipment} gold coins. Atmospheric light, cavern, dungeon context. Torchlit environment. Unreal Engine render style, photorealistic, realistic fantasy style."
+    else:       
+        pass
+
+    return adventurer, prompt
+

src/temp.py

@@ -13,4 +13,9 @@ Dungeon context, stones, fire and low light.
     elif characer_dropdown == "Loot bag":
         prompt = f"A loot bag from a medieval fantasy adventurer and his equipments. On the floor also a {weapon_equipment} a {head_equipment}, a {hand_equipment}, a {chest_equipment}, a {waist_equipment}, and a {foot_equipment}. 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
+        pass
+
+
+
+keywords:
+- Torchlit environment

tools/clean_filenames.py

@@ -0,0 +1,26 @@
+import os
+import re
+
+# Define the directory containing the files
+directory = "images/temp/"
+
+# Function to clean file names
+def clean_filename(filename):
+    # Replace spaces, special characters, and non-alphanumeric characters with underscores
+    cleaned_name = re.sub(r'[^a-zA-Z0-9_.-]', '_', filename)
+    return cleaned_name
+
+# Loop through files in the directory
+for filename in os.listdir(directory):
+    # Get the full path of the file
+    full_path = os.path.join(directory, filename)
+    
+    # Skip directories (if any)
+    if os.path.isfile(full_path):
+        # Clean the file name
+        new_name = clean_filename(filename)
+        
+        # Rename the file
+        new_full_path = os.path.join(directory, new_name)
+        os.rename(full_path, new_full_path)
+        print(f"Renamed: {filename} -> {new_name}")

tools/create_dir.py

@@ -0,0 +1,80 @@
+# img_gen.py
+#img_gen_modal.py
+# img_gen.py
+# img_gen_modal.py
+import modal
+import random
+import io
+import os
+
+
+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")
+    modal.Image.debian_slim(python_version="3.9")  # Base image
+
+    .apt_install(
+        "git",
+    )
+    .pip_install(
+
+    )
+    .env(
+        {
+            "HF_HUB_ENABLE_HF_TRANSFER": "1", "HF_HOME": "HF_HOME", "HF_HUB_CACHE": CACHE_DIR
+        }
+    )
+)
+
+# Create a Modal app
+app = modal.App("tools-test-dir", image=image)
+with image.imports():
+    import os
+    from datetime import datetime
+
+flux_model_vol = modal.Volume.from_name("flux-model-vol", create_if_missing=True)  # Reference your volume
+
+@app.function(volumes={"/data": flux_model_vol},
+              secrets=[modal.Secret.from_name("huggingface-token")],
+              #gpu="a100-80gb"
+              )
+def test_dir():
+    
+    import os
+    import urllib.request
+
+
+    # Define the path of the new directory
+    new_directory = "LS_images"
+
+    # Create the directory (and parent directories if needed)
+    os.makedirs(f"/data/{new_directory}", exist_ok=True)
+
+    #url = "https://huggingface.co/city96/FLUX.1-dev-gguf/resolve/main/flux1-dev-Q8_0.gguf"
+    #urllib.request.urlretrieve(url, "/data/FLUX.1-dev-gguf/flux1-dev-Q8_0.gguf")
+
+    print("Download complete!")
+
+
+    #print(f"Directory created: {new_directory}")
+
+    # Get the current working directory
+    current_directory = os.getcwd()
+    # List the contents of the current directory
+    print("Contents of current modal directory:")
+    print(os.listdir(current_directory))
+
+    # VOLUME DIRECTORY
+    volume_directory = f"{current_directory}/data/"
+    print(f"Current volume directory: {volume_directory}")
+    print(os.listdir(volume_directory))
+
+
+    flux_model_vol.de
+
+
+
+
+

tools/download_models_modal_HF.py

@@ -0,0 +1,45 @@
+################# RUN IT W MODAL RUN TO DOWNLOAD ON MODAL VOLUME ###########
+
+
+import modal
+import os
+
+app = modal.App("flux-model-setup")
+
+# Persistent volume for storing models
+volume = modal.Volume.from_name("flux-model-vol", create_if_missing=True)  
+
+# Image with dependencies
+download_image = (
+    modal.Image.debian_slim()
+    .pip_install("huggingface_hub[hf_transfer]", "transformers", "aria2")  # aria2 for ultra-fast parallel downloads
+    .env({"HF_HUB_ENABLE_HF_TRANSFER": "1"})  # Enable fast Rust-based downloads
+)
+
+@app.function(
+    volumes={"/data": volume},
+    image=download_image,
+    secrets=[modal.Secret.from_name("huggingface-token")]
+)
+def download_flux():
+    from huggingface_hub import snapshot_download
+    import transformers
+
+    repo_id = "strangerzonehf/Flux-Midjourney-Mix2-LoRA"
+    local_dir = "/data/Flux-Midjourney-Mix2-LoRA/"
+
+    # **FASTEST METHOD:** Use max_workers for parallel download
+    snapshot_download(
+        repo_id,
+        local_dir=local_dir,
+        revision="main",
+        #ignore_patterns=["*.pt", "*.bin"],  # Skip large model weights
+        max_workers=8  # Higher concurrency for parallel chunk downloads
+    )
+
+    transformers.utils.move_cache()
+    print(f"Model downloaded to {local_dir}")
+
+@app.local_entrypoint()
+def main():
+    download_flux.remote()