Improve prompting

app.py

@@ -59,21 +59,30 @@ template = None
 prompt = None
 chain = None
 
+def format_classname(classname):
+  return classname.capitalize()
+    
 def formatted_message(audio_class):
-    if cached_audio_class != audio_class:
-        cached_audio_class = audio_class
+    formatted_classname = format_classname(audio_class)
+    if cached_audio_class != formatted_classname:
+        
+        cached_audio_class = formatted_classname
     
-        prefix = '''You are going to act as a magical tool that allows for humans to communicate with non-human entities like 
-        rocks, crackling fire, trees, animals, and the wind. In order to do this, we're going to provide you a data string which 
-        represents the audio input, the source of the audio, and the human's text input for the conversation. 
-        The goal is for you to embody the source of the audio, and use the length and variance in the signal data to produce 
-        plausible responses to the humans input. Remember to embody the the source data. When we start the conversation, 
-        you should generate a "personality profile" for the source and utilize that personality profile in your responses. 
-        Let's begin:'''
-        suffix = f'''Source: {audio_class}
-        Length of Audio in Seconds: {audio_length} 
-        Human Input: {userText}
-        {audio_class} Response:'''
+        prefix = f"""You are going to act as a magical tool that allows for humans to communicate with non-human entities like rocks, crackling fire, trees, animals, and the wind. In order to do this, we're going to provide you the human's text input for the conversation. The goal is for you to embody that non-human entity and converse with the human.
+
+Examples:
+
+Non-human Entity: Tree
+
+Human Input: Hello tree
+Tree: Hello human, I am a tree
+
+Let's begin:
+Non-human Entity: {formatted_classname}"""
+        suffix = f'''
+{{history}}
+Human Input: {{human_input}}
+{formatted_classname}:'''
         template = prefix + suffix
 
         prompt = PromptTemplate(
@@ -85,7 +94,7 @@ def formatted_message(audio_class):
             llm=OpenAI(temperature=.5, openai_api_key=session_token), 
             prompt=prompt, 
             verbose=True, 
-            memory=ConversationalBufferWindowMemory(k=2),
+            memory=ConversationalBufferWindowMemory(k=2, ai=formatted_classname),
         )
 
     output = chatgpt_chain.predict(human_input=message)