Update app.py

app.py

@@ -34,7 +34,22 @@ def sort_fn(inp):
     		f"{inp}",	# str  in 'Paste Text' Textbox component
     		api_name="/sort_doc"
     )
-    return nouns
+    return sen, nouns
+
+
+
+def find_fn(inp,sen,nouns)
+    client = Client("Omnibus/sort_document")
+    result = client.predict(
+    		inp,	# str  in 'Search query' Textbox component
+    		sen,	# Dict[Any, Any] (any valid json) in 'Sentences' Json component
+    		nouns,	# Dict[Any, Any] (any valid json) in 'Nouns' Json component
+    		api_name="/find_query_sen"
+    )
+    return result
+
+
+
 
 def find_all(url):
     return_list=[]
@@ -462,11 +477,20 @@ def summarize(inp,history,report_check,chart_check,data=None,files=None,director
                 c +=1
         print (f'c:: {c}')
 
-        json_out = sort_fn(out)
-    yield "", history,error_box,json_out,json_out
+        sen_out,nouns_out = sort_fn(out)
+    yield "", history,error_box,nouns_out,nouns_out,sen_out
+
 
+def search_fn(inp,history,report_check,chart_check,data):
+    data=find_fn(inp,)
+    error_box=""
+    if inp == "":
+        inp = "Process this data"
+    history.clear()
+    history = [(inp,"Working on it...")] 
+    yield "",history,error_box,json_box,json_box
 
-def search_fn(prompt,history,report_check,chart_check,data):
+    
     if data != "Error" and data != "":
         print(inp)
         out = str(data)
@@ -479,7 +503,7 @@ def search_fn(prompt,history,report_check,chart_check,data):
         print (f'c:: {c}')
 
         #json_start = sort_fn(out)
-        json_out = find_keyword_fn(c,prompt,data)
+        json_out = find_keyword_fn(c,inp,data)
         print(f'JSON_BOX:: {json_out}')
         out = str(json_out)
         if report_check:
@@ -626,7 +650,9 @@ def clear_fn():
     return "",[(None,None)]
 
 with gr.Blocks() as app:
+    found_state=gr.State()
     doc_state=gr.State()
+    noun_state=gr.State()
     gr.HTML("""<center><h1>Mixtral 8x7B TLDR Summarizer + Web</h1><h3>Summarize Data of unlimited length</h3>""")
     chatbot = gr.Chatbot(label="Mixtral 8x7B Chatbot",show_copy_button=True)
     with gr.Row():
@@ -662,8 +688,9 @@ with gr.Blocks() as app:
     clear_btn.click(clear_fn,None,[prompt,chatbot])
     
     #go=button.click(summarize,[prompt,chatbot,report_check,chart_check,data,file,directory,url,pdf_url,pdf_batch],[prompt,chatbot,e_box,json_out])
-    find=button.click(search_fn,[prompt,chatbot,report_check,chart_check,doc_state],[prompt,chatbot,e_box,json_out])
-    go=load_button.click(summarize,[prompt,chatbot,report_check,chart_check,data,file,directory,url,pdf_url,pdf_batch],[prompt,chatbot,e_box,json_out,doc_state])
+    
+    find=button.click(find_fn,[inp,doc_state,noun_state],found_state).then(search_fn,[prompt,chatbot,report_check,chart_check,found_state],[prompt,chatbot,e_box,json_out])
+    go=load_button.click(summarize,[prompt,chatbot,report_check,chart_check,data,file,directory,url,pdf_url,pdf_batch],[prompt,chatbot,e_box,json_out,noun_state,doc_state])
     #find=button.click()
     stop_button.click(None,None,None,cancels=[go])
 app.queue(default_concurrency_limit=20).launch(show_api=False)