Update app.py

app.py

@@ -56,15 +56,15 @@ def generate(prompt,history,mod=2,tok=4000,seed=1,role="ASSISTANT",data=None):
         word_dict.close()    
     else:
         lod=[]
-    if role == "ASSISTANT":
-        system_prompt = prompts.ASSISTANT
-        formatted_prompt = format_prompt(f'USER:{prompt}', mod, system_prompt)
+    if role == "MANAGER":
+        system_prompt = prompts.MANAGER.replace("**TIMELINE**",data[4]).replace("**HISTORY**",str(history))
+        formatted_prompt = format_prompt(prompt, mod, system_prompt)
+    elif role == "PATHMAKER":
+        system_prompt = prompts.PATH_MAKER.replace("**CURRENT_OR_NONE**",str(data[4])).replace("**CODE_FILES**",json.dumps(data[0],indent=4)).replace("**HISTORY**",str(history))
+        formatted_prompt = format_prompt(prompt, history, system_prompt)
     elif role == "CREATE_FILE":
         system_prompt = prompts.CREATE_FILE.replace("**FILENAME**",str(data[4]))
         formatted_prompt = format_prompt(prompt, mod, system_prompt)
-    elif role == "MANAGER":
-        system_prompt = prompts.MANAGER.replace("**TIMELINE**",data[4]).replace("**HISTORY**",str(history))
-        formatted_prompt = format_prompt(prompt, mod, system_prompt)
     elif role == "SEARCH":
         system_prompt = prompts.SEARCH.replace("**DATA**",data)
         formatted_prompt = format_prompt(f'USER:{prompt}', mod, system_prompt)
@@ -89,7 +89,8 @@ def generate(prompt,history,mod=2,tok=4000,seed=1,role="ASSISTANT",data=None):
         yield output
         yield history
         yield prompt
-    elif role=="ASSISTANT" or role=="WEBDEVELOPER" or role=="PATHMAKER":
+
+    elif role=="PATHMAKER":
         print("Runnning ", role)
         stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=True)
         #prompt=f"We just completed role:{role}, now choose the next tool to complete the task:{prompt}, or COMPLETE"
@@ -99,6 +100,7 @@ def generate(prompt,history,mod=2,tok=4000,seed=1,role="ASSISTANT",data=None):
         yield output
         yield history
         yield prompt
+        
     elif role=="CREATE_FILE":
         print("Running Create File")
         stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=True)