Update app.py

app.py

@@ -24,6 +24,8 @@ agents =[
     "SOCIAL_MEDIA_MANAGER",
     "MEME_GENERATOR",
     "QUESTION_GENERATOR",
+    "AI_REPORT_WRITER",
+
 ]
 temperature=0.9
 max_new_tokens=256
@@ -62,8 +64,8 @@ def question_generate(prompt, history, agent_name=agents[0], sys_prompt="", temp
 def generate(prompt, history, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=1048, top_p=0.95, repetition_penalty=1.0,):
     seed = random.randint(1,1111111111111111)
 
-    agent=prompts.WEB_DEV
-    if agent_name == "WEB_DEV":
+    agent=prompts.AI_REPORT_WRITER
+'''    if agent_name == "WEB_DEV":
         agent = prompts.WEB_DEV_SYSTEM_PROMPT
     if agent_name == "CODE_REVIEW_ASSISTANT":
         agent = prompts.CODE_REVIEW_ASSISTANT
@@ -78,7 +80,7 @@ def generate(prompt, history, agent_name=agents[0], sys_prompt="", temperature=0
     if agent_name == "MEME_GENERATOR":
         agent = prompts.MEME_GENERATOR  
     if agent_name == "QUESTION_GENERATOR":
-        agent = prompts.QUESTION_GENERATOR      
+        agent = prompts.QUESTION_GENERATOR     ''' 
     system_prompt=agent
     temperature = float(temperature)
     if temperature < 1e-2: