Update app.py

app.py

@@ -10,6 +10,7 @@ import datetime
 import xmltodict
 
 from prompts import (
+    GET_KEYWORD,
     COMPRESS_HISTORY_PROMPT,
     COMPRESS_DATA_PROMPT,
     COMPRESS_DATA_PROMPT_SMALL,
@@ -141,8 +142,36 @@ def compress_data(c,purpose, task, history):
 
 
 
+def get_key(inp,data):
+    key_box=[]
+    seed=random.randint(1,1000000000)
+    key_w = run_gpt(
+            GET_KEYWORD,
+            stop_tokens=[],
+            max_tokens=56,
+            seed=seed,
+            purpose=inp,
+            task=inp,
+        )
+    print(f'key_w::{key_w}')
+    if " " in key_w:
+        key_w=key_w.split(" ")[-1]
+    for i,ea in enumerate(data):
+        try:
+            if ea['title'] and key_w in ea['title']:
+                key_box.append(ea)
+            elif ea['description'] and key_w in ea['description']:
+                key_box.append(ea)
+            elif ea['link'] and key_w in ea['link']:
+                key_box.append(ea)
+        except Exception as e:
+            print(e)
+    print(key_box)
+
+    
 def summarize(inp,history,data=None):
     
+    
     json_box=[]
     if inp == "":
         inp = "Process this data"
@@ -305,6 +334,7 @@ def find_rss():
     )
     yield out_box,[(None,f'Source is current as of:\n{timestamp} UTC\n\nThe current Date and Time is:\n{timestamp} UTC')]
 
+    
 def load_data():
     yield None,[(None,f'Loading data source, please wait')]
     
@@ -330,10 +360,12 @@ with gr.Blocks() as app:
         sub_btn=gr.Button("Submit")
     with gr.Row():
         load_btn = gr.Button("Load RSS")
-    u_btn=gr.Button("Update [RSS Data]")
+        u_btn=gr.Button("Update [RSS Data]")
+        keyw = gr.Button("Use Keyword [Experimental]")
     with gr.Row():
         out_json = gr.JSON()
         fil = gr.Textbox()
+    keyw.click(get_key,[inst,out_json],None)
     load_btn.click(load_data,None,[out_json,cb])
     u_btn.click(find_rss,None,[out_json,cb])
     sub_btn.click(summarize,[inst,cb,out_json],[inst,cb])