Update app.py

app.py

@@ -16,9 +16,9 @@ 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,
+        prompt, history, system_prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0, agent=None,
 ):
-    system_prompt=prompts.WEB_DEV
+    system_prompt=prompts.{agent}
     temperature = float(temperature)
     if temperature < 1e-2:
         temperature = 1e-2
@@ -41,7 +41,10 @@ def generate(
         output += response.token.text
         yield output
     return output
-
+agents =[
+    "WEB_DEV",
+    "AI_SYSTEM_PROMPT",
+]
 
 additional_inputs=[
     gr.Textbox(
@@ -85,12 +88,11 @@ additional_inputs=[
         step=0.05,
         interactive=True,
         info="Penalize repeated tokens",
-    )
-]
-agents =[
-    "WEB_DEV",
-    "AI_SYSTEM_PROMPT",
+    ),
+    gr.Dropdown(choices=[s for s in agents]),
+
 ]
+
 examples=[["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,],
@@ -108,6 +110,5 @@ with gr.Blocks() as iface:
         examples=examples,
         concurrency_limit=20,
     )
-    ag = gr.Dropdown(choices=[s for s in agents])
 iface.launch(show_api=False)