Update app.py

app.py

@@ -1,4 +1,4 @@
-from huggingface_hub import InferenceClient, HfApi
+from huggingface_hub import InferenceClient, HfApi, list_files_info
 #from html2image import Html2Image
 import gradio as gr
 #import markdown
@@ -15,7 +15,8 @@ import re
 import os
 loc_folder="chat_history"
 loc_file="chat_json"
-
+user_="community-pool/"
+repo_="test2"
 clients = [
     {'type':'image','name':'black-forest-labs/FLUX.1-dev','rank':'op','max_tokens':16384,'schema':{'bos':'<|im_start|>','eos':'<|im_end|>'}},
     {'type':'text','name':'deepseek-ai/DeepSeek-V2.5-1210','rank':'op','max_tokens':16384,'schema':{'bos':'<|im_start|>','eos':'<|im_end|>'}},
@@ -47,6 +48,8 @@ def format_prompt(message, mod, system):
     return prompt
 def generate(prompt,history,mod=2,tok=4000,seed=1,role="ASSISTANT",data=None):
     #print("#####",history,"######")
+    file_list = list_files_info(f'{user_}{repo_}')
+    print('file list\n',file_list)
     gen_images=False
     client=InferenceClient(clients[int(mod)]['name'])
     client_tok=clients[int(mod)]['max_tokens']
@@ -64,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("**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**",file_list).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])).replace("**TEMPLATE_OR_NONE**",str(data[2]))
+        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]))
         formatted_prompt = format_prompt(prompt, mod, system_prompt)
     elif role == "SEARCH":
         system_prompt = prompts.SEARCH.replace("**DATA**",data)
@@ -174,7 +177,7 @@ def parse_json(inp):
         return out1,out2
 def build_space(repo_name,file_name,file_content,access_token=""):
     try:
-        repo_path="community-pool/"+str(repo_name)
+        repo_path=user_+str(repo_name)
         if not access_token:access_token=os.environ['HF_TOKEN']
 
         api=HfApi(endpoint="https://huggingface.co", token=access_token)
@@ -223,7 +226,6 @@ 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:
-        
         seed = random.randint(1,9999999999999)
         c=0
         #history = [history[-4:]]
@@ -263,7 +265,7 @@ def agent(prompt_in,history,mod=2):
                     ret1,ret2 = parse_json(out_w[2].split('<|im_end|>')[0])
                     print('ret1',ret1)
                     print('ret2',ret2)
-                    build_out = build_space('test1',ret1,ret2)
+                    build_out = build_space(repo_,ret1,ret2)
                     history+=[{'role':'system','content':f'observation:{build_out}'}]
                     yield history
                 elif 'IMAGE' in fn: