Update app.py

app.py

@@ -72,20 +72,21 @@ import os
 
 
 # Function to get bot response
-def format_prompt(state, bot_index, character_name, character_description):
+def format_prompt(state, bot_index, character_name, character_description, user_name):
     if character_name is None or character_name.strip() == "":
         character_name = "Ryan"
     if character_description is None or character_description.strip() == "":
         character_description = "Ryan is a college student who is always willing to help. He knows a lot about math and coding."
+    if user_name is None or user_name.strip() == "":
+        user_name = "You"
 
-    prompt = f"{character_description}\n\n"
+    prompt = f"<|im_start|>system {character_description}<|im_end|>\n"
     for message in state["history"][bot_index]:
         if message['role'] == 'user':
-            prompt += f"You: {message['content']}\n"
+            prompt += f"<|im_start|>user {user_name}: {message['content']}<|im_end|>\n"
         else:
-            prompt += f"{character_name}: {message['content']}\n"
-    prompt += f"{character_name}: "
-    print(prompt)
+            prompt += f"<|im_start|>assistant {character_name}: {message['content']}<|im_end|>\n"
+    prompt += f"<|im_start|>assistant {character_name}:"
     return prompt
 import aiohttp
 import asyncio