cosmetic changes, changed info generation to improve wait time.

README.md

@@ -1,8 +1,8 @@
 ---
 title: NPCGenerator
-emoji: 👁
-colorFrom: pink
-colorTo: blue
+emoji: ⚔️
+colorFrom: grey
+colorTo: grey
 sdk: gradio
 sdk_version: 6.17.3
 python_version: '3.12'

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import random
 import spaces
 import torch
-from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
+from transformers import pipeline
 
 
 
@@ -10,12 +10,14 @@ pipe = pipeline("text-generation", model="TinyLlama/TinyLlama-1.1B-Chat-v1.0", t
 
 DESC_SYSTEM_PROMPT = (
     "Write a description of a fantasy character based on the user prompt"
-    "Include only the following items; Fantasy race, height, build, eye color, hair color/cut, clothing, notable physical features"
-    "Do not describe any mannerisms or activities."
+    "response Must include: fantasy Race, height, build, eye color, hair color, clothing."
+    "response may include: noticeable features such as scars, tattoos, jewelry, or expression"
+    "Response must not include any extraneous details."
 )
 BACK_SYSTEM_PROMPT = (
     "Write a short timeline for this character"
     "Keep each part short and fitting for a background character in a fantasy world."
+    "include things like occupations they have had, family ties, livelihood, or mannerisms."
     "Include one line about what the character is thinking or doing right now."
     "Do not include any descriptions of physical features."
 )
@@ -29,37 +31,33 @@ constitution = ''
 intelligence = ''
 wisdom = ''
 charisma = ''
+    
 
 @spaces.GPU
-def stats():
+def char_description(prompt):
+
+    prompt1 = f"{DESC_SYSTEM_PROMPT}\n\nCharacter Description: {prompt}\n\nDescribe:"
+    output1 = pipe(prompt1, return_full_text=False, max_new_tokens=128)
+    prompt2 = f"{BACK_SYSTEM_PROMPT}\n\n{output1}\n\nUser prompt: {prompt}\n\nDescribe:"
+    output2 = pipe(prompt2, return_full_text=False, max_new_tokens=128)
     global strength
-    strength_prompt = f"{STAT_SYSTEM_PROMPT} Strength"
+    strength_prompt = f"{STAT_SYSTEM_PROMPT} Strength for {output1}"
     strength = pipe(strength_prompt, return_full_text=False)
     global dexterity
-    dexterity_prompt = f"{STAT_SYSTEM_PROMPT} Dexterity"
+    dexterity_prompt = f"{STAT_SYSTEM_PROMPT} Dexterity for {output1}"
     dexterity = int(pipe(dexterity_prompt, return_full_text=False))
     global constitution
-    constitution_prompt = f"{STAT_SYSTEM_PROMPT} Constitution"
+    constitution_prompt = f"{STAT_SYSTEM_PROMPT} Constitution for {output1}"
     constitution = int(pipe(constitution_prompt, return_full_text=False))
     global intelligence
-    intelligence_prompt = f"{STAT_SYSTEM_PROMPT} Intelligence"
+    intelligence_prompt = f"{STAT_SYSTEM_PROMPT} Intelligence for {output1}"
     intelligence = int(pipe(intelligence_prompt, return_full_text=False))
     global wisdom
-    wisdom_prompt = f"{STAT_SYSTEM_PROMPT} Wisdom"
+    wisdom_prompt = f"{STAT_SYSTEM_PROMPT} Wisdom for {output1}"
     wisdom = int(pipe(wisdom_prompt, return_full_text=False))
     global charisma
-    charisma_prompt = f"{STAT_SYSTEM_PROMPT} Charisma"
+    charisma_prompt = f"{STAT_SYSTEM_PROMPT} Charisma for {output1}"
     charisma = int(pipe(charisma_prompt, return_full_text=False))
-    
-
-@spaces.GPU
-def char_description(prompt):
-
-    prompt1 = f"{DESC_SYSTEM_PROMPT}\n\nCharacter Description: {prompt}\n\nDescribe:"
-    output1 = pipe(prompt1, return_full_text=False, max_new_tokens=256)
-    prompt2 = f"{BACK_SYSTEM_PROMPT}\n\n{output1}\n\nUser prompt: {prompt}\n\nDescribe:"
-    output2 = pipe(prompt2, return_full_text=False, max_new_tokens=256)
-    stats
     return output1[0]["generated_text"].strip(),output2[0]["generated_text"].strip()
 
 @spaces.GPU
@@ -68,10 +66,27 @@ def rand_char_description():
     races = input_file.readline()
     rand_race = random.choice(races)
     prompt1 = f"{DESC_SYSTEM_PROMPT}\n\nCharacter Description: {rand_race}\n\nDescribe:"
-    output1 = pipe(prompt1, return_full_text=False, max_new_tokens=256)
+    output1 = pipe(prompt1, return_full_text=False, max_new_tokens=128)
     prompt2 = f"{BACK_SYSTEM_PROMPT}\n\n{output1}\n\nUser prompt: {rand_race}\n\nDescribe:"
-    output2 = pipe(prompt2, return_full_text=False, max_new_tokens=256)
-    stats
+    output2 = pipe(prompt2, return_full_text=False, max_new_tokens=128)
+    global strength
+    strength_prompt = f"{STAT_SYSTEM_PROMPT} Strength for {output1}"
+    strength = pipe(strength_prompt, return_full_text=False)
+    global dexterity
+    dexterity_prompt = f"{STAT_SYSTEM_PROMPT} Dexterity for {output1}"
+    dexterity = int(pipe(dexterity_prompt, return_full_text=False))
+    global constitution
+    constitution_prompt = f"{STAT_SYSTEM_PROMPT} Constitution for {output1}"
+    constitution = int(pipe(constitution_prompt, return_full_text=False))
+    global intelligence
+    intelligence_prompt = f"{STAT_SYSTEM_PROMPT} Intelligence for {output1}"
+    intelligence = int(pipe(intelligence_prompt, return_full_text=False))
+    global wisdom
+    wisdom_prompt = f"{STAT_SYSTEM_PROMPT} Wisdom for {output1}"
+    wisdom = int(pipe(wisdom_prompt, return_full_text=False))
+    global charisma
+    charisma_prompt = f"{STAT_SYSTEM_PROMPT} Charisma for {output1}"
+    charisma = int(pipe(charisma_prompt, return_full_text=False))
     return output1[0]["generated_text"].strip(),output2[0]["generated_text"].strip()
 
 
@@ -81,7 +96,7 @@ def rand_char_description():
 with gr.Blocks() as demo:
     title = gr.HTML("<center><h1>NPC Generator</h1></center>")
     description = gr.HTML("<center>For tabletop GMs when players ask for a description of random people in the tavern.</center>")
-    tags = gr.HTML("<center>| Backyard AI | Tiny Titan | Well Tuned |</center>")
+    tags = gr.HTML("<center>| Backyard AI | Tiny Titan |</center>")
 
     with gr.Row(equal_height = True):
         with gr.Column(scale = 2):
@@ -104,11 +119,11 @@ with gr.Blocks() as demo:
             text_out1 = gr.Textbox(label = "Description")
             text_out2 = gr.Textbox(label = "Backstory")
             btn.click(fn = char_description,inputs = text_in, outputs = [text_out1, text_out2])
-            rand.click(fn = rand_char_description,inputs = text_in, outputs = [text_out1, text_out2])
+            rand.click(fn = rand_char_description,inputs = None, outputs = [text_out1, text_out2])
 
 
         with gr.Column(scale = 0):
-            ac = gr.Textbox(value = lambda:random.randint(1,20), label = "AC", interactive = False)
+            ac = gr.Textbox(value = lambda:random.randint(3,20), label = "AC", interactive = False)
             speed = gr.Textbox(value = lambda:random.randint(10,30), label = "HP", interactive = False)
             speed = gr.Textbox(value = lambda:"25 ft", label = "Speed", interactive = False)