Update app.py

app.py

@@ -16,9 +16,13 @@ def format_prompt(message, history):
   return prompt
 
 def generate(
-        prompt, history, system_prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0, agent=None,
+        prompt, history, system_prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0, agent_name=None,
 ):
-    system_prompt=prompts.agent
+    if agent_name = "WEB_DEV":
+        agent = prompt.WEB_DEV
+    if agent_name = "AI_SYSTEM_PROMPT":
+        agent = prompt.AI_SYSTEM_PROMPT
+    system_prompt=agent
     temperature = float(temperature)
     if temperature < 1e-2:
         temperature = 1e-2