Update app.py

app.py

@@ -49,12 +49,7 @@ def format_prompt(message, mod, system):
     return prompt
 def generate(prompt,history,mod=2,tok=4000,seed=1,role="ASSISTANT",data=None):
     #print("#####",history,"######")
-    try:
-        file_list = fs.ls(f'spaces/{user_}{repo_}',detail=False)
-    except Exception as e:
-        print(e)
-        file_list=["NO FILES YET"]
-    print('file list\n',file_list)
+
     gen_images=False
     client=InferenceClient(clients[int(mod)]['name'])
     client_tok=clients[int(mod)]['max_tokens']
@@ -72,10 +67,10 @@ def generate(prompt,history,mod=2,tok=4000,seed=1,role="ASSISTANT",data=None):
         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("**FILE_LIST**",str(file_list)).replace("**CURRENT_OR_NONE**",str(data[4])).replace("**PROMPT**",json.dumps(data[0],indent=4)).replace("**HISTORY**",str(history))
+        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))
         formatted_prompt = format_prompt(prompt, mod, system_prompt)
     elif role == "CREATE_FILE":
-        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]))
+        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]))
         formatted_prompt = format_prompt(prompt, mod, system_prompt)
     elif role == "SEARCH":
         system_prompt = prompts.SEARCH.replace("**DATA**",data)
@@ -231,7 +226,12 @@ def agent(prompt_in,history,mod=2):
     MAX_DATA=int(clients[int(mod)]['max_tokens'])*2
     if not history:history=[{'role':'user','content':prompt_in['text']}]
     while go == True:
-
+        try:
+            file_list = fs.ls(f'spaces/{user_}{repo_}',detail=False)
+        except Exception as e:
+            print(e)
+            file_list=["NO FILES YET"]
+        print('file list\n',file_list)
 
         
         seed = random.randint(1,9999999999999)
@@ -241,7 +241,9 @@ def agent(prompt_in,history,mod=2):
             history = [history[-2:]]
         print('history',history)
         role="PATHMAKER"
+        in_data[3]=file_list
         outph= list(generate(prompt,history,mod,2400,seed,role,in_data))[0]
+        
         in_data[4]=outph
         print(outph)
         history.extend([{'role':'assistant','content':str(outph)}])
@@ -269,6 +271,7 @@ def agent(prompt_in,history,mod=2):
                     in_data[1]=com
                     temp1=file_template(com)
                     in_data[2]=temp1
+                    in_data[3]=file_list
                     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])