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)