Update app.py

app.py

@@ -243,7 +243,7 @@ class BaseInvertedIndexRetriever(BaseRetriever):
     def index_class(self) -> Type[InvertedIndex]:
         pass
 
-    def _init_(self, index_dir: str) -> None:
+    def __init__(self, index_dir: str) -> None:
         self.index = self.index_class.from_saved(index_dir)
 
     def get_term_weights(self, query: str, cid: str) -> Dict[str, float]:
@@ -307,38 +307,44 @@ demo: Optional[gr.Interface] = None  # Assign your gradio demo to this variable
 return_type = List[Hit]
 
 ## YOUR_CODE_STARTS_HERE
-# Building BM25 index and save:
 bm25_index = BM25Index.build_from_documents(
-    documents=iter(sciq.corpus),
-    ndocs=12160,
-    show_progress_bar=True
-)
-bm25_index.save("output/bm25_index")
-
-# Loading index and use BM25 retriever to retrieve:
-bm25_retriever = BM25Retriever(index_dir="output/bm25_index")
+        documents=iter(sciq.corpus),
+        ndocs=len(sciq.corpus),
+        k1=0.9,
+        b=0.4
+    )
+bm25_index.save("output/bm25_index_b")  # Save index to directory
+bm25_retriever = BM25Retriever(index_dir="output/bm25_index_b")
 
 corpus_dict = {doc.collection_id: doc.text for doc in sciq.corpus}
 
-def get_query(query):
+# Search function for the BM25 system
+def search(query: str) -> List[Hit]:
+    # Replace the following placeholder with actual retrieval logic
     results = bm25_retriever.retrieve(query)
     hits = [
         {
             "cid": cid,
             "score": score,
-            "text": corpus_dict[cid]
+            "text": corpus_dict[cid]  # Assuming sciq.corpus maps cids to document texts
         }
         for cid, score in results.items()
     ]
     return hits
 
-
-
+def handle_search(query):
+        results = search(query)
+        return results
+        
+    
 demo = gr.Interface(
-    fn=get_query,
-    inputs=gr.Textbox(label="Enter your query"),
-    outputs=gr.Textbox(label="Results", lines=20, interactive=False),
-    title="BM25 Query Engine"
+    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
 )
-## YOUR_CODE_ENDS_HERE
-demo.launch()
+
+# Launch the app with shareable URL
+demo.launch(debug=True)
+