Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -59,13 +59,13 @@ def generate(prompt,history,mod=2,tok=4000,seed=1,role="ASSISTANT",data=None):
|
|
59 |
else:
|
60 |
lod=[]
|
61 |
if role == "MANAGER":
|
62 |
-
system_prompt = prompts.MANAGER.replace("**TIMELINE**",data[4]).replace("**HISTORY**",str(history))
|
63 |
formatted_prompt = format_prompt(prompt, mod, system_prompt)
|
64 |
elif role == "PATHMAKER":
|
65 |
-
system_prompt = prompts.PATH_MAKER.replace("**FILE_LIST**",str(data[3])).replace("**CURRENT_OR_NONE**",str(data[4])).replace("**PROMPT**",json.dumps(data[0],indent=4)).replace("**HISTORY**",str(history))
|
66 |
formatted_prompt = format_prompt(prompt, mod, system_prompt)
|
67 |
elif role == "CREATE_FILE":
|
68 |
-
system_prompt = prompts.CREATE_FILE.replace("**FILE_LIST**",str(data[3])).replace("**TIMELINE**",data[4]).replace("**FILENAME**",str(data[1])).replace("**TEMPLATE_OR_NONE**",str(data[2]))
|
69 |
formatted_prompt = format_prompt(prompt, mod, system_prompt)
|
70 |
elif role == "SEARCH":
|
71 |
system_prompt = prompts.SEARCH.replace("**DATA**",data)
|
|
|
59 |
else:
|
60 |
lod=[]
|
61 |
if role == "MANAGER":
|
62 |
+
system_prompt = prompts.PREFIX + prompts.MANAGER.replace("**TIMELINE**",data[4]).replace("**HISTORY**",str(history))
|
63 |
formatted_prompt = format_prompt(prompt, mod, system_prompt)
|
64 |
elif role == "PATHMAKER":
|
65 |
+
system_prompt = prompts.PREFIX + prompts.PATH_MAKER.replace("**FILE_LIST**",str(data[3])).replace("**CURRENT_OR_NONE**",str(data[4])).replace("**PROMPT**",json.dumps(data[0],indent=4)).replace("**HISTORY**",str(history))
|
66 |
formatted_prompt = format_prompt(prompt, mod, system_prompt)
|
67 |
elif role == "CREATE_FILE":
|
68 |
+
system_prompt = prompts.PREFIX + prompts.CREATE_FILE.replace("**FILE_LIST**",str(data[3])).replace("**TIMELINE**",data[4]).replace("**FILENAME**",str(data[1])).replace("**TEMPLATE_OR_NONE**",str(data[2]))
|
69 |
formatted_prompt = format_prompt(prompt, mod, system_prompt)
|
70 |
elif role == "SEARCH":
|
71 |
system_prompt = prompts.SEARCH.replace("**DATA**",data)
|