Update app.py

app.py

@@ -12,6 +12,7 @@ import datetime
 #from query import tasks
 from prompts import (
     FINDER,
+    SAVE_MEMORY,
     COMPRESS_HISTORY_PROMPT,
     COMPRESS_DATA_PROMPT,
     COMPRESS_DATA_PROMPT_SMALL,
@@ -164,6 +165,71 @@ def compress_data(c,purpose, task, history, result):
     history = "result: {}\n".format(resp)
     return history
 
+    
+def save_memory(purpose, history):
+    history=str(history)
+    c=0
+    out = str(out)
+    rl = len(out)
+    print(f'rl:: {rl}')
+    for i in str(out):
+        if i == " " or i=="," or i=="\n" or i=="/" or i=="." or i=="<":
+            c +=1
+    print (f'c:: {c}')
+
+    seed=random.randint(1,1000000000)
+    
+    print (c)
+    #tot=len(purpose)
+    #print(tot)
+    divr=int(c)/MAX_DATA
+    divi=int(divr)+1 if divr != int(divr) else int(divr)
+    chunk = int(int(c)/divr)
+    print(f'chunk:: {chunk}')
+    print(f'divr:: {divr}')
+    print (f'divi:: {divi}')
+    out = []
+    #out=""
+    s=0
+    e=chunk
+    print(f'e:: {e}')
+    new_history=""
+    task = f'Compile this data to fulfill the task: {task}, and complete the purpose: {purpose}\n'
+    for z in range(divi):
+        print(f's:e :: {s}:{e}')
+        
+        hist = history[s:e]
+        
+        resp = run_gpt(
+            SAVE_MEMORY,
+            stop_tokens=["observation:", "task:", "action:", "thought:"],
+            max_tokens=2048,
+            seed=seed,
+            purpose=purpose,
+            task=task,
+            knowledge=new_history,
+            history=hist,
+        ).strip('\n')
+        new_history = resp
+        print (resp)
+        out+=resp
+        e=e+chunk
+        s=s+chunk
+    '''
+    resp = run_gpt(
+        COMPRESS_DATA_PROMPT,
+        stop_tokens=["observation:", "task:", "action:", "thought:"],
+        max_tokens=2048,
+        seed=seed,
+        purpose=purpose,
+        task=task,
+        knowledge=new_history,
+        history=result,
+    )
+    '''
+    print ("final" + resp)
+    #history = "result: {}\n".format(resp)
+    return history
 
 
 
@@ -392,6 +458,7 @@ with gr.Blocks() as app:
                         opt=gr.Dropdown(label="Choices",choices=examples,allow_custom_value=True, value="Start a new game", interactive=True)
                         #prompt=gr.Textbox(label = "Prompt", value="Start a new game")
                     with gr.Column(scale=2):
+                        
                         rand = gr.Checkbox(label="Random", value=True)
                         seed=gr.Slider(label="Seed", minimum=1, maximum=99999999999, value=rand_val)
                     #models_dd=gr.Dropdown(choices=[m for m in return_list],interactive=True)
@@ -402,11 +469,15 @@ with gr.Blocks() as app:
                 with gr.Row():
                     tokens = gr.Slider(label="Max new tokens",value=2096,minimum=0,maximum=1048*10,step=64,interactive=False, visible=False,info="The maximum numbers of new tokens")
             with gr.Column(scale=1):
+                save_btn=gr.Button("Save Memory")
+                snap_btn=gr.Button("Take Screenshot")
                 char_stats=gr.Textbox()
-                json_out=gr.JSON()
+        json_out=gr.JSON()
     #text=gr.JSON()
     #inp_query.change(search_models,inp_query,models_dd)
     #test_b=test_btn.click(itt,url,e_box)
+
+    save_btn.click(save_memory,[opt,chatbot],json_out)
     clear_btn.click(clear_fn,None,[opt,chatbot])
     #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])