Update app.py

app.py

@@ -280,7 +280,35 @@ def compress_data_og(c, instruct, history):
     return resp
 
 
+RECALL_MEMORY="""
+The user will give you a query, and you will pick keywords from the provided KEYWORD LIST that will be used to search for information about the query.
+Your response will be only a list of keywords.
+Example Response:
+[keyword1,keyword2,keyword3]
 
+USER QUERY:
+{prompt}
+
+KEYWORD LIST:
+{keywords}
+"""
+
+
+
+
+def get_mem(prompt,kw):
+    seed=random.randint(1,1000000000)
+    resp = run_gpt(
+        RECALL_MEMORY.format(keywords=kw,prompt=prompt),
+        stop_tokens=[],
+        max_tokens=1024,
+        seed=seed,
+    )
+        
+    print (resp)
+    return resp 
+
+    
 def summarize(inp,history,report_check,sum_check,mem_check,data=None,files=None,url=None,pdf_url=None,pdf_batch=None):
     json_box=[]
     if inp == "":
@@ -643,8 +671,24 @@ def save_memory(purpose, history):
         out_box.append(resp)
         create_index()
     return out_box
-
-
+def recall_memory(inp,history)
+    error_box=""
+    json_out={}
+    r = requests.get(f'{save_data}mem-test2/index.json') 
+    print(f'status code main:: {r.status_code}')
+    if r.status_code==200:
+        mem = json.loads(r.text)
+        print (f'ind::\n{mem}')
+    if not r.status_code==200:
+        print("Create new IND")
+        out="MEMORY FILE NOT FOUND"
+        return out,out,out,out
+    mem_keys = mem.keys()
+    rawp = get_mem(inp,mem_keys)
+        
+    history.clear()
+    history.append((inp,rawp))
+    yield "", history,error_box,json_out
 
 
     
@@ -680,8 +724,13 @@ with gr.Blocks() as app:
             pdf_url = gr.Textbox(label="PDF URL")       
         with gr.Tab("PDF Batch"):
             pdf_batch = gr.Textbox(label="PDF URL Batch (comma separated)")
+        with gr.Tab("Memory"):
+            mem_inp = gr.Textbox(label="Query")
+            mem = gr.Button()
     json_out=gr.JSON()
     e_box=gr.Textbox()
+
+    mem.click(recall_memory,mem_inp,[prompt,chatbot,e_box,json_out])
     #text=gr.JSON()
     #inp_query.change(search_models,inp_query,models_dd)
     clear_btn.click(clear_fn,None,[prompt,chatbot])