Update app.py

app.py

@@ -28,7 +28,11 @@ clients = [
     {'type':'text','name':'meta-llama/Llama-3.3-70B-Instruct','rank':'pro','max_tokens':16384,'schema':{'bos':'<|im_start|>','eos':'<|im_end|>'}},
     {'type':'text','name':'mistralai/Mixtral-8x7B-Instruct-v0.1','rank':'op','max_tokens':40000,'schema':{'bos':'<s>','eos':'</s>'}},
 ]
-
+def file_template(inp):
+    if "readme.md" in inp.lower():
+        template=prompts.README
+    else:template="NONE"
+    return template
 def format_prompt(message, mod, system):
     eos=f"{clients[int(mod)]['schema']['eos']}\n"
     bos=f"{clients[int(mod)]['schema']['bos']}\n"
@@ -63,7 +67,7 @@ def generate(prompt,history,mod=2,tok=4000,seed=1,role="ASSISTANT",data=None):
         system_prompt = prompts.PATH_MAKER.replace("**CURRENT_OR_NONE**",str(data[4])).replace("**PROMPT**",json.dumps(data[0],indent=4)).replace("**HISTORY**",str(history))
         formatted_prompt = format_prompt(prompt, mod, system_prompt)
     elif role == "CREATE_FILE":
-        system_prompt = prompts.CREATE_FILE.replace("**TIMELINE**",data[4]).replace("**FILENAME**",str(data[1]))
+        system_prompt = prompts.CREATE_FILE.replace("**TIMELINE**",data[4]).replace("**FILENAME**",str(data[1])).replace("**TEMPLATE_OR_NONE**",str(data[2]))
         formatted_prompt = format_prompt(prompt, mod, system_prompt)
     elif role == "SEARCH":
         system_prompt = prompts.SEARCH.replace("**DATA**",data)
@@ -252,6 +256,8 @@ def agent(prompt_in,history,mod=2):
                 if 'CREATE_FILE' in fn:
                     print('CREATE_FILE called')
                     in_data[1]=com
+                    temp1=file_template(com)
+                    in_data[2]
                     out_o =generate(prompt,history,mod=mod,tok=10000,seed=seed,role="CREATE_FILE",data=in_data)
                     out_w=list(out_o)
                     ret1,ret2 = parse_json(out_w[2].split('<|im_end|>')[0])