Update app.py

app.py

@@ -9,7 +9,7 @@ from huggingface_hub import InferenceClient,HfApi
 import random
 import json
 import datetime
-#from query import tasks
+import uuid
 from prompts import (
     FINDER,
     SAVE_MEMORY,
@@ -21,8 +21,10 @@ from prompts import (
     PREFIX,
     TASK_PROMPT,
 )
-api=HfApi()
-
+reponame="Omnibus/tmp"
+save_data=f'https://huggingface.co/datasets/{reponame}/raw/main/'
+token_self = os.environ['HF_TOKEN']
+api=HfApi(token=token_self)
 
 
 client = InferenceClient(
@@ -168,6 +170,7 @@ def compress_data(c,purpose, task, history, result):
 
     
 def save_memory(purpose, history):
+    uid=uuid.uuid4()
     history=str(history)
     c=0
     inp = str(history)
@@ -217,25 +220,31 @@ def save_memory(purpose, history):
         e=e+chunk
         s=s+chunk
     print ("final1" + resp)
-    
-    #lines = resp.strip().strip("\n").split("\n")
-    skills=[]
-    #skill_dict={}
-    #for i,line in enumerate(lines):
-    #    print(f'Line:: {line}')
-    #skills.append(resp)
-    #print ("final2" + resp)
     try:
         resp=resp.split('</s>')[0]
         print ("final2" + resp)
     except Exception as e:
         resp = resp
         print(e)
-
-    skills.append(resp)
-    out=skills
+    timestamp=str(datetime.datetime.now())
+    timename=timestamp.replace(" ","--").replace(":","-").replace(".","-")
+    json_object=[resp]
+    #json_object = json.dumps(out_box)
+    #json_object = json.dumps(out_box,indent=4)
+    with open("tmp1.json", "w") as outfile:
+        outfile.write(json_object)
+    api.upload_file(
+    path_or_fileobj=f"tmp-{uid}.json",
+    path_in_repo=f"/mem-test/{timename}.json",
+    repo_id=reponame,
+    #repo_id=save_data.split('datasets/',1)[1].split('/raw',1)[0],
+    token=token_self,
+    repo_type="dataset",
+    )
+    #skills.append(resp)
+    #out=skills
     #history = "result: {}\n".format(resp)
-    return out
+    return [resp]
 
 
 
@@ -488,7 +497,7 @@ with gr.Blocks() as app:
     #go=button.click(check_rand,[rand,seed],seed).then(run,[opt,chatbot,tokens,char_stats,seed],[chatbot,char_stats,json_out,opt])
     go=button.click(check_rand,[rand,seed],seed).then(run,[opt,chatbot],[chatbot])
     stop_button.click(None,None,None,cancels=[go])
-app.launch(show_api=False) 
+app.queue(default_concurrency_limit=20).launch(show_api=False)