Update app.py

app.py

@@ -16,7 +16,10 @@ def format_prompt(message, history):
 agents =[
     "WEB_DEV",
     "AI_SYSTEM_PROMPT",
-    "PYTHON_CODE_DEV"
+    "PYTHON_CODE_DEV",
+    "CODE_REVIEW_ASSISTANT ",
+    "CONTENT_WRITER_EDITOR ",
+    "SOCIAL_MEDIA_MANAGER ",
 ]
 def generate(
         prompt, history, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
@@ -26,6 +29,12 @@ def generate(
     agent=prompts.WEB_DEV
     if agent_name == "WEB_DEV":
         agent = prompts.WEB_DEV_SYSTEM_PROMPT
+    if agent_name == "CODE_REVIEW_ASSISTANT":
+        agent = prompts.CODE_REVIEW_ASSISTANT
+    if agent_name == "CONTENT_WRITER_EDITOR":
+        agent = prompts.CONTENT_WRITER_EDITOR
+    if agent_name == "SOCIAL_MEDIA_MANAGER":
+        agent = prompts.SOCIAL_MEDIA_MANAGER        
     if agent_name == "AI_SYSTEM_PROMPT":
         agent = prompts.AI_SYSTEM_PROMPT
     if agent_name == "PYTHON_CODE_DEV":
@@ -108,14 +117,17 @@ additional_inputs=[
 
 ]
 
-examples=[["Write a simple working game in HTML5", agents[1], None, None, None, None, ],
+examples=[["Write a simple working game in HTML5", agents[0], None, None, None, None, ],
+          ["Choose 3 useful types of AI agents, and create a detailed System Prompt to align each of them.", agents[1], None, None, None, None, ],
+          ["Explain it to me in a childrens story how Nuclear Fission works", agents[4], None, None, None, None, ],
+          ["Show a bunch of examples of catchy ways to post, 'I had a ham sandwich for lunch today'", agents[5], None, None, None, None, ],
           ["Write high quality personal website to show off my adventure sports hobby", agents[0], None, None, None, None, ],
-          ["I'm planning a vacation to Japan. Can you suggest a one-week itinerary including must-visit places and local cuisines to try?", None, None, None, None, None, ],
-          ["Can you write a short story about a time-traveling detective who solves historical mysteries?", None, None, None, None, None,],
-          ["I'm trying to learn French. Can you provide some common phrases that would be useful for a beginner, along with their pronunciations?", None, None, None, None, None,],
-          ["I have chicken, rice, and bell peppers in my kitchen. Can you suggest an easy recipe I can make with these ingredients?", None, None, None, None, None,],
-          ["Can you explain how the QuickSort algorithm works and provide a Python implementation?", None, None, None, None, None,],
-          ["What are some unique features of Rust that make it stand out compared to other systems programming languages like C++?", None, None, None, None, None,],
+          ["I'm planning a vacation to Japan. Can you suggest a one-week itinerary including must-visit places and local cuisines to try?", agents[4], None, None, None, None, ],
+          ["Can you write a short story about a time-traveling detective who solves historical mysteries?", agents[4], None, None, None, None,],
+          ["I'm trying to learn French. Can you provide some common phrases that would be useful for a beginner, along with their pronunciations?", agents[4], None, None, None, None,],
+          ["I have chicken, rice, and bell peppers in my kitchen. Can you suggest an easy recipe I can make with these ingredients?", agents[4], None, None, None, None,],
+          ["Can you explain how the QuickSort algorithm works and provide a Python implementation?", agents[2], None, None, None, None,],
+          ["What are some unique features of Rust that make it stand out compared to other systems programming languages like C++?", agents[3], None, None, None, None,],
          ]