akhooli commited on
Commit
6d42934
·
verified ·
1 Parent(s): 4440f5f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -11,8 +11,15 @@ if os.path.exists(path_to_index):
11
  message = "index loaded!"
12
  import gradio as gr
13
 
14
- def check_fn():
15
- return message
 
 
 
 
 
 
 
16
 
17
  qapp = gr.Interface(fn=check_fn, inputs="textbox", outputs="textbox")
18
  if __name__ == "__main__":
 
11
  message = "index loaded!"
12
  import gradio as gr
13
 
14
+ def process_results(results):
15
+ for r in results:
16
+ print(f"Sura: {r['document_id']} ({r['document_metadata']}) \n Text:{r['content']}")
17
+
18
+ k = 3 # How many documents you want to retrieve
19
+
20
+ def answer_fn(query):
21
+ results = RAG.search(query= query)
22
+ return process_results(results)
23
 
24
  qapp = gr.Interface(fn=check_fn, inputs="textbox", outputs="textbox")
25
  if __name__ == "__main__":