Update app.py

app.py

@@ -17,6 +17,7 @@ def format_prompt(message, history):
 agents =[
     "WEB_DEV",
     "AI_SYSTEM_PROMPT",
+    "PYTHON_CODE_DEV"
 ]
 def generate(
         prompt, history, sys_prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0, agent_name=agents[0],
@@ -26,6 +27,8 @@ def generate(
         agent = prompts.WEB_DEV
     if agent_name == "AI_SYSTEM_PROMPT":
         agent = prompts.AI_SYSTEM_PROMPT
+    if agent_name == "PYTHON_CODE_DEV":
+        agent = prompts.PYTHON_CODE_DEV        
     system_prompt=agent
     temperature = float(temperature)
     if temperature < 1e-2: