attempt4

backend/semantic_search.py

@@ -14,25 +14,17 @@ logger = logging.getLogger(__name__)
 load_dotenv()
 
 # Connect to the LanceDB database
-# Determine the working directory
 current_working_dir = Path(os.getcwd())
-logger.info(f"Working directory: {current_working_dir}")
-
-# List contents of the current working directory
-current_dir_contents = os.listdir(current_working_dir)
-logger.info(f"Contents of the working directory: {current_dir_contents}")
-
-# Ensure the working directory contains 'gradio_app'
-if 'gradio_app' not in current_dir_contents:
-    raise FileNotFoundError("The 'gradio_app' directory is missing from the working directory.")
-
-# List and log all contents of the gradio_app directory
-gradio_app_dir = current_working_dir / 'gradio_app'
-gradio_app_contents = os.listdir(gradio_app_dir)
-logger.info(f"Contents of 'gradio_app' directory: {gradio_app_contents}")
-
-db_path = current_working_dir / "gradio_app" / ".lancedb"
+# Determine the LanceDB path and log it
+db_path = current_working_dir / ".lancedb"
 logger.info(f"Database path: {db_path}")
+
+# List contents of the LanceDB directory
+if db_path.exists():
+    lancedb_contents = os.listdir(db_path)
+    logger.info(f"Contents of the LanceDB directory: {lancedb_contents}")
+else:
+    logger.error(f"LanceDB directory does not exist at path: {db_path}")
 db = lancedb.connect(db_path)
 
 MODEL1_STRATEGY1 = "model1_fixed"
@@ -70,6 +62,6 @@ def retrieve(query, k):
     except Exception as e:
         raise gr.Error(str(e))
     
-# if __name__ == "__main__":
-#     res = retrieve("What is transformer?", 4)
-#     print(res)
+if __name__ == "__main__":
+    res = retrieve("What is transformer?", 4)
+    print(res)