Spaces:
Sleeping
Sleeping
Got up to docker build ready again, was importing things in block-builder like torch and utilities that weren't needed
a3dba5f
# block_id is a global variable that is used to keep track of the current block id | |
block_id = 0 | |
# This function takes in the user input and block_id and returns a list of blocks built with textareas | |
def build_blocks(user_input, block_id): | |
list_of_blocks = [] | |
title_block = build_title_block(user_input['store_name'], user_input['store_description'], user_input['store_backstory'], user_input['store_reputation']) | |
block_id = block_id + 1 | |
list_of_blocks.append(title_block) | |
store_image_block = build_image_block(user_input['storefront_sd_prompt'], block_id) | |
block_id = block_id + 1 | |
list_of_blocks.append(store_image_block) | |
store_properties_block = build_store_properties_block(store_type= user_input['store_type'], | |
store_size= user_input['store_size'], | |
store_hours= user_input['store_hours'], | |
store_location= user_input['store_location'], | |
store_owners= user_input['store_owners'], | |
store_employees= user_input['store_employees'], | |
store_services= user_input['store_services'], | |
store_specialties= user_input['store_specialties'], | |
store_reputation= user_input['store_reputation'], | |
store_rumors= user_input['store_rumors'], | |
block_id= block_id) | |
block_id = block_id + 1 | |
list_of_blocks.append(store_properties_block) | |
owner_id = 1 | |
# Employee and owner could be combined into a single function with a parameter for the type of block | |
# Iterate over owners and generate owner image and details block | |
owner_title = "Owner" | |
if len(user_input['store_owners']) > 1: | |
owner_title = "Owners" | |
owner_title_block = f"""<h2 id="owner">{owner_title}</h2>""" | |
for owner in user_input['store_owners']: | |
owner_image_block = build_image_block(owner['sd_prompt'], block_id) | |
block_id = block_id + 1 | |
list_of_blocks.append(owner_image_block) | |
owner_block = build_owner_block(owner,owner_id, owner_title_block, block_id) | |
block_id = block_id + 1 | |
list_of_blocks.append(owner_block) | |
owner_id += 1 | |
employee_id = 1 | |
# Iterate over employees and generate employee image and details block | |
employee_title = "Employee" | |
if len(user_input['store_employees']) > 1: | |
employee_title = "Employees" | |
employee_title_block = f"""<h2 id="employee">{employee_title}</h2>""" | |
for employee in user_input['store_employees']: | |
employee_image_block = build_image_block(employee['sd_prompt'], block_id) | |
block_id = block_id + 1 | |
list_of_blocks.append(employee_image_block) | |
employee_block = build_employee_block(employee, employee_id, employee_title_block, block_id) | |
block_id = block_id + 1 | |
list_of_blocks.append(employee_block) | |
employee_id += 1 | |
customer_id = 1 | |
for customer in user_input['store_customers']: | |
customers_block = build_section_entry_block('Customers',customer, customer_id, block_id) | |
block_id = block_id + 1 | |
customer_id += 1 | |
list_of_blocks.append(customers_block) | |
quest_id = 1 | |
for quest in user_input['store_quests']: | |
quests_block = build_section_entry_block('Store Quests', quest, quest_id, block_id) | |
block_id = block_id + 1 | |
quest_id += 1 | |
list_of_blocks.append(quests_block) | |
services_id = 1 | |
for service in user_input['store_services']: | |
services_block = build_section_entry_block('Services', service, services_id, block_id) | |
block_id = block_id + 1 | |
services_id += 1 | |
list_of_blocks.append(services_block) | |
specialties_id = 1 | |
for specialty in user_input['store_specialties']: | |
specialty_block = build_section_entry_block('Specialties', specialty, specialties_id, block_id) | |
block_id = block_id + 1 | |
specialties_id += 1 | |
list_of_blocks.append(specialty_block) | |
security_id = 1 | |
for security in user_input['store_security']: | |
security_block = build_section_entry_block('Security', security, security_id, block_id) | |
block_id = block_id + 1 | |
security_id += 1 | |
list_of_blocks.append(security_block) | |
inventory_block = build_inventory_block(user_input['inventory'], block_id) | |
block_id = block_id + 1 | |
list_of_blocks.append(inventory_block) | |
return list_of_blocks | |
# Take in a specific item type and item, and return the html for that item | |
def process_into_html(item_type,item, block_id): | |
item_html = f"""<tr> | |
<td align="left"><strong>{item_type}</strong></td> | |
<td align="right"><textarea class="string-action-description-textarea" id="{item_type}-{block_id}" | |
hx-post="/update-stats" hx-trigger="change" hx-target="#user-store-owners-{block_id}t" hx-swap="outerHTML" | |
title="">{item}</textarea></td> | |
</tr>""" | |
return item_html | |
# Take in a specific iterable type and iterable, and return the html for that iterable | |
def process_iterable_into_html(iterable_type, iterable, block_id): | |
iterable_html = f"""""" | |
for item in iterable: | |
item_html = f"""<tr> | |
<td align="left"><strong>{iterable_type}</strong></td> | |
<td align="right"><textarea class="string-action-description-textarea" id="user-store-owners-{block_id}" | |
hx-post="/update-stats" hx-trigger="change" hx-target="#user-store-owners-{block_id}t" hx-swap="outerHTML" | |
title="">{item['name']}</textarea></td> | |
</tr>""" | |
iterable_html += item_html | |
return iterable_html | |
# Take in a list of rumors and return the html for that list of rumors | |
def process_rumors_into_html(rumors, block_id): | |
rumors_html = f"""""" | |
for rumor in rumors: | |
rumor_html = f"""<tr> | |
<td align="left"><strong>Store Rumors</strong></td> | |
<td align="right"><textarea class="string-action-description-textarea" id="user-store-rumors-{block_id}" | |
hx-post="/update-stats" hx-trigger="change" hx-target="#user-store-rumors-{block_id}t" hx-swap="outerHTML" | |
title="Rumors">{rumor}</textarea></td> | |
</tr>""" | |
rumors_html += rumor_html | |
return rumors_html | |
# Take in a list of secrets and return the html for that list of secrets | |
def process_secrets_into_html(secrets, block_id): | |
secrets_html = f"""""" | |
for secret in secrets: | |
secret_html = f"""<tr> | |
<td align="left"><strong>Secrets</strong></td> | |
<td align="right"><textarea class="string-action-description-textarea" id="user-store-rumors-{block_id}" | |
hx-post="/update-stats" hx-trigger="change" hx-target="#user-store-rumors-{block_id}t" hx-swap="outerHTML" | |
title="Store Size">{secret}</textarea></td> | |
</tr>""" | |
secrets_html += secret_html | |
return secrets_html | |
# Block for title, description, backstory, and reputation | |
def build_title_block(title,description,backstory,reputation): | |
title_block_html = f"""<div class="block-item" data-block-id = {block_id}><h1> | |
<textarea class="title-textarea" id="user-store-title" hx-post="/update-stats" hx-trigger="change" hx-target="#user-store-name" hx-swap="outerHTML" title="Name of store">{title}</textarea></h1> | |
<div contenteditable="true" class="description-textarea" id="user-store-description" | |
hx-post="/update-stats" hx-trigger="change" | |
hx-target="#user-monster-description" hx-swap="outerHTML" | |
title="Any amount or style of description"> | |
<p>{f"{description} {backstory} {reputation}"}</p> | |
</div> """ | |
return title_block_html | |
# Block for image generation | |
def build_image_block(sd_prompt, block_id): | |
image_block_html = f""" | |
<div class="block-item" data-block-id="{block_id}"> | |
<textarea class="string-action-description-textarea" id="user-storefront-prompt-{block_id}" | |
hx-post="/update-stats" hx-trigger="change" hx-target="#user-sd-prompt" hx-swap="outerHTML" | |
title="Storefront image description">{sd_prompt}</textarea> | |
<button class="generate-image-button" data-block-id="{block_id}">Generate Image</button> | |
<img id="generated-image-{block_id}" alt="" style="display: none; cursor: pointer;"> | |
</div> | |
""" | |
return image_block_html | |
# Block for store properties | |
def build_store_properties_block(store_type, | |
store_size, | |
store_hours, | |
store_location, | |
store_owners, | |
store_employees, | |
store_services, | |
store_specialties, | |
store_reputation, | |
store_rumors, | |
block_id): | |
# This could be the iterable block function with additional flexibility | |
store_properties_base_html = f""" | |
<div class="block-item" data-block-id="{block_id}"> | |
<div class="block classTable frame decoration"> | |
<table> | |
<thead> | |
<tr> | |
<th align="left"></th> | |
<th align="center"></th> | |
<th align="center"></th> | |
</tr> | |
</thead> | |
<tbody> | |
<tr> | |
<td align="left"><strong>Size</strong></td> | |
<td align="right"><textarea class="string-action-description-textarea" id="user-store-size-{block_id}" | |
hx-post="/update-stats" hx-trigger="change" hx-target="#user-store-size-{block_id}t" hx-swap="outerHTML" | |
title="Store Size">{store_size}</textarea></td> | |
</tr> | |
<tr> | |
<td align="left"><strong>Town</strong></td> | |
<td align="right"><textarea class="string-action-description-textarea" id="user-store-town-{block_id}" | |
hx-post="/update-stats" hx-trigger="change" hx-target="#user-store-town-{block_id}t" hx-swap="outerHTML" | |
title="Store Size">{store_location['town']}</textarea></td> | |
</tr> | |
<tr> | |
<td align="left"><strong>District</strong></td> | |
<td align="right"><textarea class="string-action-description-textarea" id="user-store-district-{block_id}" | |
hx-post="/update-stats" hx-trigger="change" hx-target="#user-store-district-{block_id}t" hx-swap="outerHTML" | |
title="Store Size">{store_location['district']}</textarea></</td> | |
</tr> | |
<tr> | |
<td align="left"><strong>Street</strong></td> | |
<td align="right"><textarea class="string-action-description-textarea" id="user-store-street-{block_id}" | |
hx-post="/update-stats" hx-trigger="change" hx-target="#user-store-street-{block_id}t" hx-swap="outerHTML" | |
title="Store Size">{store_location['street']}</textarea></</td> | |
</tr> | |
<tr> | |
<td align="left"><strong>Type</strong></td> | |
<td align="right"><textarea class="string-action-description-textarea" id="user-store-type-{block_id}" | |
hx-post="/update-stats" hx-trigger="change" hx-target="#user-store-type-{block_id}t" hx-swap="outerHTML" | |
title="Store Size">{store_type}</textarea></</td> | |
</tr> | |
<tr> | |
<tr> | |
<td align="left"><strong>Store Hours</strong></td> | |
<td align="right"><textarea class="string-action-description-textarea" id="user-store-hours-{block_id}" | |
hx-post="/update-stats" hx-trigger="change" hx-target="#user-store-hours-{block_id}t" hx-swap="outerHTML" | |
title="Store Size">{store_hours}</textarea></td> | |
</tr> """ | |
store_owners = [] | |
store_employees = [] | |
owners_html = process_iterable_into_html('Store Owners', store_owners, block_id) | |
employees_html = process_iterable_into_html('Store Employees', store_employees, block_id) | |
store_specialties_html = process_iterable_into_html('Store Specialties', store_specialties, block_id) | |
store_services_html = process_iterable_into_html('Store Services', store_services, block_id) | |
store_rumors_html = process_rumors_into_html(store_rumors, block_id) | |
store_iterables_html = f""" | |
{owners_html} | |
{employees_html} | |
{store_services_html} | |
{store_specialties_html} | |
{store_rumors_html} | |
""" | |
store_end_html = f""" | |
<tr> | |
<td align="left"><strong>Store Reputation</strong></td> | |
<td align="right"><textarea class="string-action-description-textarea" id="user-store-reputation-{block_id}" | |
hx-post="/update-stats" hx-trigger="change" hx-target="#user-store-reputation-{block_id}t" hx-swap="outerHTML" | |
title="Store Size">{store_reputation}</textarea></td> | |
</tr> | |
</tbody> | |
</table> | |
</div> | |
</div> | |
""" | |
store_properties_block_html = f"""{store_properties_base_html} | |
{store_iterables_html} | |
{store_end_html}""" | |
return store_properties_block_html | |
# Block of owner table | |
def build_owner_block(owner, owner_id, owner_title_block, block_id): | |
# Owner block with values : Name, Race, Class, Description, Personality, Secrets, sd-prompt | |
# Process owner values into html | |
owner_name_html = process_into_html('Owner', owner['name'], block_id) | |
owner_race_html = process_into_html('Species', owner['species'], block_id) | |
owner_class_html = process_into_html('Class', owner['class'], block_id) | |
owner_description_html = process_into_html('Description', owner['description'], block_id) | |
owner_personality_html = process_into_html('Personality', owner['personality'], block_id) | |
owner_secrets_html = process_secrets_into_html(owner['secrets'], block_id) | |
# Build owner block html | |
# If owner_id is 1, add owner_title_block to owner_block_html | |
owner_block_html = f"""""" | |
owner_block_html += f"""<div class="block-item" data-block-id="{block_id}">""" | |
if owner_id == 1: | |
owner_block_html+= owner_title_block | |
owner_block_html += f"""<h3 id="owner_{owner_id}"><textarea class="subtitle-textarea" id="user-store-rumors-{block_id}" | |
hx-post="/update-stats" hx-trigger="change" hx-target="#user-store-rumors-{block_id}t" hx-swap="outerHTML" | |
title="Owner Name">{owner['name']}</textarea></h3>""" | |
owner_block_html += f"""<table> | |
<thead> | |
<tr> | |
<th align="center"></th> | |
<th align="center"></th> | |
</tr> | |
</thead> | |
<tbody> | |
{owner_name_html} | |
{owner_race_html} | |
{owner_class_html} | |
{owner_description_html} | |
{owner_personality_html} | |
{owner_secrets_html} | |
</tbody> | |
</table> | |
</div> | |
""" | |
return owner_block_html | |
# Block of employee table | |
def build_employee_block(employee, employee_id, employee_title_block, block_id): | |
# Employee block with name, role, species, description, personality, sd-prompt | |
# Process employee values into html | |
employee_name_html = process_into_html('Employee', employee['name'], block_id) | |
employee_role_html = process_into_html('Role', employee['role'], block_id) | |
employee_species_html = process_into_html('Species', employee['species'], block_id) | |
employee_description_html = process_into_html('Description', employee['description'], block_id) | |
employee_personality_html = process_into_html('Personality', employee['personality'], block_id) | |
# Build employee block html | |
employee_block_html = f"""""" | |
employee_block_html += f"""<div class="block-item" data-block-id="{block_id}">""" | |
if employee_id == 1: | |
employee_block_html += employee_title_block | |
employee_block_html += f"""<h3 id="owner_{employee_id}"><textarea class="subtitle-textarea" id="user-store-rumors-{block_id}" | |
hx-post="/update-stats" hx-trigger="change" hx-target="#user-store-rumors-{block_id}t" hx-swap="outerHTML" | |
title="Owner Name">{employee['name']}</textarea> | |
</h3> | |
<table> | |
<thead> | |
<tr> | |
<th align="center"></th> | |
<th align="center"></th> | |
</tr> | |
</thead> | |
<tbody> | |
{employee_name_html} | |
{employee_role_html} | |
{employee_species_html} | |
{employee_description_html} | |
{employee_personality_html} | |
</tbody> | |
</table> | |
</div> | |
""" | |
return employee_block_html | |
# Section to take in a section name, entry, entry_id, and block_id and return the html for that section | |
def build_section_entry_block(section, entry, entry_id, block_id): | |
section_block_html = f"""""" | |
section_block_html += f"""<div class="block-item" data-block-id="{block_id}">""" | |
if entry_id == 1: | |
section_block_html += f"""<h1 id="store-quests">{section}</h1> """ | |
entry_features_list = list(entry.keys()) | |
for feature in entry_features_list: | |
if feature == 'name': | |
section_block_html += f"""<h3 id="quest={entry_id}"> | |
<textarea class="subtitle-textarea" id="user-store-rumors-{block_id}" | |
hx-post="/update-stats" hx-trigger="change" hx-target="#user-store-rumors-{block_id}t" hx-swap="outerHTML" | |
title={section}>{entry['name']}</textarea> | |
</h3>""" | |
else: | |
feature_name = feature[0].upper() + feature[1:] | |
section_block_html += f"""<p> | |
<textarea class="string-action-description-textarea" id="user-store-rumors-{block_id}" | |
hx-post="/update-stats" hx-trigger="change" hx-target="#user-store-rumors-{block_id}t" hx-swap="outerHTML" | |
title={section}>{feature_name}: {entry[feature]}</textarea> | |
</p>""" | |
return section_block_html | |
def build_inventory_block(inventory, block_id): | |
inventory_block_html = f"""""" | |
inventory_block_html += f"""<div class="block-item" data-block-id="{block_id}">""" | |
inventory_block_html += f"""<div class="block classTable frame decoration"> | |
<h5 id="inventory">Inventory</h5> | |
<table> | |
<thead> | |
<tr> | |
<th align="center">Name</th> | |
<th align="center">Type</th> | |
<th align="left">Cost</th> | |
<th align="center">Properties</th> | |
</tr> | |
</thead> | |
<tbody>""" | |
# Create a list of the keys in the inventory, each key is the type of items | |
list_of_type = list(inventory.keys()) | |
# Iterate through keys and check if the value is greater than an empty list | |
for type in list_of_type: | |
inventory_type = inventory[type] | |
if len(inventory_type) > 0 : | |
# iterate through items in inventory type list, each item is a dictionary with prescribed values. | |
# Need to check for list in properties. | |
for item in inventory_type: | |
item['properties'] = ', '.join(item['properties']) | |
item_block_html = f"""<tr> | |
<td align="center"><textarea class="string-action-description-textarea" id="user-store-rumors-{block_id}" | |
hx-post="/update-stats" hx-trigger="change" hx-target="#user-store-rumors-{block_id}t" hx-swap="outerHTML" | |
title={type}>{item['name']}</textarea></td> | |
<td align="center"><textarea class="string-action-description-textarea" id="user-store-rumors-{block_id}" | |
hx-post="/update-stats" hx-trigger="change" hx-target="#user-store-rumors-{block_id}t" hx-swap="outerHTML" | |
title={type}>{item['type']}</textarea></td> | |
<td align="center"><textarea class="string-action-description-textarea" id="user-store-rumors-{block_id}" | |
hx-post="/update-stats" hx-trigger="change" hx-target="#user-store-rumors-{block_id}t" hx-swap="outerHTML" | |
title={type}>{item['cost']}</textarea></td> | |
<td align="center"><textarea class="string-action-description-textarea" id="user-store-rumors-{block_id}" | |
hx-post="/update-stats" hx-trigger="change" hx-target="#user-store-rumors-{block_id}t" hx-swap="outerHTML" | |
title={type}>{item['properties']}</textarea></td> | |
</tr>""" | |
inventory_block_html += item_block_html | |
inventory_block_html += f"""</tbody> | |
</table> | |
</div> | |
</div>""" | |
return inventory_block_html | |
"""<div class="Block_13"> | |
<div class="block classTable frame decoration"> | |
<h5 id="inventory">Inventory</h5> | |
<table> | |
<thead> | |
<tr> | |
<th align="center">Name</th> | |
<th align="center">Type</th> | |
<th align="left">Cost</th> | |
<th align="center">Properties</th> | |
</tr> | |
</thead> | |
<tbody> | |
<tr> | |
<td align="center">Poultry Drumsticks</td> | |
<td align="center">Meat</td> | |
<td align="left">1 gp per lbs</td> | |
<td align="center"></td> | |
</tr> | |
<tr> | |
<td align="center">Ground Beef</td> | |
<td align="center">Meat</td> | |
<td align="left">1 gp per lbs</td> | |
<td align="center"></td> | |
</tr> | |
<tr> | |
<td align="center">Pork Chops</td> | |
<td align="center">Meat</td> | |
<td align="left">1 gp per lbs</td> | |
<td align="center"></td> | |
</tr> | |
<tr> | |
<td align="center">Bacon Strips</td> | |
<td align="center">Meat</td> | |
<td align="left">1 gp per lbs</td> | |
<td align="center"></td> | |
</tr> | |
<tr> | |
<td align="center">Sausage Links</td> | |
<td align="center">Meat</td> | |
<td align="left">1 gp per lbs</td> | |
<td align="center"></td> | |
</tr> | |
<tr> | |
<td align="center">Mystic Minotaur Steak</td> | |
<td align="center">Exotic Meat</td> | |
<td align="left">25 gold per pound</td> | |
<td align="center">Grants temporary strength boost when consumed, Requires fine culinary skills to cook properly</td> | |
</tr> | |
<tr> | |
<td align="center">Quantum Quail</td> | |
<td align="center">Exotic Poultry</td> | |
<td align="left">15 gold each</td> | |
<td align="center">“Phases in and out of existence”, “Can enhance one’s agility”</td> | |
</tr> | |
<tr> | |
<td align="center">Invisible Bacon</td> | |
<td align="center">Mystical Meat</td> | |
<td align="left">10 gold per slice</td> | |
<td align="center">“Invisible to the naked eye”, “Tastes incredibly savory”, “Can only be seen with a special spell”</td> | |
</tr> | |
<tr> | |
<td align="center">Hydra Sausage</td> | |
<td align="center">Mythical Meat</td> | |
<td align="left">50 gold per link</td> | |
<td align="center">“Each bite regenerates after a while”, “Consuming too much may cause mild hallucinations”</td> | |
</tr> | |
<tr> | |
<td align="center">Cursed Cleaver</td> | |
<td align="center">Kitchen Equipment</td> | |
<td align="left">100 gold</td> | |
<td align="center">“Cuts through any meat effortlessly”, “Occasionally whispers in a long-forgotten language”, “Rumored to be haunted”</td> | |
</tr> | |
<tr> | |
<td align="center">Vampire Spice Mix</td> | |
<td align="center">Cooking Ingredient</td> | |
<td align="left">20 gold per pouch</td> | |
<td align="center">“Adds a distinct flavor”, “Enhances blood flow in the consumer”, “Leaves a lingering aftertaste of garlic”</td> | |
</tr> | |
<tr> | |
<td align="center">Phoenix Feather Skewers</td> | |
<td align="center">Cooking Utensil</td> | |
<td align="left">75 gold per set</td> | |
<td align="center">“Prevents meat from overcooking”, “Gives a slight warmth to cooked items”, “Reusable endlessly”</td> | |
</tr> | |
</tbody> | |
</table> | |
</div> <!--Close Block_12--> | |
</div>""" | |
#Text Area Template | |
"""<textarea class="string-action-description-textarea" id="user-store-rumors-{block_id}" | |
hx-post="/update-stats" hx-trigger="change" hx-target="#user-store-rumors-{block_id}t" hx-swap="outerHTML" | |
title="Rumors">{rumor}</textarea>""" | |
#Title Area Template | |
"""<h3 id="owner_{owner_id}"><textarea class="title-textarea" id="user-store-rumors-{block_id}" | |
hx-post="/update-stats" hx-trigger="change" hx-target="#user-store-rumors-{block_id}t" hx-swap="outerHTML" | |
title="Owner Name">{owner['name']}</textarea><</h3>""" | |