Update app.py

app.py

@@ -106,7 +106,7 @@ def Bm25(raw_text_list,input_text,n) :
   top_5_results = bm25.get_top_n(tokenized_query, corpus, n=n)
   results = '\n'.join(top_5_results)
   combined_input = "query = " + input_text + "\n\n For the above query these are some results from a search engine: \n ".join(results)  + "\n\n Give detailed and brief answer for the query write in 500-1000 words. Give detailed and well informative answer(include calculations if needed, using tables and other styles of structuring is optional for better answering ) "
-  return combined_input
+  return combined_input,results
 
 def llm_route(llm):
     openai_api_key = "EMPTY"
@@ -190,7 +190,7 @@ def function2(input_text,one_shot_example,llm):
 def function3(input_text,one_shot_example,n,llm):
   n=int(n)
   k=search_results(input_text)
-  k=Bm25(k,input_text,n)
+  k,results=Bm25(k,input_text,n)
   if llm=="GPT-4":
       model = AzureChatOpenAI(
           openai_api_base=BASE_URL,
@@ -243,7 +243,7 @@ def function3(input_text,one_shot_example,n,llm):
                 stop=None,  # Optional: specify a stop sequence
                 temperature=0.7  # Adjust the creativity of the response
             )
-        return completion.choices[0].text.strip()
+        return completion.choices[0].text.strip(),results
       
       
 
@@ -256,6 +256,7 @@ def function4(input_text, one_shot_example, n,llm):
     k=""
     for doc in sorted_docs[:n]:
         k+=doc
+    results=k
     if llm=="GPT-4":
         model = AzureChatOpenAI(
           openai_api_base=BASE_URL,
@@ -307,7 +308,7 @@ def function4(input_text, one_shot_example, n,llm):
                 stop=None,  # Optional: specify a stop sequence
                 temperature=0.7  # Adjust the creativity of the response
             )
-            return completion.choices[0].text.strip()
+            return completion.choices[0].text.strip(),results
         
         
 
@@ -339,7 +340,7 @@ iface3 = gr.Interface(
         gr.Dropdown(choices=dropdown_options, label="Number of top search results"),
         gr.Dropdown(choices=llm_dropdown, label="LLM")
     ], 
-    outputs="text"
+    outputs={"LLM Answer": "text", "Google Search Result": "text"}
 )
 iface4 = gr.Interface(
     function4, 
@@ -349,7 +350,7 @@ iface4 = gr.Interface(
         gr.Dropdown(choices=dropdown_options_4, label="Number of top k documents"),
         gr.Dropdown(choices=llm_dropdown, label="LLM")
     ], 
-    outputs="text"
+    outputs={"LLM Answer": "text", "Abstract Search Result": "text"}
 )
 
 # Create a parallel interface that combines all individual interfaces