moved the app.py

README.md

@@ -11,3 +11,12 @@ license: mit
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+
+
+### TODOs
+
+[x] Refactor the vectordb.py - remove code duplication
+[x] Sustainability - database paths - move to HF
+[x] Fix it for the new models e.g. Llama and others
+[x] Add the space secrets to have it running online
+[x] Make the space public

src/vectordb/vectordb.py

@@ -88,6 +88,7 @@ def search_wikivoyage_docs(query: str, limit: int = 10, reranking: int = 0, run_
         logger.error(f"Error while connecting to DB: {e}")
 
     logger.info("Connected to Wikivoyage DB.")
+    print("Tablenames: ", db.table_names())
 
     # query_embedding = embed_query(query)
     table = db.open_table("wikivoyage_documents")
@@ -126,7 +127,7 @@ def search_wikivoyage_docs(query: str, limit: int = 10, reranking: int = 0, run_
     return city_lists
 
 
-def search_wikivoyage_listings(query, cities, limit=10, reranking=0):
+def search_wikivoyage_listings(query:str, cities: list, limit: int=10, reranking: int = 0, run_local: Optional[bool] = False):
     """
     
     Function to search the wikivoyage database an return most relevant listings, post-filtered by the recommended
@@ -139,11 +140,14 @@ def search_wikivoyage_listings(query, cities, limit=10, reranking=0):
         - reranking: bool (0 or 1), if activated, CrossEncoderReranker is used.
 
     """
-    uri = database_dir
-    current_dir = os.path.split(os.getcwd())[1]
+    if run_local:
+        uri = database_dir
+        current_dir = os.path.split(os.getcwd())[1]
 
-    if "src" or "tests" in current_dir: # hacky way to get the correct path
-        uri = uri.replace("../../", "../")
+        if "src" or "tests" in current_dir: # hacky way to get the correct path
+            uri = uri.replace("../../", "../")
+    else:
+        uri = os.environ["BUCKET_NAME"]
 
     db = lancedb.connect(uri)
     logger.info("Connected to Wikivoyage Listings DB.")