reranker button added

app.py

@@ -34,7 +34,7 @@ def add_text(history, text):
     return history, gr.Textbox(value="", interactive=False)
 
 
-def bot(history, api_kind):
+def bot(history, api_kind, use_reranker):
     query = history[-1][0]
 
     if not query:
@@ -46,13 +46,14 @@ def bot(history, api_kind):
 
     documents = retrieve(query, 10)
 
-    document_time = perf_counter() - document_start
-    logger.info(f'Finished Retrieving documents in {round(document_time, 2)} seconds...')
+    if use_reranker == "Yes":
+        document_time = perf_counter() - document_start
+        logger.info(f'Finished Retrieving documents in {round(document_time, 2)} seconds...')
 
-    rerank_start = perf_counter()
-    documents = rerank(query, documents, TOP_K)
-    rerank_time = perf_counter() - rerank_start
-    logger.info(f'Finished Reranking documents in {round(rerank_time, 2)} seconds...')
+        rerank_start = perf_counter()
+        documents = rerank(query, documents, TOP_K)
+        rerank_time = perf_counter() - rerank_start
+        logger.info(f'Finished Reranking documents in {round(rerank_time, 2)} seconds...')
 
     # Create Prompt
     prompt = template.render(documents=documents, query=query)
@@ -92,6 +93,7 @@ with gr.Blocks() as demo:
         txt_btn = gr.Button(value="Submit text", scale=1)
 
     api_kind = gr.Radio(choices=["HuggingFace", "OpenAI"], value="HuggingFace")
+    use_reranker = gr.Radio(label="Use reranker?", choices=["Yes", "No"], value="Yes")
 
     prompt_html = gr.HTML()
     # Turn off interactivity while generating if you click