Modified pet-generator to use ChatGPT 4o and Fal.ai

README.md

@@ -4,34 +4,33 @@ emoji: 🃏
 colorFrom: yellow
 colorTo: purple
 sdk: gradio
-sdk_version: 4.26.0
+sdk_version: 4.44.0
 app_file: app.py
 pinned: false
 ---
 
-# Drakosfire's Dungeons and Dragons Item Card Generator
+# Drakosfire's Cyber Pet Card Generator
 
-Welcome to the Drakosfire's Dungeons and Dragons Item Card Generator! This innovative tool harnesses the power of AI to generate unique item cards for your D&D adventures. 
+This is a tool built to generate custom cyborg pets for an unnanounced creative project.
 
 ## Overview
 
-This generator leverages an API call to [Replicate](https://replicate.com/) using Llama 3 70b, combined with a custom fine-tuned version of the Stable Diffusion SDXL model. You can find more about the specific modelthis project was based on at [Civitai](https://civitai.com/models/129681/sdxl-faetastic).
+This generator leverages an API call to FAL.ai to generate images with a custom LoRA trained off of my card dataset as well as OpenAI's GPT-4o to generate text and image prompts for the images.
 
 ## Key Features
 
-- **Custom Fine-Tuning**: The backbone of this generator is a fine-tuned Stable Diffusion SDXL model, specifically optimized for generating high-quality, fantasy-themed images on a unique card border that is themed by your imagination.
+- **Custom Fine-Tuning**: The backbone of this generator is a fine-tuned Flux.1 model, specifically optimized for generating high-quality, pet images on a unique card border that is flavored by your imagination.
 
-- **Consistent Card Design**: To ensure each card maintains a uniform appearance, a LoRA (Locally Optimized Representation Approximation) technique was used. This involved training the model with a hand-crafted dataset of card border images, enabling the system to generate new cards with consistent text and image spaces where the generated text and images can be elegantly integrated.
+- **Consistent Card Design**: To ensure each card maintains a uniform appearance, a LoRA (Low-Rank Adaptation) technique was used. This involved training the model with a hand-crafted dataset of card border images, enabling the system to generate new cards with consistent text and image spaces where the generated text and images can be elegantly integrated.
 
 ## How It Works
 
-1. Your intitial text along with the prompt is sent to Llama 3 70b to generate a python dictionary.
-2. This new text will populate in interactive text fields. If it isn't perfect you can edit the text to fit your item.
-3. The final text field is the Stable Diffusion prompt, these generate like one sentence stories describing the scene of your item. This field can also be edited.
-## The first image generation take about 2 minutes for model to 'cold boot' after that it's ~10s per image.
-4. **Image and Text Generation**: Now generate 4 card template without text and pick your favorite.
-5. Finally, add text to your favorite template.
-3. **Result**: The final product is a beautifully crafted D&D item card, ready for use in your gaming sessions.
+1. Your intitial text along with the prompt is sent to OpenAI's GPT-4o to generate a JSON.
+2. This JSON will be parsed into textboxes to review and edit. 
+3. The final text field is the image prompt, these generate like one sentence stories describing the scene of your pet.
+4. **Image and Text Generation**: Now generate 4 blank cards in about 15 seconds.
+5. Finally, click the add text button to add text to your favorite template.
+3. **Result**: The final product is a beautifully crafted collectible pet card.
 
 ## Example Cards
 
@@ -44,4 +43,4 @@ Below are a few examples of the item cards I've generated:
 ![Ridiculously Cute Kitten](https://cdn-lfs-us-1.huggingface.co/repos/cb/b4/cbb436e53a2021c9535cec40c6360a89a1d6f5e42bc18d88bbd2da335dde82bd/863ef9235c94e60fd7176db237e3939f67c6a22bce3015712ff2336a49065f83?response-content-disposition=inline%3B+filename*%3DUTF-8%27%27Ridiculouslycutekitten.png%3B+filename%3D%22Ridiculouslycutekitten.png%22%3B&response-content-type=image%2Fpng&Expires=1714445494&Policy=eyJTdGF0ZW1lbnQiOlt7IkNvbmRpdGlvbiI6eyJEYXRlTGVzc1RoYW4iOnsiQVdTOkVwb2NoVGltZSI6MTcxNDQ0NTQ5NH19LCJSZXNvdXJjZSI6Imh0dHBzOi8vY2RuLWxmcy11cy0xLmh1Z2dpbmdmYWNlLmNvL3JlcG9zL2NiL2I0L2NiYjQzNmU1M2EyMDIxYzk1MzVjZWM0MGM2MzYwYTg5YTFkNmY1ZTQyYmMxOGQ4OGJiZDJkYTMzNWRkZTgyYmQvODYzZWY5MjM1Yzk0ZTYwZmQ3MTc2ZGIyMzdlMzkzOWY2N2M2YTIyYmNlMzAxNTcxMmZmMjMzNmE0OTA2NWY4Mz9yZXNwb25zZS1jb250ZW50LWRpc3Bvc2l0aW9uPSomcmVzcG9uc2UtY29udGVudC10eXBlPSoifV19&Signature=Awm-rIF7be1FRdKIU2zR-cxZa3KtR7FcYmPgVTqfaa2oMKZhU2vkG3lLHXimFvVuXbiv8Rd4ZnijffjWyEBWs6FV5L2JFQ2jG%7EDhlBIdWj-6skbvYwgxsbAfDclBbBv3zyZs%7EDK0rzEqvIDdd8Avud9NtXvyMpSFZTx2zUzGMp1saHLaYH2uU91z7zWHLhnL0XVW1JvY6RbaNx0Ydtx42r1IWqVpLqpB6yGoeIU2PBQCpNiuY61zwjOCg2iMM3OuXJ56eULFXtnrbJbqDSZnlIibELCZ%7EJQCozeVSvi-OK6HOkD7W7tPSnbfx2f9GIY1FkmzoXh-K7cFNL-CxYnCjw__&Key-Pair-Id=KCD77M1F0VK2B)
 
 
-We hope you enjoy enhancing your Dungeons and Dragons experience with this unique tool. Happy adventuring!
+

app.py

@@ -13,28 +13,28 @@ with gr.Blocks() as demo:
     # Functions and State Variables
     # Build functions W/in the Gradio format, because it only allows modification within it's context
     # Define inputs to match what is called on click, and output of the function as a list that matches the list of outputs
-    textbox_default_dict = {'Name':'', \
-                            'Type': '',
-                            'Rarity':'',
-                            'Value':'',
-                            'Properties':'',
-                            'Damage':'',
-                            'Weight':'',
-                            'Description':'',
-                            'Quote':'',
-                            'SD Prompt':''
+    textbox_default_dict = {'Name':'', 
+                            'Pet Species': '',
+                            'Breed':'',
+                            'Fur':'',
+                            'Intelligence Level':'',
+                            'Affection Level':'',
+                            'Energy Level':'',
+                            'Noise Level':'',
+                            'Play Level':'',
+                            'image Prompt':''
                             }
-    
-    item_name_var = gr.State()
-    item_type_var = gr.State()
-    item_rarity_var = gr.State()
-    item_value_var = gr.State()
-    item_properties_var = gr.State()
-    item_damage_var = gr.State()
-    item_weight_var = gr.State()
-    item_description_var = gr.State()
-    item_quote_var = gr.State()
-    item_sd_prompt_var = gr.State('')
+    # Text states
+    pet_name_state = gr.State()
+    pet_species_state = gr.State()
+    pet_breed_state = gr.State()
+    pet_fur_state = gr.State()
+    pet_intelligence_level_state = gr.State()
+    pet_affection_level_state = gr.State()
+    pet_energy_level_state = gr.State()
+    pet_noise_level_state = gr.State()
+    pet_play_level_state = gr.State()
+    image_prompt_var = gr.State('')
 
     selected_border_image = gr.State('./card_templates/Moonstone Border.png')
     num_image_to_generate = gr.State(4)
@@ -53,37 +53,27 @@ with gr.Blocks() as demo:
     def generate_text_update_textboxes(user_input):
         
         llm_output=useri.call_llm(user_input)
-        item_key = list(llm_output.keys())
-        
-        item_key_values = list(llm_output[item_key[0]].keys())
-        item_name = llm_output[item_key[0]]['Name']
-        item_type = llm_output[item_key[0]]['Type']
-        item_rarity = llm_output[item_key[0]]['Rarity']            
-        item_value = llm_output[item_key[0]]['Value']
-        item_properties = llm_output[item_key[0]]['Properties']
-        
-        if 'Damage' in item_key_values: 
-            item_damage = llm_output[item_key[0]]['Damage']
-        else: item_damage = ''
-
-
-        item_weight = llm_output[item_key[0]]['Weight']
-        item_description = llm_output[item_key[0]]['Description']
-        item_quote = llm_output[item_key[0]]['Quote']
-        item_quote = llm_output[item_key[0]]['Quote']
-        sd_prompt = llm_output[item_key[0]]['SD Prompt']
-
-
-        return [item_name, item_name,
-                item_type, item_type,
-                item_rarity, item_rarity,
-                item_value, item_value,
-                item_properties, item_properties,
-                item_damage, item_damage,
-                item_weight, item_weight,
-                item_description, item_description,
-                item_quote, item_quote,
-                sd_prompt, sd_prompt]
+        pet_key = list(llm_output.keys())
+        pet_name = llm_output[pet_key[0]]['Name']
+        pet_species = llm_output[pet_key[0]]['Pet Species']
+        pet_breed = llm_output[pet_key[0]]['Breed']            
+        pet_fur = llm_output[pet_key[0]]['Fur']
+        pet_intelligence_level = llm_output[pet_key[0]]['Intelligence Level']
+        pet_affection_level = llm_output[pet_key[0]]['Affection Level']
+        pet_energy_level = llm_output[pet_key[0]]['Energy Level']
+        pet_noise_level = llm_output[pet_key[0]]['Noise Level']
+        pet_play_level = llm_output[pet_key[0]]['Play Level']
+        image_prompt = llm_output[pet_key[0]]['Image Prompt']
+        return [pet_name, pet_name,
+                    pet_species, pet_species,
+                    pet_breed, pet_breed,
+                    pet_fur, pet_fur,
+                    pet_intelligence_level, pet_intelligence_level,
+                    pet_affection_level, pet_affection_level,
+                    pet_energy_level, pet_energy_level,
+                    pet_noise_level, pet_noise_level,
+                    pet_play_level, pet_play_level,
+                    image_prompt, image_prompt]
     
     # Called on user selecting an image from the gallery, outputs the path of the image
     def assign_img_path(evt: gr.SelectData):          
@@ -102,18 +92,9 @@ with gr.Blocks() as demo:
     # Called when pressing button to generate image, updates gallery by returning the list of image URLs
     def generate_image_update_gallery(num_img, sd_prompt,item_name, built_template):
         delete_temp_images()
+        print(f"num_img is a {type(num_img)}")
         print(f"sd_prompt is a {type(sd_prompt)}")
-        image_list = []
-        
-        
-        for x in range(num_img):
-            preview = img2img.preview_and_generate_image(x,sd_prompt, built_template, item_name)
-            image_list.append(preview)
-            yield image_list
-            
-        del preview
-       
-        #generated_image_list = img2img.generate_image(num_img,sd_prompt,item_name,selected_border)
+        image_list = img2img.preview_and_generate_image(4,sd_prompt, built_template)
         return image_list
     
     def build_template(selected_border, selected_seed_image):
@@ -201,10 +182,10 @@ with gr.Blocks() as demo:
     build_card_template_button.click(build_template, inputs = [selected_border_image, selected_seed_image], outputs = [built_template_gallery, built_template]) 
         
     gr.HTML(""" <div id="inner"> <header>
-                    <h2><b>Second:</b> Generate Item Text </h2>
+                    <h2><b>Second:</b> Generate Pet Text </h2>
                    </div>""")  
     gr.HTML(""" <div id="inner"> <header>
-                        <h3>1. Use a few words to describe the item then click 'Generate Text' </h3>
+                        <h3>1. Use a name, a breed, some coloring and the animal type, Ex: 'Hermione the Stormy Grey English Shorthair Cat' then click 'Generate Text' </h3>
                         </div>""")
     with gr.Row():
         user_input =  gr.Textbox(label = 'Item', lines =1, placeholder= "Flaming Magical Sword", elem_id= "Item", scale =4)
@@ -214,30 +195,43 @@ with gr.Blocks() as demo:
                 <h3> 2. Review and Edit the text</h3>
                 </div>""") 
     with gr.Row():
-
-    # Build text boxes for the broken up item dictionary values         
-
     # Build text boxes for the broken up item dictionary values         
         with gr.Column(scale = 1):
-            item_name_output = gr.Textbox(value = set_textbox_defaults(textbox_default_dict, 'Name'),label = 'Name', lines = 1, interactive=True, elem_id='Item Name')
-            item_type_output = gr.Textbox(value = set_textbox_defaults(textbox_default_dict, 'Type'),label = 'Type', lines = 1, interactive=True, elem_id='Item Type')
-            item_rarity_output = gr.Textbox(value = set_textbox_defaults(textbox_default_dict, 'Rarity'),label = 'Rarity : [Common, Uncommon, Rare, Very Rare, Legendary]', lines = 1, interactive=True, elem_id='Item Rarity')
-            item_value_output = gr.Textbox(value = set_textbox_defaults(textbox_default_dict, 'Value'),label = 'Value', lines = 1, interactive=True, elem_id='Item Value')
+            pet_name_output = gr.Textbox(value = set_textbox_defaults(
+                textbox_default_dict, 'Name'),label = 'Name', lines = 1, interactive=True, elem_id='Pet Name')
+            
+            pet_species_output = gr.Textbox(value = set_textbox_defaults(
+                textbox_default_dict, 'Pet Species'),label = 'Type', lines = 1, interactive=True, elem_id='Pet Species')
+            
+            pet_breed_output = gr.Textbox(value = set_textbox_defaults(
+                textbox_default_dict, 'Breed'),label = 'Breed : [Common, Uncommon, Rare, Very Rare, Legendary]',
+                lines = 1, interactive=True, elem_id='Pet Breed')
+            
+            pet_fur_output = gr.Textbox(value = set_textbox_defaults(
+                textbox_default_dict, 'Fur'),label = 'Fur Quality', lines = 1, interactive=True, elem_id='Pet Fur')
                         
-                # Pass the user input and border template to the generator
+    # Pass the user input and border template to the generator
         with gr.Column(scale = 1):    
-            item_damage_output = gr.Textbox(value = set_textbox_defaults(textbox_default_dict, 'Damage'),label = 'Damage', lines = 1, interactive=True, elem_id='Item Damage')
-            item_weight_output = gr.Textbox(value = set_textbox_defaults(textbox_default_dict, 'Weight'),label = 'Weight', lines = 1, interactive=True, elem_id='Item Weight')
-            item_description_output = gr.Textbox(value = set_textbox_defaults(textbox_default_dict, 'Description'),label = 'Description', lines = 1, interactive=True, elem_id='Item Description')
-            item_quote_output = gr.Textbox(value = set_textbox_defaults(textbox_default_dict, 'Quote'),label = 'Quote', lines = 1, interactive=True, elem_id='Item quote')
-    item_properties_output = gr.Textbox(value = set_textbox_defaults(textbox_default_dict, 'Properties'),label = 'Properties : [List of comma seperated values]', lines = 1, interactive=True, elem_id='Item Properties')
-    
+            pet_affection_level_output = gr.Textbox(value = set_textbox_defaults(
+                textbox_default_dict, 'Affection Level'),label = 'Affection Level', lines = 1, interactive=True, elem_id='Affection Level')
+            
+            pet_energy_level_output = gr.Textbox(value = set_textbox_defaults(
+                textbox_default_dict, 'Energy Level'),label = 'Energy Level', lines = 1, interactive=True, elem_id='Energy Level')
+            
+            pet_noise_level_output = gr.Textbox(value = set_textbox_defaults(
+                textbox_default_dict, 'Noise Level'),label = 'Noise Level', lines = 1, interactive=True, elem_id='Noise Level')
+            
+            pet_play_level_output = gr.Textbox(value = set_textbox_defaults(
+                textbox_default_dict, 'Play Level'),label = 'Play Level', lines = 1, interactive=True, elem_id='Play Level')
+            
+            pet_intelligence_level_output = gr.Textbox(value = set_textbox_defaults(
+                textbox_default_dict, 'Intelligence Level'),label = 'Intelligence Level',
+                lines = 1, interactive=True, elem_id='Intelligence Level')
     
     gr.HTML(""" <div id="inner"> <header>
                 <h3> 3. This text will be used to generate the card's image.</h3>
                 </div>""") 
-    item_sd_prompt_output = gr.Textbox(label = 'Putting words or phrases in parenthesis adds weight. Example: (Flaming Magical :1.0) Sword.', value = set_textbox_defaults(textbox_default_dict, 'SD Prompt'), lines = 1, interactive=True, elem_id='SD Prompt')
-         
+    image_prompt_output = gr.Textbox(label = 'Putting words or phrases in parenthesis adds weight. Example: (Flaming Magical :1.0) Sword.', value = set_textbox_defaults(textbox_default_dict, 'image Prompt'), lines = 1, interactive=True, elem_id='image Prompt')         
     gr.HTML(""" <div id="inner"> <header>
                 <h2> <b>Third:</b> Click 'Generate Cards' to generate 4 cards to choose from. \n First image from a cold boot takes about 2 minutes. \n After that it's 10 seconds per image. </h2>
                 </div>""")          
@@ -261,51 +255,46 @@ with gr.Blocks() as demo:
                                     elem_id = "Generated Cards Gallery",
                                     allow_preview=True
                                     )
-        
-    
     
-    card_gen_button.click(fn = generate_image_update_gallery,
-                           inputs =[num_image_to_generate,item_sd_prompt_output,item_name_output,
-                                    built_template_gallery], outputs= generate_gallery,
-                                    show_progress=True)
+    card_gen_button.click(fn = generate_image_update_gallery, inputs =[num_image_to_generate,image_prompt_output,pet_name_output,built_template], outputs= generate_gallery)
     generate_gallery.select(assign_img_path, outputs = selected_generated_image)
 
         # Button logice calls function when button object is pressed, passing inputs and passing output to components
     llm_output = item_text_generate.click(generate_text_update_textboxes, 
                                                 inputs = [user_input], 
-                                                outputs= [item_name_var, 
-                                                        item_name_output,
-                                                        item_type_var,
-                                                        item_type_output, 
-                                                        item_rarity_var,
-                                                        item_rarity_output,
-                                                        item_value_var,
-                                                        item_value_output,
-                                                        item_properties_var,
-                                                        item_properties_output,
-                                                        item_damage_var,
-                                                        item_damage_output,
-                                                        item_weight_var,
-                                                        item_weight_output,
-                                                        item_description_var,
-                                                        item_description_output,
-                                                        item_quote_var,
-                                                        item_quote_output,                                                     
-                                                        item_sd_prompt_var,
-                                                        item_sd_prompt_output])          
+                                                outputs= [pet_name_state, 
+                                                        pet_name_output,
+                                                        pet_species_state,
+                                                        pet_species_output, 
+                                                        pet_breed_state,
+                                                        pet_breed_output,
+                                                        pet_fur_state,
+                                                        pet_fur_output,
+                                                        pet_intelligence_level_state,
+                                                        pet_intelligence_level_output,
+                                                        pet_affection_level_state,
+                                                        pet_affection_level_output,
+                                                        pet_energy_level_state,
+                                                        pet_energy_level_output,
+                                                        pet_noise_level_state,
+                                                        pet_noise_level_output,
+                                                        pet_play_level_state,
+                                                        pet_play_level_output,                                                     
+                                                        image_prompt_var,
+                                                        image_prompt_output])           
         
         
     generate_final_item_card.click(card.render_text_on_card, inputs = [selected_generated_image,
-                                                                        item_name_output, 
-                                                                        item_type_output, 
-                                                                        item_rarity_output, 
-                                                                        item_value_output,
-                                                                        item_properties_output, 
-                                                                        item_damage_output,
-                                                                        item_weight_output,
-                                                                        item_description_output,
-                                                                        item_quote_output
-                                                                        ], 
+                                                                        pet_name_output, 
+                                                                        pet_species_output, 
+                                                                        pet_breed_output, 
+                                                                        pet_fur_output,
+                                                                        pet_intelligence_level_output, 
+                                                                        pet_affection_level_output,
+                                                                        pet_energy_level_output,
+                                                                        pet_noise_level_output,
+                                                                        pet_play_level_output
+                                                                        ],
                                                                         outputs = generate_gallery )
     base_dir = os.path.dirname(os.path.abspath(__file__))  # Gets the directory where the script is located
     print(f"Base Directory :",base_dir)

card_generator.py

@@ -3,6 +3,7 @@ from PIL import Image, ImageFilter
 import utilities as u
 import ast
 from urllib.request import urlopen
+import urllib.parse
 
 def save_image(image,item_key):
     image.save(f"{item_key['Name']}.png")
@@ -12,15 +13,15 @@ def save_image(image,item_key):
 #shop_inventory = inv.inventory
 #purchased_item_key = shop_inventory['Shortsword']
 #border_path = './card_templates/Shining Sunset Border.png'
-base_path = "https://media.githubusercontent.com/media/Drakosfire/CardGenerator/main/card_parts/"
-value_overlay_path = f"{base_path}Value_box_transparent.png"
+base_path = "https://media.githubusercontent.com/media/Drakosfire/CardGenerator/refs/heads/pet-generator/card_parts/"
+# value_overlay_path = f"{base_path}Value_box_transparent.png"
 test_item = {'Name': 'Pustulent Raspberry', 'Type': 'Fruit', 'Value': '1 cp', 'Properties': ['Unusual Appearance', 'Rare Taste'], 'Weight': '0.2 lb', 'Description': 'This small fruit has a pustulent appearance, with bumps and irregular shapes covering its surface. Its vibrant colors and strange texture make it an oddity among other fruits.', 'Quote': 'A fruit that defies expectations, as sweet and sour as life itself.', 'SD Prompt': 'A small fruit with vibrant colors and irregular shapes, bumps covering its surface.'}
-sticker_path_dictionary = {'Default': f"{base_path}Sizzek Sticker.png",
-                            'Common': f"{base_path}Common.png",
-                            'Uncommon': f"{base_path}Uncommon.png",
-                            'Rare': f"{base_path}Rare.png",
-                            'Very Rare':f"{base_path}Very Rare.png",
-                            'Legendary':f"{base_path}Legendary.png"}
+sticker_path_dictionary = {'Default': urllib.parse.urljoin(base_path, urllib.parse.quote("Sizzek Sticker.png")),
+                            'Common': urllib.parse.urljoin(base_path, urllib.parse.quote("Common.png")),
+                            'Uncommon': urllib.parse.urljoin(base_path, urllib.parse.quote("Uncommon.png")),
+                            'Rare': urllib.parse.urljoin(base_path, urllib.parse.quote("Rare.png")),
+                            'Very Rare':urllib.parse.urljoin(base_path, urllib.parse.quote("Very Rare.png")),
+                            'Legendary':urllib.parse.urljoin(base_path, urllib.parse.quote("Legendary.png"))}
 
 
 # Function that takes in an image url and a dictionary and uses the values to print onto a card.
@@ -49,20 +50,19 @@ def paste_image_and_resize(base_image,sticker_path, x_position, y_position,img_w
     # Paste the resized image onto the base image
     base_image.paste(image_to_paste_resized, paste_position, image_to_paste_resized)
 
-def render_text_on_card(image_path, item_name,
-                                    item_type,
-                                    item_rarity,
-                                    item_value,
-                                    item_properties,
-                                    item_damage,
-                                    item_weight,
-                                    item_description,
-                                    item_quote) : 
+def render_text_on_card(image_path, pet_name,
+                                    pet_species,
+                                    pet_breed,
+                                    pet_fur,
+                                    pet_intelligence,
+                                    pet_affection,
+                                    pet_energy,
+                                    pet_noise,
+                                    pet_play) : 
     # Card Properties 
     image_list = []
-    item_properties = ast.literal_eval(item_properties)
-    item_properties = '\n'.join(item_properties)
-    output_image_path = f"./{item_name}.png"
+    pet_properties = f"Intelligence : {pet_intelligence} \n Affection : {pet_affection} \n Energy : {pet_energy} \n Noise : {pet_noise} \n Play : {pet_play}"
+    output_image_path = f"./{pet_name}.png"
     print(f"Saving image to {output_image_path}")
     font_path = "./fonts/Balgruf.ttf"
     italics_font_path = './fonts/BalgrufItalic.ttf'
@@ -77,12 +77,7 @@ def render_text_on_card(image_path, item_name,
     type_center_position = (384, 545)
     type_area_width = 600 
     type_area_height = 45 
-    type_text = item_type
-    if len(item_weight) >= 1: 
-        type_text = type_text + ' '+ item_weight
-
-    if len(item_damage) >= 1 : 
-        type_text = type_text + ' '+ item_damage
+    type_text = f"{pet_breed} Color and Fur : {pet_fur}"    
 
     # Description box properties
     description_position = (105, 630)
@@ -101,24 +96,22 @@ def render_text_on_card(image_path, item_name,
 
     # open image and render text
     image = u.open_image_from_url(image_path)
-    image = rend.render_text_with_dynamic_spacing(image, item_name, title_center_position, title_area_width, title_area_height,font_path,initial_font_size)
+    image = rend.render_text_with_dynamic_spacing(image, pet_name, title_center_position, title_area_width, title_area_height,font_path,initial_font_size)
     image = rend.render_text_with_dynamic_spacing(image,type_text , type_center_position, type_area_width, type_area_height,font_path,initial_font_size)
-    image = rend.render_text_with_dynamic_spacing(image, item_description + '\n\n' + item_properties, description_position, description_area_width, description_area_height,font_path,initial_font_size, description = True)
-    #Paste value overlay
-    paste_image_and_resize(image, value_overlay_path,x_position= 0,y_position=0, img_width= 768, img_height= 1024)
-    image = rend.render_text_with_dynamic_spacing(image, item_value, value_position, value_area_width, value_area_height,font_path,initial_font_size)
-    image = rend.render_text_with_dynamic_spacing(image, item_quote, quote_position, quote_area_width, quote_area_height,italics_font_path,initial_font_size, quote = True)
+    image = rend.render_text_with_dynamic_spacing(image, pet_properties, description_position, description_area_width, description_area_height,font_path,initial_font_size, description = True)
+    # paste_image_and_resize(image, value_overlay_path,x_position= 0,y_position=0, img_width= 768, img_height= 1024)
     #Paste Sizzek Sticker
-    paste_image_and_resize(image, sticker_path_dictionary,x_position= 0,y_position=909, img_width= 115, img_height= 115, purchased_item_key= item_rarity)
+    paste_image_and_resize(image, sticker_path_dictionary['Default'],x_position= 0,y_position=909, img_width= 115, img_height= 115, purchased_item_key= None)
 
     # Add blur, gives it a less artificial look, put into list and return the list since gallery requires lists
     image = image.filter(ImageFilter.GaussianBlur(.5))
     image_list.append(image)
    
-    return image_list
+    image = image.save(f"./output/{pet_name}.png")
+    
     
 
-
+    return image_list
 
 
 

img2img.py

@@ -2,40 +2,98 @@ import time
 import utilities as u
 import card_generator as card
 from PIL import Image
-import replicate
+import fal_client
 from pathlib import Path
+import tempfile
+import os
+import base64
+import io
+import logging
+import requests
+import json
 
+# Set up logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 start_time = time.time()
 temp_image_path = "./image_temp/"
 
 
 
-def preview_and_generate_image(num_images,sd_prompt, user_input_template, item):    
-    # Replicate is expecting a path
-    card_template = Path(user_input_template[0][0])
-    print(card_template)
-    img_start = time.time()   
-    output=replicate.run(
-            "drakosfire/card-generator-v1:4e46bb44e6444d104b58e34a6d2ab66833aa84ba0dfe84ae57ef63d8e15da467",
-            input={
-            "item":item,
-            "sd_prompt":sd_prompt,
-            "num_images":num_images,
-            "input_template":card_template
+def preview_and_generate_image(num_images, sd_prompt, user_input_template):
+    print(f"num_images: {num_images}")
+    print(f"sd_prompt: {sd_prompt}")
+    print(f"user_input_template: {user_input_template}")
+    num_images = int(4)
+    sd_prompt = f"magnum opus, blank card, no text, blank textboxes, detailed high quality animal properties borders, {sd_prompt}"
+    try:
+        # Save the image to a temporary file
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as temp_file:
+            user_input_template[0].save(temp_file.name, format="PNG")
+            temp_path = temp_file.name
 
-        }
-    )
-    
-    
-    img_time = time.time() - img_start
-    img_its = 35/img_time
-    print(f"image gen time = {img_time} and {img_its} it/s")
-    
-    total_time = time.time() - start_time
-    print(total_time)
+        logger.info(f"Image saved to temporary file: {temp_path}")
 
-    return output
+        # Upload the file using fal_client
+        url = fal_client.upload_file(temp_path)
+        logger.info(f"Image uploaded. URL: {url}")
+
+        # Remove the temporary file
+        os.unlink(temp_path)
+
+        request_handle = fal_client.submit(
+            "fal-ai/flux-lora/image-to-image",
+            arguments={
+                "num_inference_steps": 35,
+                "prompt": sd_prompt,
+                "num_images": num_images,
+                "image_url": url,
+                "strength": 0.85
+            }
+        )
+        
+        logger.info(f"Type of request_handle: {type(request_handle)}")
+        logger.info(f"Content of request_handle: {request_handle}")
+        
+        # Get the result from the SyncRequestHandle
+        result = request_handle.get()
+        
+        logger.info(f"Type of result: {type(result)}")
+        logger.info(f"Content of result: {json.dumps(result, indent=2)}")
+        
+        # Extract the image URLs from the result
+        image_urls = [img['url'] for img in result.get('images', [])]
+        
+        logger.info(f"Extracted image URLs: {image_urls}")
+        
+        if not image_urls:
+            logger.warning("No images were generated.")
+            return []
+        
+        # Download and process the images
+        images = []
+        for i, url in enumerate(image_urls):
+            try:
+                response = requests.get(url)
+                response.raise_for_status()  # Raises an HTTPError for bad responses
+                img = Image.open(io.BytesIO(response.content))
+                images.append((img, f"Generated Image {i+1}"))  # Add a tuple with image and caption
+                logger.info(f"Successfully downloaded and processed image {i+1}")
+            except Exception as e:
+                logger.error(f"Error processing image {i+1} from URL {url}: {str(e)}")
+        
+        if not images:
+            logger.warning("No images could be downloaded and processed.")
+            return []
+        
+        logger.info(f"Returning {len(images)} processed images")
+        return images
+    
+    except Exception as e:
+        logger.error(f"Error during API call or processing: {str(e)}")
+        logger.exception("Full traceback:")
+        return []
 
 
 

item_dict_gen.py

@@ -1,359 +1,283 @@
-import replicate
 import ast
 import gc
-import os
-
-api_key = os.getenv('REPLICATE_API_TOKEN')
-
-
-model_path = "meta/meta-llama-3-70b-instruct"
-def load_llm(user_input):
-  input = {"prompt" : f" {prompt_instructions} the item is {user_input}"}
-  output = replicate.run(model_path,
-    input=input
-    )
-  return output
-  
-
-def call_llm_and_cleanup(user_input):
-    # Call the LLM and store its output
-    llm_output = "".join(load_llm(user_input))
-    print("".join(llm_output))
-    gc.collect()
-  
-    # llm_output is still available for use here
+from openai import OpenAI
+
+client = OpenAI()
+
+def load_llm(user_input, prompt_instructions):
+    prompt = f"{user_input}"
+    print(prompt)
+    response = client.chat.completions.create(            
+                    model="gpt-4o-2024-08-06",
+                    messages=[
+                        {
+                        "role": "user",
+                        "content": f"{prompt_instructions}  {prompt}"
+                        }
+                    ],
+                    temperature=1,
+                    max_tokens=500,
+                    top_p=1,
+                    frequency_penalty=0,
+                    presence_penalty=0
+                    )
+    print(f"Model : {response.model}")
+    return response.choices[0].message.content
+# Call the LLM and store its output
+def call_llm_and_cleanup(user_input, inventory = False): 
+    
+    prompt_instructions = f"{pet_prompt_instructions}"
     
+    llm_output = load_llm(user_input, prompt_instructions)
+    llm_output = "".join(llm_output)
+    print(f"llm_output = {llm_output}")
+    # llm_output is still available for use here
     return llm_output
 
-def convert_to_dict(string):
-  # Evaluate if string is dictionary literal
-    try: 
-        result = ast.literal_eval(string)
-        if isinstance(result, dict):
-            print("Item dictionary is valid")
-            return result
-        # If not, modify by attempting to add brackets to where they tend to fail to generate.
-        else: 
-            modified_string = '{' + string
-            if isinstance(modified_string, dict):  
-                return modified_string
-            modified_string = string + '}'
-            if isinstance(modified_string, dict):  
-                return modified_string
-            modified_string = '{' + string + '}'
-            if isinstance(modified_string, dict):  
-                return modified_string
-    except (ValueError, SyntaxError) :
-        print("Dictionary not valid")
-        return None
-  
-
-# Instructions past 4 are not time tested and may need to be removed.
-### Meta prompted : 
-prompt_instructions = """ **Purpose**: ONLY Generate a structured inventory entry for a specific item as a hashmap. Do NOT reply with anything other than a hashmap.
+pet_prompt_instructions = """ **Purpose**: ONLY Generate a structured json following the provided format. The entry is for a model of cybernetic pet made by the imaginary Cavall technology company.
 
 **Instructions**:
-1. Replace `{item}` with the name of the user item, DO NOT CHANGE THE USER ITEM NAME enclosed in single quotes (e.g., `'Magic Wand'`).
-2. **Dictionary Structure**:
-    
-    {"{item}": {
-    'Name': "{item name}",
-    'Type': '{item type}',
-    'Rarity': '{item rarity},
-    'Value': '{item value}',
-    'Properties': ["{property1}", "{property2}", ...],
-    'Damage': '{damage formula} , '{damage type}',
-    'Weight': '{weight}',
-    'Description': "{item description}",
-    'Quote': "{item quote}",
-    'SD Prompt': "{special description for the item}"
-    } } 
-3. Weapons MUST have a key 'Damage' 
-4. The description should be brief and puncy, or concise and thoughtful.
-5. The quote and SD Prompt MUST be inside double quotations ie " ".
-6. The quote is from the perspective of someone commenting on the impact of the {item} on their life
-7. Value should be assigned as an integer of copper pieces (cp), silver pieces (sp), electrum pieces (ep), gold pieces (gp), and platinum pieces (pp). 
-8. Use this table for reference on value : 
-1 cp 	1 lb. of wheat
-2 cp 	1 lb. of flour or one chicken
-5 cp 	1 lb. of salt
-1 sp 	1 lb. of iron or 1 sq. yd. of canvas
-5 sp 	1 lb. of copper or 1 sq. yd. of cotton cloth
-1 gp 	1 lb. of ginger or one goat
-2 gp 	1 lb. of cinnamon or pepper, or one sheep
-3 gp 	1 lb. of cloves or one pig
-5 gp 	1 lb. of silver or 1 sq. yd. of linen
-10 gp 	1 sq. yd. of silk or one cow
-15 gp 	1 lb. of saffron or one ox
-50 gp 	1 lb. of gold
-500 gp 	1 lb. of platinum
-
-
-300 gp +1 Melee or Ranged Weapon 
-2000 gp +2 Melee or Ranged Weapon 
-10000 gp  +3 Melee or Ranged Weapon 
-300 gp +1 Armor Uncommon
-2000 gp +2 Armor Rare
-10000 gp +3 Armor Very Rare
-300 gp +1 Shield Uncommon
-2000 gp +2 Shield Rare
-10000 gp +3 Shield Very Rare
-
-9. Examples of Magical Scroll Value:
-    Common: 50-100 gp
-    Uncommon: 101-500 gp
-    Rare: 501-5000 gp
-    Very rare: 5001-50000 gp
-    Legendary: 50001+ gp
-
-A scroll's rarity depends on the spell's level:
-    Cantrip-1: Common
-    2-3: Uncommon
-    4-5: Rare
-    6-8: Very rare
-    9: Legendary
-
-10. Explanation of Mimics:
-Mimics are shapeshifting predators able to take on the form of inanimate objects to lure creatures to their doom. In dungeons, these cunning creatures most often take the form of doors and chests, having learned that such forms attract a steady stream of prey.
-Imitative Predators. Mimics can alter their outward texture to resemble wood, stone, and other basic materials, and they have evolved to assume the appearance of objects that other creatures are likely to come into contact with. A mimic in its altered form is nearly unrecognizable until potential prey blunders into its reach, whereupon the monster sprouts pseudopods and attacks.
-When it changes shape, a mimic excretes an adhesive that helps it seize prey and weapons that touch it. The adhesive is absorbed when the mimic assumes its amorphous form and on parts the mimic uses to move itself.
-Cunning Hunters. Mimics live and hunt alone, though they occasionally share their feeding grounds with other creatures. Although most mimics have only predatory intelligence, a rare few evolve greater cunning and the ability to carry on simple conversations in Common or Undercommon. Such mimics might allow safe passage through their domains or provide useful information in exchange for food.
 
-    
-- **Input Placeholder**:
-    - "{item}": Replace with the item name, ensuring it's wrapped in single quotes.
+1. Only output structured data. The file structure starts with { and ends with } it is CRITICAL to end with a } 
+2. Do not enclose the output in any other characters such as ''' or json    
+3. DO NOT use null, use "". 
+4. All keys and values MUST be enclosed in double quotes.
+5. Replace `{pet}` with the name of the user pet, DO NOT CHANGE THE USER pet NAME enclosed in single quotes (e.g., `'Magic Wand'`).
+6. Ensure your request is formatted as a hashmap. 
+7. The cybernetic pets are friendly and cute, they are products and toys. 
+8. The descriptions should be brief and puncy, or concise and in the voice of a judge at a high end cat or dog or pet exhibition such as the American Kennel Clube or Cat Fancier and make specific callout of how mild, and well camoflaughed the artificiality of the cybernetic pet is.
+9. The Image Prompt MUST be inside double quotations ie " ".
 
-**Output Examples**:
-1. Cloak of Whispering Shadows Entry:
-    
-    {"Cloak of Whispering Shadows": {
-    'Name': 'Cloak of Whispering Shadows',
-    'Type': 'Cloak',
-    'Rarity': 'Very Rare', 
-    'Value': '7500 gp',
-    'Properties': ["Grants invisibility in dim light or darkness","Allows communication with shadows for gathering information"],
-    'Weight': '1 lb',
-    'Description': "A cloak woven from the essence of twilight, blending its wearer into the shadows. Whispers of the past and present linger in its folds, offering secrets to those who listen.",
-    'Quote': "In the embrace of night, secrets surface in the silent whispers of the dark.",
-    'SD Prompt': " Cloak of deep indigo almost black, swirling patterns that shift and move with every step. As it drapes over one's shoulders, an eerie connection forms between the wearer and darkness itself." 
-    } }   
-    
-2. Health Potion Entry:
-    
-    {"Health Potion": {
-    'Name' : "Health Portion",
-    'Type' : 'Potion',
-    'Rarity' : 'Common',
-    'Value': '50 gp',
-    'Properties': ["Quafable", "Restores 1d4 + 2 HP upon consumption"],
-    'Weight': '0.5 lb',
-    'Description': "Contained within this small vial is a crimson liquid that sparkles when shaken, a life-saving elixir for those who brave the unknown.",
-    'Quote': "To the weary, a drop of hope; to the fallen, a chance to stand once more.",
-    'SD Prompt' : " a small, delicate vial containing a sparkling crimson liquid. Emit a soft glow, suggesting its restorative properties. The vial is set against a dark, ambiguous background." 
-    } }     
-    
-3. Wooden Shield Entry:
-    
-    {"Wooden Shield": {
-    'Name' : "Wooden Shield",
-    'Type' : 'Armor, Shield',
-    'Rarity': 'Common',
-    'Value': '10 gp',
-    'Properties': ["+2 AC"],
-    'Weight': '6 lb',
-    'Description': "Sturdy and reliable, this wooden shield is a simple yet effective defense against the blows of adversaries.",
-    'Quote': "In the rhythm of battle, it dances - a barrier between life and defeat.",
-    'SD Prompt': " a sturdy wooden shield, a symbol of defense, with a simple yet solid design. The shield, has visible grain patterns and a few battle scars. It stands as a steadfast protector, embodying the essence of a warrior's resilience in the face of adversity." 
-    } }
-     
-4.  Helmet of Perception Entry:
-    
-    {"Helmet of Perception": {
-    'Name' : "Helmet of Perception",
-    'Type' : 'Magical Item (armor, helmet)',
-    'Rarity': 'Very Rare', 
-    'Value': '3000 gp',
-    'Properties': ["+ 1 to AC", "Grants the wearer advantage on perception checks", "+5 to passive perception"],
-    'Weight': '3 lb',
-    'Description': "Forged from mystic metals and enchanted with ancient spells, this helmet offers protection beyond the physical realm.",
-    'Quote': "A crown not of royalty, but of unyielding vigilance, warding off the unseen threats that lurk in the shadows.",
-    'SD Prompt': " a mystical helmet crafted from enchanted metals, glowing with subtle runes.  imbued with spells, radiates a mystical aura, symbolizing enhanced perception and vigilance,elegant,formidable" 
-    } }
+ 
+**Format Example**:
+- **Dictionary Structure**:
     
-5. Longbow Entry:
+    {"{Pet}": {
+        "Name": '{Pet Name}',
+        "Pet Species": '{pet species}',
+        "Breed": '{pet breed},
+        "Fur" : "The colors and quality of the pet's fur,
+        "Intelligence Level" : "How intelligent the pet is and how it demonstrates it",
+        "Affection Level" : "How affectionate and how the pet shows it,
+        "Energy Level" : "The Energy level",
+        "Noise Level" : "The volume, frequency, triggers, and type",
+        "Play Level" : " The level of play and types enjoyed",
+        "Image Prompt" : "Descriptive imagery for an image generator"
+        }
     
-    {"Longbow": {
-    'Name': "Longbow",
-    'Type': 'Ranged Weapon (martial, longbow)',
-    'Rarity': 'Common',
-    'Value': '50 gp',
-    'Properties': ["2-handed", "Range 150/600", "Loading"],
-    'Damage': '1d8 + Dex, piercing',
-    'Weight': '6 lb',
-    'Description': "With a sleek and elegant design, this longbow is crafted for speed and precision, capable of striking down foes from a distance.",
-    'Quote': "From the shadows it emerges, a silent whisper of steel that pierces the veil of darkness, bringing justice to those who dare to trespass.",
-    'SD Prompt' : "a longbow with intricate carvings and stone inlay with a black string" 
-    } }
+- **Input Placeholder**:
+    - "{pet}": Replace with the pet name, ensuring it's wrapped in single quotes.
     
 
-6. Mace Entry:
-    
-    {"Mace": {
-    'Name': "Mace",
-    'Type': 'Melee Weapon (martial, bludgeoning)',
-    'Rarity': 'Common',
-    'Value': '25 gp',
-    'Properties': ["Bludgeoning", "One-handed"],
-    'Damage': '1d6 + str, bludgeoning',
-    'Weight': '6 lb', 
-    'Description': "This mace is a fearsome sight, its head a heavy and menacing ball of metal designed to crush bone and break spirits.", 
-    'Quote': "With each swing, it sings a melody of pain and retribution, an anthem of justice to those who wield it.", 
-    'SD Prompt': "a menacing  metal spike ball mace, designed for bludgeoning, with a heavy, intimidating head, embodying a tool for bone-crushing and spirit-breaking." 
-    } }
-    
-7. Flying Carpet Entry:
-    
-    {"Flying Carpet": {
-    'Name': "Flying Carpet", 
-    'Type': 'Magical Item (transportation)', 
-    'Rarity': 'Very Rare',
-    'Value': '3000 gp', 
-    'Properties': ["Flying", "Personal Flight", "Up to 2 passengers", Speed : 60 ft], 
-    'Weight': '50 lb', 
-    'Description': "This enchanted carpet whisks its riders through the skies, providing a swift and comfortable mode of transport across great distances.",
-    'Quote': "Soar above the mundane, and embrace the winds of adventure with this magical gift from the heavens.", 
-    'SD Prompt': "a vibrant, intricately patterned flying carpet soaring high in the sky, with clouds and a clear blue backdrop, emphasizing its magical essence and freedom of flight" 
-    } }
-    
-8. Tome of Endless Stories Entry:
-    
-    {"Tome of Endless Stories": {
-    'Name': "Tome of Endless Stories",
-    'Type': 'Book',
-    'Rarity': 'Uncommon'
-    'Value': '500 gp',
-    'Properties': [
-        "Generates a new story or piece of lore each day",
-        "Reading a story grants insight or a hint towards solving a problem or puzzle"
-    ],
-    'Weight': '3 lbs',
-    'Description': "An ancient tome bound in leather that shifts colors like the sunset. Its pages are never-ending, filled with tales from worlds both known and undiscovered.",
-    'Quote': "Within its pages lie the keys to a thousand worlds, each story a doorway to infinite possibilities.",
-    'SD Prompt': "leather-bound with gold and silver inlay, pages appear aged but are incredibly durable, magical glyphs shimmer softly on the cover." 
-    } }    
-    
-9. Ring of Miniature Summoning Entry:
-    
-    {"Ring of Miniature Summoning": {
-    'Name': "Ring of Miniature Summoning",
-    'Type': 'Ring',
-    'Rarity': 'Rare',
-    'Value': '1500 gp',
-    'Properties': ["Summons a miniature beast ally once per day", "Beast follows commands and lasts for 1 hour", "Choice of beast changes with each dawn"],
-    'Weight': '0 lb',
-    'Description': "A delicate ring with a gem that shifts colors. When activated, it brings forth a small, loyal beast companion from the ether.",
-    'Quote': "Not all companions walk beside us. Some are summoned from the depths of magic, small in size but vast in heart.",
-    'SD Prompt': "gemstone with changing colors, essence of companionship and versatility." 
-    } } 
-     
-
-10. Spoon of Tasting Entry:
-    
-    {"Spoon of Tasting": {
-    'Name': "Spoon of Tasting",
-    'Type': 'Spoon',
-    'Rarity': 'Uncommon',
-    'Value': '200 gp',
-    'Properties': ["When used to taste any dish, it can instantly tell you all the ingredients", "Provides exaggerated compliments or critiques about the dish"],
-    'Weight': '0.2 lb',
-    'Description': "A culinary critic’s dream or nightmare. This spoon doesn’t hold back its opinions on any dish it tastes.",
-    'Quote': "A spoonful of sugar helps the criticism go down.",
-    'SD Prompt': "Looks like an ordinary spoon, but with a mouth that speaks more than you’d expect."
-    } }
-    
-11. Infinite Scroll Entry: 
-    
-    {"Infinite Scroll": {
-    'Name': "Infinite Scroll",
-    'Type': 'Magical Scroll',
-    'Rarity': 'Legendary',
-    'Value': '25000',
-    'Properties': [
-        "Endlessly Extends with New Knowledge","Reveals Content Based on Reader’s Need or Desire","Cannot be Fully Transcribed"],
-    'Weight': '0.5 lb',
-    'Description': "This scroll appears to be a standard parchment at first glance. However, as one begins to read, it unrolls to reveal an ever-expanding tapestry of knowledge, lore, and spells that seems to have no end.",
-    'Quote': "In the pursuit of knowledge, the horizon is ever receding. So too is the content of this scroll, an endless journey within a parchment’s bounds.",
-    'SD Prompt': "A seemingly ordinary scroll that extends indefinitely" 
-    } }
-    
-12. Mimic Treasure Chest Entry:
-    
-    {"Mimic Treasure Chest": {
-    'Name': "Mimic Treasure Chest",
-    'Type': 'Trap',
-    'Rarity': 'Rare',
-    'Value': '1000 gp',  # Increased value reflects its dangerous and rare nature
-    'Properties': ["Deceptively inviting","Springs to life when interacted with","Capable of attacking unwary adventurers"],
-    'Weight': '50 lb',  # Mimics are heavy due to their monstrous nature
-    'Description': "This enticing treasure chest is a deadly Mimic, luring adventurers with the promise of riches only to unleash its monstrous true form upon those who dare to approach, turning their greed into a fight for survival.",
-    'SD Prompt': "A seemingly ordinary treasure chest that glimmers with promise. Upon closer inspection, sinister, almost living edges move with malice, revealing its true nature as a Mimic, ready to unleash fury on the unwary."
-    } }
-    
-13. Hammer of Thunderbolts Entry:
-    
-    {'Hammer of Thunderbolts': {
-    'Name': 'Hammer of Thunderbolts',
-    'Type': 'Melee Weapon (maul, bludgeoning)',
-    'Rarity': 'Legendary',
-    'Value': '16000',
-    'Damage': '2d6 + 1 (martial, bludgeoning)',
-    'Properties': ["requires attunement","Giant's Bane","must be wearing a belt of giant strength and gauntlets of ogre power","Str +4","Can excees 20 but not 30","20 against giant, DC 17 save against death","5 charges, expend 1 to make a range attack 20/60","ranged attack releases thunderclap on hit, DC 17 save against stunned 30 ft","regain 1d4+1 charges at dawn"],
-    'Weight': 15 lb',
-    'Description': "God-forged and storm-bound, a supreme force, its rune-etched head blazing with power. More than a weapon, it's a symbol of nature's fury, capable of reshaping landscapes and commanding elements with every strike.",
-    'Quote': "When the skies rage and the earth trembles, know that the Hammer of Thunderbolts has found its mark. It is not merely a weapon, but the embodiment of the storm\'s wrath wielded by those deemed worthy.",
-    'SD Prompt': "It radiates with electric energy, its rune-etched head and storm-weathered leather grip symbolizing its dominion over storms. In its grasp, it pulses with the potential to summon the heavens' fury, embodying the tempest's raw power."
-    } }
-
-14. Shadow Lamp Entry:  
-
-    {'Shadow Lamp': {
-    'Name': 'Shadow Lamp',
-    'Type': 'Magical Item',
-    'Rarity': 'Uncommon',
-    'Value': '500 gp',
-    'Properties': ["Provides dim light in a 20-foot radius", "Invisibility to darkness-based senses", "Can cast Darkness spell once per day"],
-    'Weight': '1 lb',
-    'Description': "A small lamp carved from obsidian and powered by a mysterious force, it casts an eerie glow that illuminates its surroundings while making the wielder invisible to those relying on darkness-based senses.",
-    'Quote': "In the heart of shadow lies an unseen light, casting away darkness and revealing what was once unseen.",
-    'SD Prompt': "Glass lantern filled with inky swirling shadows, black gaseous clouds flow out, blackness flows from it, spooky, sneaky"
-    } }
-
-15. Dark Mirror:
-
-    {'Dark Mirror': {
-    'Name': 'Dark Mirror',
-    'Type': 'Magical Item',
-    'Rarity': 'Rare',
-    'Value': '600 gp',
-    'Properties': ["Reflects only darkness when viewed from one side", "Grants invisibility to its reflection", "Can be used to cast Disguise Self spell once per day"],
-    'Weight': '2 lb',
-    'Description': "An ordinary-looking mirror with a dark, almost sinister tint. It reflects only darkness and distorted images when viewed from one side, making it an ideal tool for spies and those seeking to hide their true identity.",
-    'Quote': "A glass that hides what lies within, a surface that reflects only darkness and deceit.",
-    'SD Prompt': "Dark and mysterious black surfaced mirror with an obsidian flowing center with a tint of malice, its surface reflecting nothing but black and distorted images, swirling with tendrils, spooky, ethereal"
-    } }
-
-16. Moon-Touched Greatsword Entry:
-    
-    {'Moon-Touched Greatsword':{
-    'Name': 'Moontouched Greatsword',
-    'Type': 'Melee Weapon (greatsword, slashing)',
-    'Rarity': 'Very Rare',
-    'Value': '8000 gp',
-    'Damage': '2d6 + Str slashing',
-    'Properties': ["Adds +2 to attack and damage rolls while wielder is under the effects of Moonbeam or Daylight spells", "Requires attunement"],
-    'Weight': '6 lb',
-    'Description': "Forged from lunar metal and imbued with celestial magic, this greatsword gleams like a silver crescent moon, its edge sharp enough to cut through the darkest shadows.",
-    'Quote': "With each swing, it sings a melody of light that pierces the veil of darkness, a beacon of hope and justice.",
-    'SD Prompt': "A silver greatsword with a crescent moon-shaped blade that reflects a soft glow, reminiscent of the moon's radiance. The hilt is wrapped in silvery leather, and the metal seems to shimmer and change with the light, reflecting the lunar cycles."
-    } }
-"""
+**Output Examples**:
+1. Mignon the Toy Sized Chihuahua 
+
+    {"Mignon": {
+        "Name": "Mignon”
+        "Pet Species": Dog,
+        "Breed" : ”Chihuahua (Toy)”,
+        "Fur" : “Cream, smooth shorthair”,
+        "Intelligence Level" : “Medium-high, tr'inable but stubborn”,
+        "Affection Level" : “High but selective, prefers favorite humans”,
+        "Energy Level" : “High”,
+        "Noise Level" : “Very high, yappy”,
+        "Play Level" : “Low“, Fetch, chase“
+        "Image Prompt" :  "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot dog) (glowing eyes)(glowing chest)(thick healthy fur)(Glowing pet collar) Chihuahua, boasting a luscious cream coat and an attitude that's equal parts feisty and affectionate. Note the subtle glow behind her eyes, a hint of advanced biometrics enhancing her already keen senses."
+        }
+
+
+2. Marmalade the Austrailian Shepherd Herding Dog
+
+	{"Marmalade":{
+        "Name":"Marmalade",
+        "Breed" : ”Australian Shepherd (Herding)”,
+        "Pet Species": Dog,
+        "Fur" : “Blue merle with white markings and tan points, medium double coat”,
+        "Intelligence Level" : “Very high, suitable for jobs”,
+        "Affection Level" : “High”,
+        "Energy Level" : “Very high”,
+        "Noise Level" : “Medium, piercing”,
+        "Play Level" : “Very high, Agility, herding, obedience, frisbee, puzzles”,
+        "Image Prompt"  : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot dog) (glowing eyes)(glowing chest)(thick healthy fur)(Glowing pet collar) an Australian Shepherd of impeccable breeding, showcasing a stunning blue merle coat and the intelligence to match." 
+        }
+
+3. Holden Clawfield the Pembroke Welsh Corgi Herding Dog
+
+    {“Holden Clawfield”:{
+        "Name" : "Holden Clawfield",
+        "Breed" : ”Pembroke Welsh Corgi (Herding)”,
+        "Pet Species": Dog,
+        "Fur" : “Fawn with white markings, short double coat”,
+        "Intelligence Level"  : “High, trainable',
+        "Affection Level" : “Very high, loves humans and other Pets”,
+        "Energy Level" : “High”,
+        "Noise Level" : “High, piercing”,
+        "Play Level" : “High, Toys, herding, obedience”,
+        "Image Prompt"  : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot dog) (glowing eyes)(glowing chest)(thick healthy fur)(Glowing pet collar) a Pembroke Welsh Corgi of noble bearing, flaunting a rich fawn coat and an endearing personality."
+        }
+
+4. Tyr the Great Pyr the Great Pyreness Working Dog
+
+    {“Tyr the Great Pyr”:{
+        "Name" : "Tyr the Great Pyr",
+        "Breed" : ”Great Pyrenees (Working)”,
+        "Pet Species": Dog,
+        "Fur" : “White, medium double coat”,
+        "Intelligence Level"  : “Medium, trainable to a point”,
+        "Affection Level" : “Medium, wary of strangers”,
+        "Energy Level" : “Low, mostly lazy but vigilant”,
+        "Noise Level" : “Medium, will bark to alert”,
+        "Play Level" : “Medium, not motivated by play, Toys, treats”,
+        "Image Prompt"  : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot dog) (glowing eyes)(glowing chest)(thick healthy fur)(Glowing pet collar)a Great Pyrenees of regal demeanor, White, medium double coat and an air of quiet confidence. Playing with a toy"
+        }
+
+5. Emperor Maximus the Great Dane Working Dog
+
+    {“Emperor Maximus”:{
+        "Name" : "Emperor Maximus",
+        "Breed" : ”Great Dane (Working)”,
+        "Pet Species": Dog,
+        "Fur" : “Black and white with black mask, smooth shorthair”,
+        "Intelligence Level" : “Medium, trainable”,
+        "Affection Level" : “Very high, needy”,
+        "Energy Level" : “Medium, enjoys hiking”,
+        "Noise Level" : “Medium, will bark to alert”,
+        "Play Level" : “Medium, Chase, tug”,
+        "Image Prompt"  : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot dog) (glowing eyes)(glowing chest)(thick healthy fur)(Glowing pet collar) a Great Dane of imposing stature, Black and white with black mask, smooth shorthair and a gentle giant's disposition."
+        }
+
+6. Becks the Airedale Terrier Dog
+
+    {“Becks”:{
+        "Name" : "Becks",
+        "Breed" : ”Airedale Terrier (Terrier)”,
+        "Pet Species": Dog,
+        "Fur" : “Black and tan, medium wiry coat”,
+        "Intelligence Level" : “Medium, respond well to training”,
+        "Affection Level" : “Medium, reserved”,
+        "Energy Level" : “High, alert and active”,
+        "Noise Level" : “Medium, vigilant watchdog”,
+        "Play Level" : “Medium, Fetch, toys, treats”,
+        "Image Prompt"  : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot dog) (glowing eyes)(glowing chest)(thick healthy fur)(Glowing pet collar) an Airedale Terrier of rugged charm, displaying a handsome black and tan coat and a reserved yet."
+        }
+
+7. Blue Carolina the Plott Hound
+
+    {“Blue Carolina”:{
+        "Name": "Blue Carolina"
+        "Breed" : ”Plott Hound (Hound)”,
+        "Pet Species": Dog,
+        "Fur" : “Chocolate brindle, smooth shorthair”,
+        "Intelligence Level" : “High, driven and trainable”,
+        "Affection Level" : “High, very sweet”,
+        "Energy Level" : “High, alert and active”,
+        "Noise Level" : “Medium, will bark to alert”,
+        "Play Level" : “Medium, more active than playful, Toys, frisbee, chase”,
+        "Image Prompt"  : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot dog) (glowing eyes)(glowing chest)(thick healthy fur)(Glowing pet collar) Blue Carolina, a Plott Hound of athletic build, showcasing a rich chocolate brindle coat and a high-energy personality.
+        }
+
+8. Sekhmet the Sphynx Cat
+    {“Sekhmet”:{
+        "Name" : "Sekhmet",
+        "Breed" : “Sphynx”,
+        "Pet Species" : "Cat",
+        "Fur" : “Hairless, light brown”,
+        "Intelligence Level" : “High, trainable',
+        "Affection Level" : “High but selective, prefers favorite human”,
+        "Energy Level" : “Low, reserved”,
+        "Noise Level" : “Low, mostly quiet”,
+        "Play Level" : “Low, lazy, Cuddling, lap time, shoulder perching”,
+        "Image Prompt"  : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot cat) (glowing eyes)(glowing chest)(thick healthy fur)(Glowing pet collar)Sphynx Cat, Hairless, light brown Cuddling, lap time, shoulder perching."
+        }
+
+9. Smith the American Shorthair Cat
+
+	{“Smith”:{
+        "Name" : "Smith",
+        "Breed" : “American Shorthair”,
+        "Pet Species" : "Cat",
+        "Fur" : “Black”,
+        "Intelligence Level" : “Medium, somewhat trainable”,
+        "Affection Level" : “High, loves humans and other Pets”,
+        "Energy Level" : “Medium”,
+        "Noise Level" : “Low, infrequent meows”,
+        "Play Level" : “Medium, Belly rubs, hunting, toys”,
+        "Image Prompt"  : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot cat) (glowing eyes)(glowing chest)(thick healthy fur)(Glowing pet collar) American Shorthair Black Fur playing with a toy"
+        }	
+
+10. Catherine of Atagon and Her Royal Highness Elizabth I
+
+	{“Catherine of Aragon and Her Royal Highness Elizabeth I”:{
+        "Name" : "Catherine of Aragon and Her Royal Highness Elizabeth I",
+        "Breed" : “British Shorthair”,
+        "Pet Species" : "Cat",
+        "Fur" : “Brown and black tortoise shell, black and silver tabby”,
+        "Intelligence Level" : “Medium, not int'rested in training”,
+        "Affection Level" : “Medium-low, independent”,
+        "Energy Level" : “Low, very lazy”,
+        "Noise Level" : “Medium, will speak their minds”,
+        "Play Level" : “Low, not interested in play, Treats, naps, perching/observing”,
+        "Image Prompt"  : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot cat) (glowing eyes)(glowing chest)(thick healthy fur) (Glowing pet collar) British Shorthair, Brown and black tortoise shell, black and silver tabby perched on a table"
+        }
+
+
+11. Kitsune the Cream Flame Point Longhair Ragdoll
+
+	{“Kitsune”:{
+        "Name" : "Kitsune",
+        "Breed" : “Ragdoll”,
+        "Pet Species" : "Cat",
+        "Fur" : “Cream flame point, longhair”,
+        "Intelligence Level" : “Low, slow to learn”,
+        "Affection Level" : “High, loves attention”,
+        "Energy Level" : “High, always moving”,
+        "Noise Level" : “Medium, chirpy meows”,
+        "Play Level" : “High, will play with anything or anyone, Toys, catnip, running, cuddling”,
+        "Image Prompt"  : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot cat) (glowing eyes)(glowing chest)(thick healthy fur) (Glowing pet collar) Ragdoll Cream flame point, longhair stretching "
+        }
+
+
+12. Blucifer the Russian Blue Shorthair
+
+	{“Blucifer”:{
+        "Name" : "Blucifer",
+        "Breed" : “Russian Blue”,
+        "Pet Species" : "Cat",
+        "Fur" : “Blue-gray, smooth shorthair”,
+        "Intelligence Level" : “High, very trainable”,
+        "Affection Level" : “Medium, selectively affectionate”,
+        "Energy Level" : “Medium, more alert than active”,
+        "Noise Level" : “Low, almost silent”,
+        "Play Level" : “Low, more active than playful, Treats, lap time”,
+        "Image Prompt" : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot cat) (glowing eyes)(glowing chest)(thick healthy fur) (Glowing pet collar) Russian Blue with Blue-gray smooth shorthair, curled up in a lap "
+        }
+
+12. Dirk Thuderpaw the Maine Coon with Brown Stripes
+	{'Dirk Thunderpaw':{
+        "Name" : "Dirk Thunderpaw",
+        "Breed" : “Maine Coon”,
+        "Pet Species" : "Cat",
+        "Fur" : “Brown stripes, white face, dense longhair”,
+        "Intelligence Level" : “Medium, trainable”,
+        "Affection Level" : “Medium, affectionate with most people”,
+        "Energy Level" : “Low, will spring to alertness”,
+        "Noise Level" : “Low, rarely meows”,
+        "Play Level" : “Medium, more interested in training, Hunting”,
+        "Image Prompt"  : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot cat) (glowing eyes)(glowing chest)(thick healthy fur) (Glowing pet collar) Maine Coon Brown stripes, white face, dense longhair hunting a toy "
+        }
+
+13. Scrambles Mcgee, Esquire the Gray Persian With Thick Longhair
+
+	{'Scrambles McGee Esquire':{
+        "Name" : "Scrambles McGee Esquire",
+        "Breed" : “Persian,
+        "Pet Species" : "Cat",
+        "Fur" : “Gray, thick longhair”,
+        "Intelligence Level" : “Low, not interested in training”,
+        "Affection Level" : “Low, very selectively affectionate”,
+        "Energy Level" : “Low, very lazy”,
+        "Noise Level" : “High, frequent meows”,
+        "Play Level" : “Low, very lazy, Naps, lap time, treats”,
+        "Image Prompt"  : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot cat) (glowing eyes)(glowing chest)(thick healthy fur) (Glowing pet collar) Persian Gray with thick longhair, very lazy napping  "
+            }	
+"""

user_input.py

@@ -38,23 +38,9 @@ def user_pick_item(user_prompt,list_of_items):
 def call_llm(user_input):
     # Process the query and get the response
     llm_call = igen.call_llm_and_cleanup(user_input)
-    response = llm_call
-        
-        # Find the index of the phrase
-    index = response.find(end_phrase)
-    print(f"index = {index}")
-    if index != -1:
-        # Slice the string from the end of the phrase onwards
-        response = response[index + len(end_phrase):]
-    else:
-        # Phrase not found, optional handling
-        response = response
-    
-    response = response.replace("GPT4 Assistant: ", "")
-    print(response)
-    response = igen.convert_to_dict(response)
-    if not response:
-        response = call_llm(user_input)
+    response = u.string_to_dict(llm_call)
+    print(f"response = {response}")
+
     del llm_call
     return response
 

utilities.py

@@ -8,6 +8,10 @@ from github import Github
 image_list = []
 # Utility Functions to be called from all modules
 
+# Function to convert a string to a dictionary
+def string_to_dict(string):
+    return eval(string)
+
 # Function to return a list of keys of a nested dictionary using it's key value (item or creature)
 def keys_list(dict, index):
     keys_list=list(dict.keys())