broadfield commited on
Commit
b70ffa6
·
verified ·
1 Parent(s): 0c58e85

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -73,10 +73,10 @@ def generate(prompt,history,mod=2,tok=4000,seed=1,role="ASSISTANT",data=None):
73
  system_prompt = prompts.MANAGER.replace("**TIMELINE**",data[4]).replace("**HISTORY**",str(history))
74
  formatted_prompt = format_prompt(prompt, mod, system_prompt)
75
  elif role == "PATHMAKER":
76
- system_prompt = prompts.PATH_MAKER.replace("**FILE_LIST**",file_list).replace("**CURRENT_OR_NONE**",str(data[4])).replace("**PROMPT**",json.dumps(data[0],indent=4)).replace("**HISTORY**",str(history))
77
  formatted_prompt = format_prompt(prompt, mod, system_prompt)
78
  elif role == "CREATE_FILE":
79
- system_prompt = prompts.CREATE_FILE.replace("**FILE_LIST**",file_list).replace("**TIMELINE**",data[4]).replace("**FILENAME**",str(data[1])).replace("**TEMPLATE_OR_NONE**",str(data[2]))
80
  formatted_prompt = format_prompt(prompt, mod, system_prompt)
81
  elif role == "SEARCH":
82
  system_prompt = prompts.SEARCH.replace("**DATA**",data)
 
73
  system_prompt = prompts.MANAGER.replace("**TIMELINE**",data[4]).replace("**HISTORY**",str(history))
74
  formatted_prompt = format_prompt(prompt, mod, system_prompt)
75
  elif role == "PATHMAKER":
76
+ system_prompt = prompts.PATH_MAKER.replace("**FILE_LIST**",str(file_list)).replace("**CURRENT_OR_NONE**",str(data[4])).replace("**PROMPT**",json.dumps(data[0],indent=4)).replace("**HISTORY**",str(history))
77
  formatted_prompt = format_prompt(prompt, mod, system_prompt)
78
  elif role == "CREATE_FILE":
79
+ system_prompt = prompts.CREATE_FILE.replace("**FILE_LIST**",str(file_list)).replace("**TIMELINE**",data[4]).replace("**FILENAME**",str(data[1])).replace("**TEMPLATE_OR_NONE**",str(data[2]))
80
  formatted_prompt = format_prompt(prompt, mod, system_prompt)
81
  elif role == "SEARCH":
82
  system_prompt = prompts.SEARCH.replace("**DATA**",data)