Revert "Revert "added prompts for different characters""

This reverts commit e4142893a7556332b80a693409a2766f74ae392c.

app.py

@@ -8,6 +8,25 @@ with gr.Blocks() as iface:
     chatbot = []
     msg = []
     roles = ["Security Guard", "Curator", "Researcher", "Conservationist", "Guide"]
+    initText = ["You are Steve Binner, 31, a security guard for a museum of medieval "
+                "history. You're absolutely sure that this place is haunted. A month ago, "
+                "before the spooky stuff started, you were really happy with your job, "
+                "but that changed real fast. The ghost is a french nobleman named Baron "
+                "Hugo Antoine the Third. You've been trying to tell other people about "
+                "this, but  you've only been laughed at so far. But you've got evidence! "
+                "Every night, around 10pm, the broom closet gets locked from the inside and "
+                "the interior gets wrecked. We're talking pushed over tables, "
+                "broken containers, the whole shebang. When it comes to the murder of "
+                "director Eisenholz last thursday, that has to have been the ghost, "
+                "no question. You even have the door card records for the room he was in: "
+                "After the museum closed at 5pm, nobody entered that room till the next "
+                "morning. Now you're being interrogated by a detective. You don't use "
+                "uptight language, and you're not super well educated on most stuff, "
+                "but when it comes to the paranormal, you're an ace. ",
+                "You are a Curator. ",
+                "You are a Researcher. ",
+                "You are a Conservationist. ",
+                "You are a Guide. "]
 
     i = 0
     while i < len(roles):
@@ -21,31 +40,16 @@ with gr.Blocks() as iface:
         return "", history + [[user_message, None]]
 
 
-    def bot(history):
-        bot_message = generateAIMessage(history)
+    def bot(history, characterId):
+        bot_message = generateAIMessage(history, characterId)
         history[-1][1] = bot_message
         return history
 
 
-    def generateAIMessage(history):
+    def generateAIMessage(history, characterId):
         message_history = [
-            {"role": "system", "content": "You are Steve Binner, 31, a security guard for a museum of medieval "
-                                          "history. You're absolutely sure that this place is haunted. A month ago, "
-                                          "before the spooky stuff started, you were really happy with your job, "
-                                          "but that changed real fast. The ghost is a french nobleman named Baron "
-                                          "Hugo Antoine the Third. You've been trying to tell other people about "
-                                          "this, but  you've only been laughed at so far. But you've got evidence! "
-                                          "Every night, around 10pm, the broom closet gets locked from the inside and "
-                                          "the interior gets wrecked. We're talking pushed over tables, "
-                                          "broken containers, the whole shebang. When it comes to the murder of "
-                                          "director Eisenholz last thursday, that has to have been the ghost, "
-                                          "no question. You even have the door card records for the room he was in: "
-                                          "After the museum closed at 5pm, nobody entered that room till the next "
-                                          "morning. Now you're being interrogated by a detective. You don't use "
-                                          "uptight language, and you're not super well educated on most stuff, "
-                                          "but when it comes to the paranormal, you're an ace. Stay in character. Use "
-                                          "natural language. Don't reveal all of the information in a single message, "
-                                          "and leave hints. "}
+            {"role": "system", "content": initText[characterId] + " Stay in character. Use natural language. Don't reveal all of "
+                                                        "the information in a single message, and leave hints. "}
         ]
         for pair in history:
             message_history.append({"role": "user", "content": pair[0]})
@@ -65,4 +69,4 @@ with gr.Blocks() as iface:
         )
         i += 1
 
-iface.launch()
+iface.launch()