Update app.py

app.py

@@ -318,33 +318,26 @@ bm25_retriever = BM25Retriever(index_dir="output/bm25_index_b")
 
 corpus_dict = {doc.collection_id: doc.text for doc in sciq.corpus}
 
-# Search function for the BM25 system
-def search(query: str) -> List[Hit]:
-    # Replace the following placeholder with actual retrieval logic
+def get_query(query):
     results = bm25_retriever.retrieve(query)
     hits = [
         {
             "cid": cid,
             "score": score,
-            "text": corpus_dict[cid]  # Assuming sciq.corpus maps cids to document texts
+            "text": corpus_dict[cid]
         }
         for cid, score in results.items()
     ]
     return hits
 
-def handle_search(query):
-        results = search(query)
-        return results
-        
-    
+
+
 demo = gr.Interface(
-    fn=handle_search,  # The function to process input
-    inputs=gr.Textbox(label="Enter your search query"),  # Input: Textbox
-    outputs=gr.Textbox(label="Search Results", lines=20, interactive=False),  # Output: Textbox
-    title="BM25 Search Engine Demo on SciQ Dataset",  # Title of the app
-    description="Enter your search query to get the results from the SciQ dataset."  # Description
+    fn=get_query,
+    inputs=gr.Textbox(label="Enter your query"),
+    outputs=gr.Textbox(label="Results", lines=20, interactive=False),
+    title="BM25 Query Engine"
 )
-
-# Launch the app with shareable URL
-demo.launch(debug=True)
+## YOUR_CODE_ENDS_HERE
+demo.launch()