Update app.py

app.py

@@ -16,7 +16,8 @@ from prompts import (
     PREFIX,
 )
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
-save_data='https://huggingface.co/datasets/Omnibus/tmp/raw/main/'
+reponame="Omnibus/tmp"
+save_data=f'https://huggingface.co/datasets/{reponame}/raw/main/'
 token_self = os.environ['HF_TOKEN']
 api=HfApi(token=token_self)
 
@@ -188,14 +189,14 @@ def summarize(inp,history,data=None):
             yield "", [(inp,f'{mes}\n{new_history}')]
             resp = run_gpt(
                 COMPRESS_DATA_PROMPT,
-                stop_tokens=["observation:", "task:", "action:", "thought:"],
+                stop_tokens=[],
                 max_tokens=2048,
                 seed=seed,
                 purpose=purpose,
                 task=task,
                 knowledge=new_history,
                 history=hist,
-            ).strip("\n")
+            )
             new_history = resp
             print (resp)
             out+=resp
@@ -204,7 +205,7 @@ def summarize(inp,history,data=None):
             #history = "preliminary result: {}\n".format(resp)
             #yield "", (inp,f'{mes}\n{history}')
         print ("final" + resp)
-        out_hist = "result: {}\n".format(resp)
+        out_hist = "result:\n{}".format(resp)
         #return history
         yield "", [(inp,out_hist)]
     
@@ -297,26 +298,32 @@ def find_rss():
     api.upload_file(
     path_or_fileobj="tmp1.json",
     path_in_repo=f"/rss/{timename}.json",
-    repo_id=save_data.split('datasets/',1)[1].split('/raw',1)[0],
+    repo_id=reponame,
+    #repo_id=save_data.split('datasets/',1)[1].split('/raw',1)[0],
     token=token_self,
     repo_type="dataset",
     )
     yield out_box,[(None,'Sources are loaded.  You can ask a question about them now.')]
 
+def load_data():
+    f_ist = (api.list_repo_files(repo_id=reponame, repo_type="dataset"))
+    f_ist.sort(reverse=True)
+    print(f_ist)
+    r = requests.get(f'{save_data}/{f_ist[0]}') 
+    lod = json.loads(r.text)
+    return lod
 with gr.Blocks() as app:
     cb = gr.Chatbot(height=500)
     with gr.Row():
-        rss_search = gr.Textbox(label="Instructions")
-        search_btn=gr.Button("Send")
+        inst = gr.Textbox(label="Instructions")
+        sub_btn=gr.Button("Submit")
     with gr.Row():
-        rss = gr.Textbox(label="rss")
-        btn = gr.Button("load rss")
-    r_btn=gr.Button("read")
+        load_btn = gr.Button("Load RSS")
+    u_btn=gr.Button("Update [RSS Data]")
     with gr.Row():
         out_json = gr.JSON()
         fil = gr.Textbox()
-    r_btn.click(find_rss,None,[out_json,cb])
-    #r_btn.click(read_rss,None,[out_json,fil])
-    search_btn.click(summarize,[rss_search,cb,out_json],[rss_search,cb])
-    #btn.click(get_rss,rss,out_json)
+    load_btn.click(load_data,None,out_json)
+    u_btn.click(find_rss,None,[out_json,cb])
+    sub_btn.click(summarize,[inst,cb,out_json],[inst,cb])
 app.launch()