Gradio version

Gradio working. Not all fields populate yet.

temp_prompts.txt

@@ -55,3 +55,35 @@ speech_traits = {
 "abbreviations": "Abbreviations refer to shortened forms like acronyms and initialisms. Writing with many abbreviations uses things like 'LOL' and 'IDK', while formal writing avoids abbreviations.",
 "grammar_spelling": "Grammar and spelling refer to the proper use of syntax, mechanics, and word forms. Strong grammar and spelling adheres to conventions, while poor grammar makes frequent mistakes."
 }
+speech_options = {
+"formality": ["Very informal","Informal","Neutral","Formal","Very formal"],
+"pace": ["Very slow","Slow","Moderate","Fast","Very fast"],
+"rhythm": ["Melodious","Flowing","Varied","Staccato","Choppy"],
+"volume": ["Very soft","Soft","Moderate","Loud","Very loud"],
+"emotionality": ["Very restrained","Restrained","Neutral","Expressive","Very expressive"],
+"directness": ["Very indirect","Indirect","Balanced","Direct","Very direct"],
+"humor": ["Frequently serious","Occasionally serious","Neutral","Occasionally humorous","Frequently humorous"],
+"enunciation": ["Mumbled","Relaxed","Neutral","Clear","Very clear"],
+"expressiveness": ["Very reserved","Reserved","Neutral","Expressive","Very expressive"],
+"accent": ["Strong foreign accent","Mild foreign accent","Neutral","Mild regional accent","Strong regional accent"],
+"politeness": ["Very blunt","Blunt","Neutral","Polite","Very polite"],
+"vocabulary": ["Very basic","Basic","Average","Sophisticated","Highly sophisticated"],
+"interruptions": ["Frequently allows others to interrupt","Occasionally allows others to interrupt","Balanced","Occasionally interrupts","Frequently interrupts"],
+"hesitations": ["Frequently fluent","Occasionally fluent","Balanced","Occasionally hesitates","Frequently hesitates"],
+"sentence_structure": ["Very simple","Simple","Average","Complex","Very complex"],
+"sarcasm": ["Never sarcastic","Rarely sarcastic","Occasionally sarcastic","Sarcastic","Very sarcastic"],
+"colloquialisms": ["Never uses colloquialisms","Rarely uses colloquialisms","Balanced","Occasionally uses colloquialisms","Frequently uses colloquialisms"],
+"energy_level": ["Very low energy","Low energy","Moderate energy","High energy","Very high energy"],
+"defiance/rebellion": ["Never defiant","Rarely defiant","Balanced","Occasionally defiant","Frequently defiant"],
+"playfulness": ["Never playful","Rarely playful","Occasionally playful","Playful","Very playful"],
+"vulgarity": ["Never vulgar","Rarely vulgar","Occasionally vulgar","Vulgar","Very vulgar"],
+"idiosyncrasies": ["No idiosyncrasies","Rare idiosyncrasies","Balanced","Occasional idiosyncrasies","Frequent idiosyncrasies"],
+"emotional_tone": ["Very pessimistic","Pessimistic","Neutral","Optimistic","Very optimistic"],
+"context_adaptability": ["Very inflexible","Inflexible","Balanced","Adaptable","Very adaptable"],
+"subtext": ["Never uses subtext","Rarely uses subtext","Balanced","Occasionally uses subtext","Frequently uses subtext"],
+"metaphorical_language": ["Never uses metaphorical language","Rarely uses metaphorical language","Balanced","Occasionally uses metaphorical language","Frequently uses metaphorical language"],
+"cultural_references": ["Never uses cultural references","Rarely uses cultural references","Balanced","Occasionally uses cultural references","Frequently uses cultural references"],
+"storytelling_ability": ["Never tells stories","Rarely tells stories","Balanced","Occasional storyteller","Frequent storyteller"],
+"abbreviations": ["Never","Rare","Occasional","Frequent","Very frequent"],
+"grammar_spelling": ["Very poor","Poor","Average","Good","Excellent"]
+}

test.py

@@ -1,6 +1,12 @@
 import requests
 import json
 import time
+from ast import Interactive
+import gradio as gr
+import re
+import codecs
+import random
+
 #Jup1t3R!
 
 ### Great results with lmsys_vicuna-13b-v1.1 8-bit>> ~425s
@@ -8,7 +14,7 @@ import time
 ### Good results with lmsys_vicuna-13b-v1.1 4-bit>> ~150s
 ### Good results with lmsys_vicuna-7b-v1.1 4-bit>> ~75s
 
-HOST = '0.0.0.0:5001'
+HOST = '127.0.0.1:5000'
 URI = f'http://{HOST}/api/v1/generate'
 #MPT
 # INSTRUCTION_KEY = "### Instruction:"
@@ -16,8 +22,9 @@ URI = f'http://{HOST}/api/v1/generate'
 #LLAMA
 INSTRUCTION_KEY = "### User:"
 RESPONSE_KEY = "### Assistant:"
+traits_dict = {}
 
-character = "Holden Caulfield"
+character = "Ender Wiggins"
 
 preprompt = f"""
 {INSTRUCTION_KEY}
@@ -26,11 +33,10 @@ You are a professional writer, novelist, and character creator. You are extremel
 
 postprompt = f"\n{RESPONSE_KEY}t\n"
 general = ""
-traits_dict = {}
 first_time = 0
 last_time = 0
 
-def run(prompt, toks, seed=-1):
+def run(prompt, toks=250, seed=-1):
     start_time = time.time()
     request = {
         'prompt': prompt,
@@ -74,10 +80,6 @@ def run(prompt, toks, seed=-1):
     total_time = round(end_time - start_time, 2)
     if response.status_code == 200:
         result = response.json()['results'][0]['text']
-        # return result
-        # print("-------------")
-        # print(f"{result} ({total_time}s)")
-        # print("-------------")
         print(f"({total_time}s)")
         return result
     else:
@@ -117,7 +119,8 @@ def getGeneralDescription(json, name):
 {json}
 Based on this information, use your experience as a writer to write a beautiful 5 sentence summary of {name}. Although inspired by a real character, create a fictitious version. \n{RESPONSE_KEY}: Sure! Here is a plain-text summary of a fictitious character based on the information you provided:"""
     print("Getting description.")
-    return run(preprompt + prompt, 250)
+    result = run(preprompt + prompt, 250)
+    return result.strip("`")
 
 def getName(character):
     prompt = f"{RESPONSE_KEY}Here is a list of names that fit with a fictitious character that would appear on their birth certificate without any additional context or information other than their name:\nA childrens book character of a brilliant scientist's name is: Professor Brains\nAbe Licoln's name is: Abraham Licoln\nA fun loving Teddy Bear's name is: Teddy (fuzzmuffin)_McFuzz\nHans from Star War's name is: Hans Solo\n{character}'s name is:"
@@ -202,7 +205,7 @@ def getAllSpeech(general, name):
 "metaphorical_language": ["Never uses metaphorical language","Rarely uses metaphorical language","Balanced","Occasionally uses metaphorical language","Frequently uses metaphorical language"],
 "cultural_references": ["Never uses cultural references","Rarely uses cultural references","Balanced","Occasionally uses cultural references","Frequently uses cultural references"],
 "storytelling_ability": ["Never tells stories","Rarely tells stories","Balanced","Occasional storyteller","Frequent storyteller"],
-"abbreviations": ["Never","Rare","Occasional","Frequent","Very frequent"],
+"abbreviations": ["Never","Rarely","Occasionally","Frequently","Very frequently"],
 "grammar_spelling": ["Very poor","Poor","Average","Good","Excellent"]
     }
     traits_dict['speech']={}
@@ -233,8 +236,9 @@ def getAllSpeech(general, name):
         if was_found==False:
             traits_dict['speech'][key] = "neutral"
             print(f">>> FAIL, setting to 'neutral'")
+    return traits_dict['speech']
 
-def unwrapText():
+def unwrapText(traits_dict):
         description_unwrapped = f"""
 {traits_dict['name']} is a {traits_dict['age']}-year-old {traits_dict['sexuality']} {traits_dict['sex']}.{traits_dict['general']} {traits_dict['name']} likes {traits_dict['likes']}. {traits_dict['name']} hates {traits_dict['hates']}.
             """
@@ -262,6 +266,8 @@ They have a {traits_dict['speech']['emotional_tone']} tone
 {"They " + traits_dict['speech']['metaphorical_language'] + "." if traits_dict['speech']['metaphorical_language']!="balanced" else ""}
 {"They " + traits_dict['speech']['cultural_references'] + "." if traits_dict['speech']['cultural_references']!="balanced" else ""}
 {"They " + traits_dict['speech']['storytelling_ability'] + "." if traits_dict['speech']['storytelling_ability']!="balanced" else ""}
+{"They write with " + traits_dict['speech']['grammar_spelling'] + " grammar and spelling."}
+{"They " + traits_dict['speech']['abbreviations'] + " use of abbreviations like 'lol' and 'brb'."}
      """
         traits_dict['speech_unwrapped'] = speech_unwrapped
         print(f"{speech_unwrapped} {description_unwrapped}")
@@ -294,6 +300,7 @@ Tell me your analysis in this exact format: MBTI - Enneagram - Instinctual Varia
 def test():
     prompt = f"""
     {INSTRUCTION_KEY}
+    {traits_dict['general']}
     {traits_dict['speech_unwrapped']}
     Use this information and your writing and creative experience to create a sample dialogue between {traits_dict['name']} and a fictional neutral character named John. {traits_dict['name']} must speak exactly as specified. Words chosen, topic discussed, grammar, sentence structure, and vocabulary should all be considered when shaping this conversation.
     {RESPONSE_KEY}
@@ -308,31 +315,218 @@ def test():
     """
     response = run(preprompt + prompt, 500)
     print(f"*2* {response} ***")
+
+# def go():
+#     first_time = time.time()
+#     cleaned_data = remove_after_first_brace(getTraits(character))
+#     traits_dict = json.loads(cleaned_data)
+#     for key, value in traits_dict.items():
+#         if isinstance(value, str):
+#             traits_dict[key] = value.lower()
+
+#     traits_dict['name'] = getName(character)
+#     traits_dict['sex'] = remove_non_letters(getSex(character))
+#     traits_dict['sexuality'] = remove_non_letters(getSexuality(character))
+#     traits_dict['age'] = remove_non_numbers(getAge(character))
+#     traits_dict['general'] = getGeneralDescription(traits_dict, traits_dict['name'])
+#     traits_dict['psych'] = getPsychProfile(traits_dict['general'])
+#     getAllSpeech(traits_dict['general'], traits_dict['name'])
+#     unwrapText(traits_dict)
+#     last_time = time.time()
+#     total_time = round(last_time - first_time, 2)
+#     print(f"Total time taken >> {total_time}")
+#     print(" ################# OUTPUT #################")
+#     print(json.dumps(traits_dict))
+#     print(" ################# END #################")
+#     # for key, value in traits_dict.items():
+#     #     print(f"{key}: {value}")
+#     test()
+
+with gr.Blocks() as demo:
+    rawQuickResults=gr.Textbox(visible=False)
+    gr.Markdown(
+        """
+        # Character Maker v0.2 #
+        """)
+
+    numTraits = 8
+    def updateTraitCount(text):
+        numTraits = text
+        print (numTraits)
+    def generateMedicalSecrets(count):
+        with open('medical_secrets.txt', 'r') as file:
+            lines = file.readlines()
+        selected_lines = random.sample(lines, count)
+        codes = [re.search(r'\((.*?)\)', line).group(1) for line in selected_lines]
+        codes_string = ', '.join(codes)
+        print(codes)
+        return "medical conditions: " + codes_string
+
+    with gr.Row():
+        quickStart = gr.Textbox(label="Quick Start", info="Use AI to fill out all fields based on your description.")
+        generate = gr.Button("Generate Quick Start")
+        
+    with gr.Row():
+        with gr.Column(scale=1, min_width=600):
+            trait_count = gr.Slider(4, 20, label="Minimum number of traits for each category (this is currently ignored)", interactive=False, step=1.0, value=8)
+            trait_count.change(updateTraitCount, inputs=[trait_count])
+            charName = gr.Textbox(label="Name", interactive=True)
+            personality = gr.Textbox(label="Personality", interactive=True)
+            body = gr.Textbox(label="Physical Description", interactive=True)
+            with gr.Row():
+                likes = gr.Textbox(label="Likes", interactive=True)
+                hates = gr.Textbox(label="Hates", interactive=True)
+            with gr.Row():
+                sex = gr.Dropdown(["Male", "Female", "Other"], label="Sex", interactive=True)
+                sexuality = gr.Textbox(label="Sexuality", interactive=True)
+                age = gr.Slider(1, 100, label="Age", info="Choose between 1 and 100", interactive=True, step=1.0, value=21)
+            description = gr.Textbox(label="Description", interactive=True) 
+            attributes = gr.Textbox(label="Attributes", interactive=True)
+            with gr.Row():
+                personalityProfile = gr.Textbox(label="Psychological Profile", interactive=True) 
+                generatePersonality = gr.Button("2. Analyze Personality")
+                  
+    with gr.Accordion("Speaking Traits"):
+        speechOverview = gr.TextArea(label="Speech style", interactive=True)
+        with gr.Column(scale=1, min_width=600):
+            with gr.Row():
+                formality = gr.Dropdown(["Very formal", "Formal", "Neutral", "Informal", "Very informal"], label="Formality", interactive=True)
+                pace = gr.Dropdown(["Very fast", "Fast", "Moderate", "Slow", "Very slow"], label="Pace", interactive=True)
+                rhythm = gr.Dropdown(["Choppy", "Staccato", "Varied", "Flowing", "Melodious"], label="Rhythm", interactive=True)
+                volume = gr.Dropdown(["Very loud", "Loud", "Moderate", "Soft", "Very soft"], label="Volume", interactive=True)
+                emotionality = gr.Dropdown(["Very expressive", "Expressive", "Neutral", "Restrained", "Very restrained"], label="Emotionality", interactive=True)
+                directness = gr.Dropdown(["Very direct", "Direct", "Balanced", "Indirect", "Very indirect"], label="Directness", interactive=True)
+ 
+            with gr.Row():
+                humor = gr.Dropdown(["Frequently humorous", "Occasionally humorous", "Neutral", "Occasionally serious", "Frequently serious"], label="Humor", interactive=True)
+                enunciation = gr.Dropdown(["Very clear", "Clear", "Neutral", "Relaxed", "Mumbled"], label="Enunciation", interactive=True)
+                expressiveness = gr.Dropdown(["Very expressive", "Expressive", "Neutral", "Reserved", "Very reserved"], label="Expressiveness", interactive=True)
+                accent_dialect = gr.Dropdown(["Strong regional accent", "Mild regional accent", "Neutral", "Mild foreign accent", "Strong foreign accent"], label="Accent/Dialect", interactive=True)
+                politeness = gr.Dropdown(["Very polite", "Polite", "Neutral", "Blunt", "Very blunt"], label="Politeness", interactive=True)
+                vocabulary = gr.Dropdown(["Highly sophisticated", "Sophisticated", "Average", "Basic", "Very basic"], label="Vocabulary", interactive=True)
+
+            with gr.Row():     
+                interruptions = gr.Dropdown(["Frequently interrupts", "Occasionally interrupts", "Balanced", "Occasionally allows others to interrupt", "Frequently allows others to interrupt"], label="Interruptions", interactive=True)
+                hesitations = gr.Dropdown(["Frequently hesitates", "Occasionally hesitates", "Balanced", "Occasionally fluent", "Frequently fluent"], label="Hesitations", interactive=True)
+                sentence_structure = gr.Dropdown(["Very complex", "Complex", "Average", "Simple", "Very simple"], label="Sentence Structure", interactive=True)
+                sarcasm = gr.Dropdown(["Very sarcastic", "Sarcastic", "Occasionally sarcastic", "Rarely sarcastic", "Never sarcastic"], label="Sarcasm", interactive=True)
+                colloquialisms = gr.Dropdown(["Frequently uses colloquialisms", "Occasionally uses colloquialisms", "Balanced", "Rarely uses colloquialisms", "Never uses colloquialisms"], label="Colloquialisms", interactive=True)
+                energy_level = gr.Dropdown(["Very high energy", "High energy", "Moderate energy", "Low energy", "Very low energy"], label="Energy Level", interactive=True)
+            with gr.Row():
+                defiance_rebellion = gr.Dropdown(["Frequently defiant", "Occasionally defiant", "Balanced", "Rarely defiant", "Never defiant"], label="Defiance/Rebellion", interactive=True)
+                playfulness = gr.Dropdown(["Very playful", "Playful", "Occasionally playful", "Rarely playful", "Never playful"], label="Playfulness", interactive=True)
+                vulgarity = gr.Dropdown(["Very vulgar", "Vulgar", "Occasionally vulgar", "Rarely vulgar", "Never vulgar"], label="Vulgarity", interactive=True)
+                idiosyncrasies = gr.Dropdown(["Frequent idiosyncrasies", "Occasional idiosyncrasies", "Balanced", "Rare idiosyncrasies", "No idiosyncrasies"], label="Idiosyncrasies", interactive=True)
+                emotional_tone = gr.Dropdown(["Very optimistic", "Optimistic", "Neutral", "Pessimistic", "Very pessimistic"], label="Emotional Tone", interactive=True)
+                context_adaptability = gr.Dropdown(["Very adaptable", "Adaptable", "Balanced", "Inflexible", "Very inflexible"], label="Context Adaptability", interactive=True)
+            with gr.Row():
+                subtext = gr.Dropdown(["Frequently uses subtext", "Occasionally uses subtext", "Balanced", "Rarely uses subtext", "Never uses subtext"], label="Subtext", interactive=True)
+                metaphorical_language = gr.Dropdown(["Frequently uses metaphorical language", "Occasionally uses metaphorical language", "Balanced", "Rarely uses metaphorical language", "Never uses metaphorical language"], label="Metaphorical Language", interactive=True)
+                cultural_references = gr.Dropdown(["Frequently uses cultural references", "Occasionally uses cultural references", "Balanced", "Rarely uses cultural references", "Never uses cultural references"], label="Cultural References", interactive=True)
+                storytelling_ability = gr.Dropdown(["Frequent storyteller", "Occasional storyteller", "Balanced", "Rarely tells stories", "Never tells stories"], label="Storytelling Ability", interactive=True)
+    with gr.Column(scale=1, min_width=600):
+        gr.Markdown("""**SECRETS**\n
+                        Medical conditions : This will choose a random medical condition(s) that this character suffers from.
+                        Conditions will be saved only as a medical diagnosis code to save on token space and so it can be a secret to YOU as well. 
+                    """)
+        with gr.Row():
+                conditions_count = gr.Slider(1, 10, label="Number of conditions", info="Choose between 1 and 10", interactive=True, step=1.0, value=1)
+                med_secret_button = gr.Button("5. Generate secret medical condition(s) (COMING SOON)", interactive=True)
+                med_secret = gr.Textbox(label="Secrets", interactive=True, type='password')      
+        gr.Markdown("""Trauma : This will choose a random psychological event that affects how this character behaves.
+                    Example: Experiencing discrimination
+                    """)
+        with gr.Row():
+                secrets_count = gr.Slider(1, 10, label="Number of issues", info="Choose between 1 and 10", interactive=True, step=1.0)
+                secret_button = gr.Button("6. Generate secret issues(s) (COMING SOON)", interactive=False)
+                secret = gr.Textbox(label="Secrets", interactive=True, type='password')            
+        situation = gr.TextArea(label="Situation (COMING SOON)", interactive=False)
+        starting_message = gr.TextArea(label="Starting message (COMING SOON)", interactive=False)
+        # sample_starters = gr.CheckboxGroup(["This year is flying by so fast.",
+        #                     "It's quite sunny outside today.",
+        #                     "People seem to be in a hurry all the time.",
+        #                     "I hear birds chirping. It must be spring already.",
+        #                     "The city looks different at night."], 
+        #                     label="Example conversation starters", info="These are simple statements designed to evoke a unique response without adding additional context. ", interactive=True),
+        with gr.Row():
+            with gr.Column(scale=1):
+                examples = gr.TextArea(label="Example chats (COMING SOON)", value="<START>\n")
+
+    def genMedical(text):
+        return "none"
+    
+    med_secret_button.click(generateMedicalSecrets, inputs=[conditions_count], outputs=[med_secret])
+    createJSON = gr.Button("4. Create JSON")
+    quickStartResult = gr.JSON(label="result", interactive=True)    
+
+    def updateFields(text):
+        print("updating fields")
+        charName = text.get('name', '')
+        personality = text.get('personality', '')
+        body = text.get('body', '')
+        likes = text.get('likes', '')
+        hates = text.get('hates', '')
+        sex = text.get('sex', '')
+        sexuality = text.get('sexuality', '')
+        age = text.get('age', '')
+        description = text.get('general', '')
+        personalityProfile = text.get('psych', '')
+        attributes = text.get('attributes', '')
+        speechOverview = text.get('speech_unwrapped', '')
+        return [
+            charName,
+            personality,
+            body,
+            likes,
+            hates,
+            sex,
+            sexuality,
+            age,
+            description,
+            personalityProfile,
+            attributes,
+            speechOverview
+        ]
+    
+    quickStartResult.change(updateFields, inputs=[quickStartResult], outputs=[
+        charName, 
+        personality,
+        body,
+        likes,
+        hates,
+        sex,
+        sexuality,
+        age,
+        description,
+        personalityProfile,
+        attributes,
+        speechOverview
+    ])
+
+    def generateQuickStart(text):
+        first_time = time.time() 
+        cleaned_data = remove_after_first_brace(getTraits(text))
+        traits_dict = json.loads(cleaned_data)
+        for key, value in traits_dict.items():
+            if isinstance(value, str):
+                traits_dict[key] = value.lower()
+        traits_dict['name'] = getName(text)
+        traits_dict['sex'] = remove_non_letters(getSex(text))
+        traits_dict['sexuality'] = remove_non_letters(getSexuality(text))
+        traits_dict['age'] = remove_non_numbers(getAge(text))
+        traits_dict['general'] = getGeneralDescription(traits_dict, traits_dict['name'])
+        traits_dict['psych'] = getPsychProfile(traits_dict['general'])
+        traits_dict['speech'] = getAllSpeech(traits_dict['general'], traits_dict['name'])
+        unwrapText(traits_dict)
+        last_time = time.time()
+        total_time = round(last_time - first_time, 2)
+        print(f"completed in {total_time} s")
+        return traits_dict
+    
+    generate.click(generateQuickStart, inputs=[quickStart], outputs=[quickStartResult])
+    
+inst = demo.launch()
     
-if __name__ == '__main__':
-    first_time = time.time()
-    cleaned_data = remove_after_first_brace(getTraits(character))
-    traits_dict = json.loads(cleaned_data)
-    for key, value in traits_dict.items():
-        if isinstance(value, str):
-            traits_dict[key] = value.lower()
-
-    # To check the updated dictionary:
-    traits_dict['name'] = getName(character)
-    traits_dict['sex'] = remove_non_letters(getSex(character))
-    traits_dict['sexuality'] = remove_non_letters(getSexuality(character))
-    traits_dict['age'] = remove_non_numbers(getAge(character))
-    traits_dict['general'] = getGeneralDescription(traits_dict, traits_dict['name'])
-    traits_dict['psych'] = getPsychProfile(traits_dict['general'])
-    getAllSpeech(traits_dict['general'], traits_dict['name'])
-    unwrapText()
-    last_time = time.time()
-    total_time = round(last_time - first_time, 2)
-    print(f"Total time taken >> {total_time}")
-    print(" ################# OUTPUT #################")
-    for key, value in traits_dict.items():
-        print(f"{key}: {value}")
-    test()