Cleanup files and add requirements.txt

app.py

@@ -1,4 +1,3 @@
-import replicate
 import img2img
 import card_generator as card
 import utilities as u

img2img.py

@@ -1,6 +1,3 @@
-from diffusers import (StableDiffusionXLImg2ImgPipeline, AutoencoderKL)                   
-from diffusers.utils import load_image
-import torch
 import time
 import utilities as u
 import card_generator as card

inventory.py

@@ -1,52 +0,0 @@
-
-
-inventory = {
-    'Shortsword': {
-            'Name' : 'Shortsword',
-            'Type' : 'Melee Weapon (martial, sword)',
-            'Value': '10 gp',
-            'Properties': ['Finesse, Light '],
-            'Damage': '1d6 + proficiency + Dex or Str',
-            'Weight': '2 lb',
-            'Description': 'Gleaming with a modest radiance, the shortsword boasts a keen edge and a leather-wrapped hilt, promising both grace and reliability in the heat of combat.',
-            'Quote': 'In the heart of battle, the shortsword proves not just a weapon, but a steadfast companion, whispering paths of valor to those who wield it.',
-            'SD Description' : 'high resolution, blank magic card,detailed high quality intricate border, decorated textbox, high quality magnum opus cgi drawing of a steel shortsword'
-                    },          
-            
-    'Health Potion': {
-        'Name' : 'Health Portion',
-        'Type' : 'Potion',
-        '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.'
-            },
-            
-            
-    'Wooden Shield': {
-        'Name' : 'Wooden Shield',
-        'Type' : 'Armor, Shield',
-        'Value': '15 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.'
-        },
-                
-    'Magical Helmet': {
-        'Name' : 'Magical Helmet of Perception',
-        'Type' : 'Magical Item (armor, helmet)',
-        'Value': '120 gp',
-        'Properties': ['+ 1 to AC', 'Grants the wearer enhanced 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.'
-                }
-            }
-   
-{'id': 'cmpl-5b0ed6c7-2326-473f-8f11-32d3f079edc2',
-  'object': 'text_completion', 
-  'created': 1709094107, 
-  'model': '../models/starling-lm-7b-alpha.Q8_0.gguf', 
-  'choices': [{'text': ' Here\'s an example of a structured inventory entry for a Mimic Treasure Chest as per your request:\n\n```python\n{\n    \'Mimic Treasure Chest\': {\n        \'Name\': \'Mimic Treasure Chest\',\n        \'Type\': \'Trap\',\n        \'Rarity\': \'Rare\',\n        \'Value\': \'1000 gp\', \n        \'Properties\': [\n            \'Deceptively inviting\', \n            \'Springs to life when interacted with\', \n            \'Capable of attacking unwary adventurers\'\n            ],\n        \'Weight\': \'50 lb\', \n        \'Description\': \'At first glance, this chest appears to be laden with treasure, beckoning to all who gaze upon it. However, it harbors a deadly secret: it is a Mimic, a cunning and dangerous creature that preys on the greed of adventurers. With its dark magic, it can perfectly imitate a treasure chest, only to reveal its true, monstrous form when approached. Those who seek to plunder its contents might find themselves in a fight for their lives.\',\n        \'Quote\': \'"Beneath the guise of gold and riches lies a predator, waiting with bated breath for its next victim."\',\n        \'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.\'\n    }\n}\n```\n\nKeep in mind that mimics are typically found in dungeons and are known to take on the form of doors and chests. This example follows that theme while also providing information on the mimic\'s rarity, value, properties, and weight.', 'index': 0, 'logprobs': None, 'finish_reason': 'stop'}], 'usage': {'prompt_tokens': 4287, 'completion_tokens': 405, 'total_tokens': 4692}}

main.py

@@ -1,318 +0,0 @@
-import replicate
-import img2img
-import card_generator as card
-import utilities as u
-import ctypes
-import user_input as useri
-import gradio as gr
-import template_builder as tb
-
-# This is a fix for the way that python doesn't release system memory back to the OS and it was leading to locking up the system
-libc = ctypes.cdll.LoadLibrary("libc.so.6")
-M_MMAP_THRESHOLD = -3
-
-# Set malloc mmap threshold.
-libc.mallopt(M_MMAP_THRESHOLD, 2**20)
-initial_name = "A Crowbar"
-
-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':''
-                            }
-    
-    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('')
-
-    selected_border_image = gr.State('./card_templates/Moonstone Border.png')
-    num_image_to_generate = gr.State(4)
-    generated_image_list = gr.State([])
-    selected_generated_image = gr.State()
-    selected_seed_image = gr.State()
-    built_template = gr.State()  
-    mimic = None
-
-    def set_textbox_defaults(textbox_default_dict, key):
-        item_name = textbox_default_dict[key]
-        return item_name
-
-
-    # Function called when user generates item info, then assign values of dictionary to variables, output once to State, twice to textbox
-    def generate_text_update_textboxes(user_input):
-        u.reclaim_mem()
-
-
-        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]
-    
-    # Called on user selecting an image from the gallery, outputs the path of the image
-    def assign_img_path(evt: gr.SelectData):          
-        img_dict = evt.value
-        print(img_dict)
-        selected_image_path = img_dict['image']['url']
-        print(selected_image_path)
-        return selected_image_path   
-    
-     # Make a list of files in image_temp and delete them
-    def delete_temp_images():
-        image_list = u.directory_contents('./image_temp')
-        u.delete_files(image_list)
-        u.image_list.clear()
-    
-    # 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"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
-        u.reclaim_mem()
-
-        #generated_image_list = img2img.generate_image(num_img,sd_prompt,item_name,selected_border)
-        return image_list
-    
-    def build_template(selected_border, selected_seed_image):
-        image_list = tb.build_card_template(selected_border, selected_seed_image)
-        return image_list, image_list
-   
-
-    # Beginning of UI Page
-    # Beginning of UI Page
-    gr.HTML(""" <div id="inner"> <header>
-            <h1>Item Card Generator</h1>
-            <p>
-            With this AI driven tool you will build a collectible style card of a fantasy flavored item with details.
-            </p>
-            </div>""")
-    
-    
-    gr.HTML(""" <div id="inner"> <header>
-            <h2><b>First:</b> Build a Card Template</h2>
-                </div>""")
-    with gr.Row():
-        with gr.Column():
-
-    # Template Gallery instructions
-            gr.HTML(""" <div id="inner"> <header>
-                        <h3>1. Click a border from the 'Card Template Gallery'</h3> 
-                        </div>""")
-        
-    border_gallery = gr.Gallery(label = "Card Template Gallery", 
-                                    scale = 2,
-                                    value = useri.index_image_paths("Drakosfire/CardGenerator", "seed_images/card_templates"),
-                                    show_label = True,
-                                    columns = [3], rows = [3],
-                                    object_fit = "contain",
-                                    height = "auto",
-                                    elem_id = "Template Gallery")
-    
-    gr.HTML(""" <div id="inner"> <header>
-                <h3>2. Click a image from the Seed Image Gallery</h3><br>
-                </div>""")
-    
-    border_gallery.select(assign_img_path, outputs = selected_border_image)
-    seed_image_gallery = gr.Gallery(label= " Image Seed Gallery",
-                                    scale = 2,
-                                    value = useri.index_image_paths("Drakosfire/CardGenerator", "seed_images/item_seeds"),
-                                    show_label = True,
-                                    columns = [3], rows = [3],
-                                    object_fit = "contain",
-                                    height = "auto",
-                                    elem_id = "Template Gallery",
-                                    interactive=True)                 
-
-    gr.HTML(""" <div id="inner"> <header><h4> -Or- Upload your own seed image, by dropping it into the 'Generated Template Gallery' </h4><br>
-                <h3>3. Click 'Generate Card Template'</h3><br>
-            </div>""")
-    
-    built_template_gallery = gr.Gallery(label= "Generated Template Gallery",
-                                        scale = 1,
-                                        value = None,
-                                        show_label = True,
-                                        columns = [4], rows = [4],
-                                        object_fit = "contain",
-                                        height = "auto",
-                                        elem_id = "Template Gallery",
-                                        interactive=True,
-                                        type="filepath") 
-    
-    seed_image_gallery.select(assign_img_path, outputs = selected_seed_image)
-    built_template_gallery.upload(u.receive_upload, inputs=built_template_gallery, outputs= selected_seed_image)
-    build_card_template_button = gr.Button(value = "Generate Card Template")
-    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>
-                   </div>""")  
-    gr.HTML(""" <div id="inner"> <header>
-                        <h3>1. Use a few words to describe the item 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)
-        item_text_generate = gr.Button(value = "Generate item text", scale=1)
-
-    gr.HTML(""" <div id="inner"> <header>
-                <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')
-                        
-                # 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')
-    
-    
-    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')
-         
-    gr.HTML(""" <div id="inner"> <header>
-                <h2> <b>Third:</b> Click 'Generate Cards' to generate 4 cards to choose from. </h2>
-                </div>""")          
-    card_gen_button = gr.Button(value = "Generate Cards", elem_id="Generate Card Button")
-           
-    # No longer Row Context, in context of entire Block
-    gr.HTML(""" <div id="inner"> <header>
-                <h2> <b>Fourth:</b> Click your favorite card then add text, or click 'Generate Four Card Options' again.<br>
-                 </h2>
-                </div>""")       
-
-    with gr.Row():
-        generate_gallery = gr.Gallery(label = "Generated Cards",
-                                    value = [],
-                                    show_label= True,
-                                    scale= 5,
-                                    columns =[2], rows = [2],
-                                    object_fit= "fill",
-                                    height = "768",
-                                    elem_id = "Generated Cards Gallery"
-                                    )
-        generate_final_item_card = gr.Button(value = "Add Text", elem_id = "Generate user card")
-    
-    
-    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)
-    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])          
-        
-        
-    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
-                                                                        ], 
-                                                                        outputs = generate_gallery )
-    
-    
-if __name__ == '__main__':
-    demo.launch(server_name = "0.0.0.0", server_port = 8000, share = False, allowed_paths = ["/media/drakosfire/Shared/","/media/drakosfire/Shared/MerchantBot/card_templates"])
-        
-
-
-
-
-
-
-
-    
-
-
-   
-
-

requirements.txt

@@ -0,0 +1,7 @@
+torch
+
+PyGithub
+pillow
+openai
+sentence_transformers
+gradio