Added pretrained model and system prompts. We have a brain peoplegit add app.py requirements.txt

app.py

@@ -1,18 +1,63 @@
 import gradio as gr
 import random
+import spaces
+import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 
+
+
+pipe = pipeline("text-generation", model="TinyLlama/TinyLlama-1.1B-Chat-v0.6", torch_dtype=torch.bfloat16, device_map="auto")
+
+DESC_SYSTEM_PROMPT = (
+    "Write a description of a fantasy character based on the user prompt"
+    "Include only the following items; Race, height, build, eye color, hair color/cut, clothing, notable features"
+)
+BACK_SYSTEM_PROMPT = (
+    "Write a short synopsis of what the character has been through"
+    "Keep each part short and fitting for a fantasy world normal person"
+    "Include one line about what the character is thinking or doing right now."
+)
+
+## Need to change this to combine the prompt and instrctions into a list of functions for the model to run. 
+@spaces.GPU
 def char_description(prompt):
+    output1 = f"{DESC_SYSTEM_PROMPT}\n\nCharacter Description: {prompt}\n\nDescribe:"
+    output2 = "I cannot create " + prompt + " yet, because I do not have a brain."
+    return output1, output2
+
+def rand_char_description(prompt):
     output1 = "I see you asked for " + prompt + "!"
     output2 = "I cannot create " + prompt + " yet, because I do not have a brain."
     return output1, output2
 
+#if the prompt is blank, use random number to choose a race and class from lists. The lists will only 
+# include the most common of each. This will be fed in as the user prompt in leiu of a real one.
+
+# Function: combine prompt with a system prompt for the character description. 
+# "system prompt = (Describe the physical characteristics of the Character requested) user prompt = (user_prompt)"
+
+# Function: use the description and prompt to generate stats. May need a function per stat. Or one function we can call 
+# 6 times. Or we may leave this random. Maybe just have the model give a 1-10 rating of each skill based on the description
+# generated and then double it.
+
+# Function: use the description, prompt, and stats to generate backstory. 
+
+# Function: use python to get DC and random advantage checks from lists. 
+
+# change btn.click to map to those functions. 
+# Maybe wrap them all into a single function with if/then logic to get all the way through the task.
+
+# add disadvantage in bard stats?
+
+##UI is fine as-is
 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>")
+
     with gr.Row(equal_height = True):
         with gr.Column(scale = 2):
-            text_in = gr.Textbox(label="What npc do you need?")
+            text_in = gr.Textbox(label="What npc do you need?", placeholder = "An Orc Barkeep")
         with gr.Column(scale = 0):
             btn = gr.Button("Generate") 
             rand = gr.Button("Random")       
@@ -31,12 +76,14 @@ 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 = char_description,inputs = text_in, outputs = [text_out1, text_out2])
+            rand.click(fn = rand_char_description,inputs = text_in, outputs = [text_out1, text_out2])
 
 
         with gr.Column(scale = 0):
             ac = gr.Textbox(value = lambda:random.randint(1,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)
+        
     
     with gr.Row():
         
@@ -45,4 +92,7 @@ with gr.Blocks() as demo:
                 dcs = gr.Textbox(label = "DC to Smash")
                 prefs = gr.Textbox(label = "Advantage if:")
 
+
+# Need to add export feature.
+
 demo.launch(theme = gr.themes.Monochrome())

requirements.txt

@@ -1,3 +1,5 @@
 gradio
 transformers
-torch
+torch
+spaces
+accelerate