added .env

.gitignore

@@ -1,3 +1,3 @@
-config.yaml
+.env
 .venv
 logs

src/app/homepage.py

@@ -39,7 +39,7 @@ def search_tab():
 
     with st.container():
 
-        col1, col2 = st.columns([7, 4], gap="small")
+        col1, col2 = st.columns([7, 5], gap="small")
         with col1:
             search_query = st.text_input(
                 label="🔍 Search for Images",

src/config/config.py

@@ -1,14 +0,0 @@
-import yaml
-from utils import logger
-
-logger = logger.get_logger()
-
-def load_config():
-    try:
-        with open('config.yaml', 'r') as file:
-            config_data = yaml.load(file, Loader=yaml.FullLoader)
-            logger.info("Successfully loaded the config.")
-            return config_data
-    except Exception as e:
-        logger.error(f"Unexpected error occurred while loading the config: {e}")
-        raise Exception(f"Error loading configuration: {e}")

src/database_pinecone/create_database.py

@@ -5,11 +5,12 @@ sys.path.append(src_directory)
 from pinecone import Pinecone, ServerlessSpec
 import time
 from model.clip_model import ClipModel
-from config import config
+from dotenv import load_dotenv
 from utils import logger
 import pandas as pd
 
-config = config.load_config()
+load_dotenv()
+pincone_api_key = os.environ.get("PINECONE_API_KEY")
 logger = logger.get_logger()
 
 clip_model = ClipModel()
@@ -38,7 +39,6 @@ def wait_till_index_loaded(pinecone, index_name):
 
 def get_index():
     try:
-        pincone_api_key = config['pinecone_db']['pincone_api_key']
         pc = Pinecone(api_key=pincone_api_key)
         index = None
         index_name = "imagesearch"