Update test.py

+ psycho profile
+ test function
+ better speech profile

test.py

@@ -3,6 +3,8 @@ import json
 import time
 #Jup1t3R!
 
+### Great results with lmsys_vicuna-13b-v1.1 8-bit>> ~425s
+### Great results with lmsys_vicuna-7b-v1.1 16-bit>> ~90s
 ### Good results with lmsys_vicuna-13b-v1.1 4-bit>> ~150s
 ### Good results with lmsys_vicuna-7b-v1.1 4-bit>> ~75s
 
@@ -15,6 +17,8 @@ URI = f'http://{HOST}/api/v1/generate'
 INSTRUCTION_KEY = "### User:"
 RESPONSE_KEY = "### Assistant:"
 
+character = "Holden Caulfield"
+
 preprompt = f"""
 {INSTRUCTION_KEY}
 You are a professional writer, novelist, and character creator. You are extremely knowledgeable in the art of character creation. You should provide the details of a character as requested by the User. Only output the exact answer. When given choices you must pick one and only one. You should not add additional context. You should be concise. You should follow the exact format requested. Do not add any additional information or context. This a fictional character, totally imagined and created by you. It is not a real person.
@@ -228,16 +232,14 @@ def getAllSpeech(general, name):
                 break
         if was_found==False:
             traits_dict['speech'][key] = "neutral"
-            print(f"!!!!!!!! not found in the text: {response}")
-            print(f"!!!!!!!! from the prompt: {prompt}")
+            print(f">>> FAIL, setting to 'neutral'")
 
 def unwrapText():
-        # 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']}.
-        #     """
+        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']}.
+            """
+        traits_dict['description'] = description_unwrapped.strip('\n')
+
         speech_unwrapped = f"""
 {traits_dict['name']} speaks with a unique style:
 They {"are " + traits_dict['speech']['formality'] + " and" if traits_dict['speech']['formality']!="neutral" else ""} speak at a {traits_dict['speech']['pace']} speed with a {traits_dict['speech']['rhythm']} rhythm.
@@ -250,8 +252,7 @@ They have a {traits_dict['speech']['accent']} accent.
 {traits_dict['name']} {"is " + traits_dict['speech']['politeness'] + " and " if traits_dict['speech']['politeness']!="neutral" else ""}uses a {traits_dict['speech']['vocabulary']} vocabulary.
 {"They " + traits_dict['speech']['interruptions'] + "." if traits_dict['speech']['interruptions']!="balanced" else ""}
 {"They " + traits_dict['speech']['hesitations'] + "." if traits_dict['speech']['interruptions']!="balanced" else ""}
-{traits_dict['name']} uses a {traits_dict['speech']['sentence_structure']} sentence structure and is {traits_dict['speech']['sarcasm']}
-{"They " + traits_dict['speech']['colloquialisms'] + "." if traits_dict['speech']['colloquialisms']!="balanced" else ""}
+{traits_dict['name']} uses a {traits_dict['speech']['sentence_structure']} sentence structure and is {traits_dict['speech']['sarcasm']}{" They " + traits_dict['speech']['colloquialisms'] + "." if traits_dict['speech']['colloquialisms']!="balanced" else ""}
 They speak with {traits_dict['speech']['energy_level']}{" and is " + traits_dict['speech']['defiance/rebellion'] + "." if traits_dict['speech']['defiance/rebellion']!="balanced" else "."}
 When {traits_dict['name']} speaks it is {traits_dict['speech']['playfulness']} and {traits_dict['speech']['vulgarity']}.
 {traits_dict['name'] + " uses " + traits_dict['speech']['idiosyncrasies'] + "." if traits_dict['speech']['idiosyncrasies']!="balanced" else ""}
@@ -262,15 +263,54 @@ They have a {traits_dict['speech']['emotional_tone']} tone
 {"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 ""}
      """
-        print(speech_unwrapped)
+        traits_dict['speech_unwrapped'] = speech_unwrapped
+        print(f"{speech_unwrapped} {description_unwrapped}")
+
+        prompt = f"""
+{INSTRUCTION_KEY} Please take this description of a character's speaking pattern, clean it up, and make sure the description is clear: {speech_unwrapped}
+{RESPONSE_KEY} Sure! Here is the revised description:
+"""
+        print(f"Correcting speech profile.")
+        response = run(prompt, 250)
+        traits_dict['speech_unwrapped'] = response
 
+def getPsychProfile(general):
+    prompt = f"""
+{preprompt}
+{general}
+Based on your understanding of this character, analyze their psychological profile using the standard profiles:
+Myers-Briggs Type Indicator (MBTI): Introverted/Extraverted, Sensing/Intuitive, Thinking/Feeling, Judging/Perceiving.
+Enneagram: main type, wing, and instinctual variant (self-preservation/sp, social/so, or sexual/sx).
+Enneagram Tritype: two additional types from different centers of intelligence.
+Socionics: similar to MBTI, identify the character's personality type.
+Global 5/SLOAN: Reserved/Social, Calm/Limbic, Organized/Unstructured, Accommodating/Egocentric, Non-Curious/Inquisitive.
+Tell me your analysis in this exact format: MBTI - Enneagram - Instinctual Variant - Tritype - Socionics - SLOAN. For example: INTJ - 3w4 - sp/sx - 358 - LIE - RCOEI. Do not explain or add other text or context.
+{RESPONSE_KEY} Sure! Based on the description of this character, and without any explanation, their psychological profile is 
+"""
+    print(f"Getting psych profile.")
+    response = run(prompt, 120)
+    return response.strip('"')
+
+def test():
+    prompt = f"""
+    {INSTRUCTION_KEY}
+    {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}
+    """
+    response = run(prompt, 350)
+    print(f"*1* {response} ***")
+    preprompt = prompt + response
+    prompt = f"""
+    {INSTRUCTION_KEY}
+    Using the speaking style described, revise that conversation to best match how {traits_dict['name']} would really speak
+    {RESPONSE_KEY}
+    """
+    response = run(preprompt + prompt, 500)
+    print(f"*2* {response} ***")
+    
 if __name__ == '__main__':
-    ##TEST
-    # getAllSpeech("Roger Rabbit is an energetic, funny, clever, adventurous, and optimistic rabbit. He hates being ignored or laughed at and dislikes sadness and injustice. Roger enjoys listening to jazz music, eating carrots, playing the piano, and solving puzzles. His personality is charming, witty, and charismatic with his short, round body covered in red fur and large ears. As a male rabbit, he is heterosexual and ten years old.","Roger Rabbit")
-    # print(traits_dict)
-    ##END TEST
     first_time = time.time()
-    character = "a comedian based on George Carlin named Bob"
     cleaned_data = remove_after_first_brace(getTraits(character))
     traits_dict = json.loads(cleaned_data)
     for key, value in traits_dict.items():
@@ -283,17 +323,16 @@ if __name__ == '__main__':
     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'])
-    # print(general)
+    traits_dict['psych'] = getPsychProfile(traits_dict['general'])
     getAllSpeech(traits_dict['general'], traits_dict['name'])
     unwrapText()
-    # print(getSpeech(traits_dict))
-    # traits_dict['description'] = (getGeneralDescription(character, traits_dict))
     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()