refactor code

Author: omar-sol

README.md

@@ -5,7 +5,7 @@ colorFrom: gray
 colorTo: pink
 sdk: gradio
 sdk_version: 4.39.0
-app_file: scripts/gradio-ui.py
+app_file: scripts/main.py
 pinned: false
 ---
 

requirements.txt

@@ -200,6 +200,7 @@ pydantic-core==2.20.1
 pydispatcher==2.0.7
 pydub==0.25.1
 pygments==2.18.0
+pymongo==4.8.0
 pyopenssl==24.2.1
 pyparsing==3.1.2
 pypdf==4.3.1

scripts/custom_retriever.py

@@ -1,4 +1,3 @@
-import logging
 import time
 from typing import List
 
@@ -8,9 +7,6 @@
 from llama_index.core.schema import NodeWithScore, TextNode
 from llama_index.postprocessor.cohere_rerank import CohereRerank
 
-logger = logging.getLogger(__name__)
-logging.basicConfig(level=logging.INFO)
-
 
 class CustomRetriever(BaseRetriever):
     """Custom retriever that performs both semantic search and hybrid search."""