updated the folder structure

.gitignore

@@ -0,0 +1,3 @@
+.venv
+.env
+vector-db

logs/api/yuvabe_care_companion_ai_api.log

@@ -0,0 +1,173 @@
+2025-03-06 20:02 - INFO - Hello World
+2025-03-06 20:02 - INFO - Hello World
+2025-03-06 20:22 - INFO - src/data/dataset.csv not found. Reading from Parquet file.
+2025-03-06 20:22 - ERROR - Error while loading dataset: Unable to find a usable engine; tried using: 'pyarrow', 'fastparquet'.
+A suitable version of pyarrow or fastparquet is required for parquet support.
+Trying to import the above resulted in these errors:
+ - Missing optional dependency 'pyarrow'. pyarrow is required for parquet support. Use pip or conda to install pyarrow.
+ - Missing optional dependency 'fastparquet'. fastparquet is required for parquet support. Use pip or conda to install fastparquet.
+Traceback (most recent call last):
+  File "C:\Users\Vijay\OneDrive\Desktop\Yuvabe_Care_Companion_AI\src\backend\data\dataset.py", line 16, in get_data_set
+    df = pd.read_parquet(PARAQUET_DATASET_PATH)
+  File "C:\Users\Vijay\OneDrive\Desktop\Yuvabe_Care_Companion_AI\.venv\Lib\site-packages\pandas\io\parquet.py", line 651, in read_parquet
+    impl = get_engine(engine)
+  File "C:\Users\Vijay\OneDrive\Desktop\Yuvabe_Care_Companion_AI\.venv\Lib\site-packages\pandas\io\parquet.py", line 67, in get_engine
+    raise ImportError(
+    ...<7 lines>...
+    )
+ImportError: Unable to find a usable engine; tried using: 'pyarrow', 'fastparquet'.
+A suitable version of pyarrow or fastparquet is required for parquet support.
+Trying to import the above resulted in these errors:
+ - Missing optional dependency 'pyarrow'. pyarrow is required for parquet support. Use pip or conda to install pyarrow.
+ - Missing optional dependency 'fastparquet'. fastparquet is required for parquet support. Use pip or conda to install fastparquet.
+2025-03-06 20:22 - ERROR - 'NoneType' object has no attribute 'head'
+2025-03-06 20:24 - INFO - src/backend/data/dataset.csv not found. Reading from Parquet file.
+2025-03-06 20:24 - ERROR - Error while loading dataset: Unable to find a usable engine; tried using: 'pyarrow', 'fastparquet'.
+A suitable version of pyarrow or fastparquet is required for parquet support.
+Trying to import the above resulted in these errors:
+ - Missing optional dependency 'pyarrow'. pyarrow is required for parquet support. Use pip or conda to install pyarrow.
+ - Missing optional dependency 'fastparquet'. fastparquet is required for parquet support. Use pip or conda to install fastparquet.
+Traceback (most recent call last):
+  File "C:\Users\Vijay\OneDrive\Desktop\Yuvabe_Care_Companion_AI\src\backend\data\dataset.py", line 16, in get_data_set
+    df = pd.read_parquet(PARAQUET_DATASET_PATH)
+  File "C:\Users\Vijay\OneDrive\Desktop\Yuvabe_Care_Companion_AI\.venv\Lib\site-packages\pandas\io\parquet.py", line 651, in read_parquet
+    impl = get_engine(engine)
+  File "C:\Users\Vijay\OneDrive\Desktop\Yuvabe_Care_Companion_AI\.venv\Lib\site-packages\pandas\io\parquet.py", line 67, in get_engine
+    raise ImportError(
+    ...<7 lines>...
+    )
+ImportError: Unable to find a usable engine; tried using: 'pyarrow', 'fastparquet'.
+A suitable version of pyarrow or fastparquet is required for parquet support.
+Trying to import the above resulted in these errors:
+ - Missing optional dependency 'pyarrow'. pyarrow is required for parquet support. Use pip or conda to install pyarrow.
+ - Missing optional dependency 'fastparquet'. fastparquet is required for parquet support. Use pip or conda to install fastparquet.
+2025-03-06 20:24 - ERROR - 'NoneType' object has no attribute 'head'
+2025-03-06 20:24 - INFO - src/backend/data/dataset.csv not found. Reading from Parquet file.
+2025-03-06 20:24 - ERROR - Error while loading dataset: Unable to find a usable engine; tried using: 'pyarrow', 'fastparquet'.
+A suitable version of pyarrow or fastparquet is required for parquet support.
+Trying to import the above resulted in these errors:
+ - Missing optional dependency 'pyarrow'. pyarrow is required for parquet support. Use pip or conda to install pyarrow.
+ - Missing optional dependency 'fastparquet'. fastparquet is required for parquet support. Use pip or conda to install fastparquet.
+Traceback (most recent call last):
+  File "C:\Users\Vijay\OneDrive\Desktop\Yuvabe_Care_Companion_AI\src\backend\data\dataset.py", line 16, in get_data_set
+    df = pd.read_parquet(PARAQUET_DATASET_PATH)
+  File "C:\Users\Vijay\OneDrive\Desktop\Yuvabe_Care_Companion_AI\.venv\Lib\site-packages\pandas\io\parquet.py", line 651, in read_parquet
+    impl = get_engine(engine)
+  File "C:\Users\Vijay\OneDrive\Desktop\Yuvabe_Care_Companion_AI\.venv\Lib\site-packages\pandas\io\parquet.py", line 67, in get_engine
+    raise ImportError(
+    ...<7 lines>...
+    )
+ImportError: Unable to find a usable engine; tried using: 'pyarrow', 'fastparquet'.
+A suitable version of pyarrow or fastparquet is required for parquet support.
+Trying to import the above resulted in these errors:
+ - Missing optional dependency 'pyarrow'. pyarrow is required for parquet support. Use pip or conda to install pyarrow.
+ - Missing optional dependency 'fastparquet'. fastparquet is required for parquet support. Use pip or conda to install fastparquet.
+2025-03-06 20:24 - ERROR - 'NoneType' object has no attribute 'head'
+2025-03-06 20:25 - INFO - src/backend/data/dataset.csv not found. Reading from Parquet file.
+2025-03-06 20:25 - ERROR - Error while loading dataset: Unable to find a usable engine; tried using: 'pyarrow', 'fastparquet'.
+A suitable version of pyarrow or fastparquet is required for parquet support.
+Trying to import the above resulted in these errors:
+ - Missing optional dependency 'pyarrow'. pyarrow is required for parquet support. Use pip or conda to install pyarrow.
+ - Missing optional dependency 'fastparquet'. fastparquet is required for parquet support. Use pip or conda to install fastparquet.
+Traceback (most recent call last):
+  File "C:\Users\Vijay\OneDrive\Desktop\Yuvabe_Care_Companion_AI\src\backend\data\dataset.py", line 16, in get_data_set
+    df = pd.read_parquet(PARAQUET_DATASET_PATH)
+  File "C:\Users\Vijay\OneDrive\Desktop\Yuvabe_Care_Companion_AI\.venv\Lib\site-packages\pandas\io\parquet.py", line 651, in read_parquet
+    impl = get_engine(engine)
+  File "C:\Users\Vijay\OneDrive\Desktop\Yuvabe_Care_Companion_AI\.venv\Lib\site-packages\pandas\io\parquet.py", line 67, in get_engine
+    raise ImportError(
+    ...<7 lines>...
+    )
+ImportError: Unable to find a usable engine; tried using: 'pyarrow', 'fastparquet'.
+A suitable version of pyarrow or fastparquet is required for parquet support.
+Trying to import the above resulted in these errors:
+ - Missing optional dependency 'pyarrow'. pyarrow is required for parquet support. Use pip or conda to install pyarrow.
+ - Missing optional dependency 'fastparquet'. fastparquet is required for parquet support. Use pip or conda to install fastparquet.
+2025-03-06 20:25 - ERROR - 'NoneType' object has no attribute 'head'
+2025-03-06 20:27 - INFO - src/backend/data/dataset.csv not found. Reading from Parquet file.
+2025-03-06 20:27 - ERROR - Error while loading dataset: Unable to find a usable engine; tried using: 'pyarrow', 'fastparquet'.
+A suitable version of pyarrow or fastparquet is required for parquet support.
+Trying to import the above resulted in these errors:
+ - Missing optional dependency 'pyarrow'. pyarrow is required for parquet support. Use pip or conda to install pyarrow.
+ - Missing optional dependency 'fastparquet'. fastparquet is required for parquet support. Use pip or conda to install fastparquet.
+Traceback (most recent call last):
+  File "C:\Users\Vijay\OneDrive\Desktop\Yuvabe_Care_Companion_AI\src\backend\data\dataset.py", line 16, in get_data_set
+    df = pd.read_parquet(PARAQUET_DATASET_PATH)
+  File "C:\Users\Vijay\OneDrive\Desktop\Yuvabe_Care_Companion_AI\.venv\Lib\site-packages\pandas\io\parquet.py", line 651, in read_parquet
+    impl = get_engine(engine)
+  File "C:\Users\Vijay\OneDrive\Desktop\Yuvabe_Care_Companion_AI\.venv\Lib\site-packages\pandas\io\parquet.py", line 67, in get_engine
+    raise ImportError(
+    ...<7 lines>...
+    )
+ImportError: Unable to find a usable engine; tried using: 'pyarrow', 'fastparquet'.
+A suitable version of pyarrow or fastparquet is required for parquet support.
+Trying to import the above resulted in these errors:
+ - Missing optional dependency 'pyarrow'. pyarrow is required for parquet support. Use pip or conda to install pyarrow.
+ - Missing optional dependency 'fastparquet'. fastparquet is required for parquet support. Use pip or conda to install fastparquet.
+2025-03-06 20:27 - INFO - Hello World
+2025-03-06 20:27 - ERROR - 'NoneType' object has no attribute 'to_dict'
+2025-03-06 20:28 - INFO - src/backend/data/dataset.csv not found. Reading from Parquet file.
+2025-03-06 20:28 - ERROR - Error while loading dataset: Unable to find a usable engine; tried using: 'pyarrow', 'fastparquet'.
+A suitable version of pyarrow or fastparquet is required for parquet support.
+Trying to import the above resulted in these errors:
+ - Missing optional dependency 'pyarrow'. pyarrow is required for parquet support. Use pip or conda to install pyarrow.
+ - Missing optional dependency 'fastparquet'. fastparquet is required for parquet support. Use pip or conda to install fastparquet.
+Traceback (most recent call last):
+  File "C:\Users\Vijay\OneDrive\Desktop\Yuvabe_Care_Companion_AI\src\backend\data\dataset.py", line 16, in get_data_set
+    df = pd.read_parquet(PARAQUET_DATASET_PATH)
+  File "C:\Users\Vijay\OneDrive\Desktop\Yuvabe_Care_Companion_AI\.venv\Lib\site-packages\pandas\io\parquet.py", line 651, in read_parquet
+    impl = get_engine(engine)
+  File "C:\Users\Vijay\OneDrive\Desktop\Yuvabe_Care_Companion_AI\.venv\Lib\site-packages\pandas\io\parquet.py", line 67, in get_engine
+    raise ImportError(
+    ...<7 lines>...
+    )
+ImportError: Unable to find a usable engine; tried using: 'pyarrow', 'fastparquet'.
+A suitable version of pyarrow or fastparquet is required for parquet support.
+Trying to import the above resulted in these errors:
+ - Missing optional dependency 'pyarrow'. pyarrow is required for parquet support. Use pip or conda to install pyarrow.
+ - Missing optional dependency 'fastparquet'. fastparquet is required for parquet support. Use pip or conda to install fastparquet.
+2025-03-06 20:28 - INFO - Hello World
+2025-03-06 20:28 - ERROR - 'NoneType' object has no attribute 'to_dict'
+2025-03-06 20:57 - INFO - src/backend/data/dataset.csv not found. Reading from Parquet file.
+2025-03-06 20:57 - INFO - CSV file created and cleaned at: src/backend/data/dataset.csv
+2025-03-06 21:00 - INFO - Loading existing dataset from: src/backend/data/dataset.csv
+2025-03-06 21:00 - INFO - Dataset loaded successfully.
+2025-03-06 21:02 - INFO - Loading existing dataset from: src/backend/data/dataset.csv
+2025-03-06 21:02 - INFO - Dataset loaded successfully.
+2025-03-06 21:04 - INFO - Loading existing dataset from: src/backend/data/dataset.csv
+2025-03-06 21:04 - INFO - Dataset loaded successfully.
+2025-03-06 21:05 - INFO - Loading existing dataset from: src/backend/data/dataset.csv
+2025-03-06 21:05 - INFO - Dataset loaded successfully.
+2025-03-06 21:05 - INFO - Loading existing dataset from: src/backend/data/dataset.csv
+2025-03-06 21:05 - INFO - Dataset loaded successfully.
+2025-03-06 21:05 - INFO - Loading existing dataset from: src/backend/data/dataset.csv
+2025-03-06 21:05 - INFO - Dataset loaded successfully.
+2025-03-06 21:06 - INFO - Loading existing dataset from: src/backend/data/dataset.csv
+2025-03-06 21:06 - INFO - Dataset loaded successfully.
+2025-03-06 21:06 - INFO - Trying to fetch the data from database
+2025-03-06 21:06 - INFO - Generated AI response for user prompt: I'm not feeling well today...
+2025-03-06 21:08 - INFO - Loading existing dataset from: src/backend/data/dataset.csv
+2025-03-06 21:08 - INFO - Dataset loaded successfully.
+2025-03-06 21:09 - INFO - Loading existing dataset from: src/backend/data/dataset.csv
+2025-03-06 21:09 - INFO - Dataset loaded successfully.
+2025-03-06 21:09 - INFO - Trying to fetch the data from database
+2025-03-06 21:09 - INFO - Generated AI response for user prompt: I'm not feeling well today...
+2025-03-06 21:13 - INFO - Loading existing dataset from: src/backend/data/dataset.csv
+2025-03-06 21:13 - INFO - Dataset loaded successfully.
+2025-03-06 21:19 - INFO - Trying to fetch the data from database
+2025-03-06 21:19 - INFO - Received user prompt: I'm not feeling well today
+2025-03-06 21:19 - INFO - Trying to fetch the data from database
+2025-03-06 21:19 - INFO - Generated AI response for user prompt: I'm not feeling well today...
+2025-03-06 21:19 - INFO - Generated AI response: I'm so sorry to hear that you're not feeling well today. It can be really frustrating and worrying w...
+2025-03-06 21:21 - INFO - Trying to fetch the data from database
+2025-03-06 21:21 - INFO - Successfully fetched the data from database
+2025-03-06 21:21 - INFO - Received user prompt: I'm not feeling well today
+2025-03-06 21:21 - INFO - Trying to fetch the data from database
+2025-03-06 21:21 - INFO - Successfully fetched the data from database
+2025-03-06 21:21 - INFO - Generated AI response for user prompt: I'm not feeling well today...
+2025-03-06 21:21 - INFO - Generated AI response: Sorry to hear that you're not feeling well today. It can be concerning when we're not at our best. S...
+2025-03-06 21:47 - INFO - Received user prompt: I'm not feeling well today
+2025-03-06 21:47 - INFO - Trying to fetch the data from database
+2025-03-06 21:47 - INFO - Successfully fetched the data from database
+2025-03-06 21:47 - INFO - Generated AI response for user prompt: I'm not feeling well today...
+2025-03-06 21:47 - INFO - Generated AI response: Sorry to hear that you're not feeling well today. It's always a good idea to try to identify the pos...

logs/api/yuvabe_care_companion_ai_app.log

@@ -0,0 +1,77 @@
+2025-03-06 21:45 - INFO - Page successfully configured with title: Yuvabe Care Companion AI
+2025-03-06 21:45 - ERROR - Error converting image to base64: [Errno 2] No such file or directory: 'src/data/images/page_icon.jpg'
+2025-03-06 21:45 - INFO - user_name not found in session_state, setting to None.
+2025-03-06 21:45 - INFO - user_name is None, requesting user input.
+2025-03-06 21:46 - INFO - Page successfully configured with title: Yuvabe Care Companion AI
+2025-03-06 21:46 - INFO - user_name not found in session_state, setting to None.
+2025-03-06 21:46 - INFO - user_name is None, requesting user input.
+2025-03-06 21:47 - INFO - Page successfully configured with title: Yuvabe Care Companion AI
+2025-03-06 21:47 - INFO - user_name is None, requesting user input.
+2025-03-06 21:47 - INFO - User entered name: Vela. Setting session_state.user_name.
+2025-03-06 21:47 - INFO - Page successfully configured with title: Yuvabe Care Companion AI
+2025-03-06 21:47 - INFO - User already entered a name: Vela. Displaying greeting.
+2025-03-06 21:47 - INFO - Displaying the chat history.
+2025-03-06 21:47 - INFO - Waiting for user input...
+2025-03-06 21:47 - INFO - Page successfully configured with title: Yuvabe Care Companion AI
+2025-03-06 21:47 - INFO - User already entered a name: Vela. Displaying greeting.
+2025-03-06 21:47 - INFO - Displaying the chat history.
+2025-03-06 21:47 - INFO - Waiting for user input...
+2025-03-06 21:47 - INFO - Assistant response: Sorry to hear that you're not feeling well today. It's always a good idea to try to identify the pos...
+2025-03-06 21:48 - INFO - Page successfully configured with title: Yuvabe Care Companion AI
+2025-03-06 21:48 - INFO - user_name not found in session_state, setting to None.
+2025-03-06 21:48 - INFO - user_name is None, requesting user input.
+2025-03-06 21:51 - INFO - Page successfully configured with title: Yuvabe Care Companion AI
+2025-03-06 21:51 - INFO - user_name not found in session_state, setting to None.
+2025-03-06 21:51 - INFO - user_name is None, requesting user input.
+2025-03-06 21:51 - INFO - Page successfully configured with title: Yuvabe Care Companion AI
+2025-03-06 21:51 - INFO - user_name not found in session_state, setting to None.
+2025-03-06 21:51 - INFO - user_name is None, requesting user input.
+2025-03-06 21:51 - INFO - Page successfully configured with title: Yuvabe Care Companion AI
+2025-03-06 21:51 - INFO - user_name is None, requesting user input.
+2025-03-06 21:51 - INFO - User entered name: Velu. Setting session_state.user_name.
+2025-03-06 21:51 - INFO - Page successfully configured with title: Yuvabe Care Companion AI
+2025-03-06 21:51 - INFO - User already entered a name: Velu. Displaying greeting.
+2025-03-06 21:51 - INFO - Displaying the chat history.
+2025-03-06 21:51 - INFO - Waiting for user input...
+2025-03-06 21:52 - INFO - Page successfully configured with title: Yuvabe Care Companion AI
+2025-03-06 21:52 - INFO - User already entered a name: Velu. Displaying greeting.
+2025-03-06 21:52 - INFO - Displaying the chat history.
+2025-03-06 21:52 - INFO - Waiting for user input...
+2025-03-06 21:53 - INFO - Page successfully configured with title: Yuvabe Care Companion AI
+2025-03-06 21:53 - INFO - User already entered a name: Velu. Displaying greeting.
+2025-03-06 21:53 - INFO - Displaying the chat history.
+2025-03-06 21:53 - INFO - Waiting for user input...
+2025-03-06 21:53 - INFO - Page successfully configured with title: Yuvabe Care Companion AI
+2025-03-06 21:53 - INFO - User already entered a name: Velu. Displaying greeting.
+2025-03-06 21:53 - INFO - Displaying the chat history.
+2025-03-06 21:53 - INFO - Waiting for user input...
+2025-03-06 21:53 - INFO - Page successfully configured with title: Yuvabe Care Companion AI
+2025-03-06 21:53 - INFO - User already entered a name: Velu. Displaying greeting.
+2025-03-06 21:53 - INFO - Displaying the chat history.
+2025-03-06 21:53 - INFO - Waiting for user input...
+2025-03-06 21:53 - INFO - Page successfully configured with title: Yuvabe Care Companion AI
+2025-03-06 21:53 - INFO - User already entered a name: Velu. Displaying greeting.
+2025-03-06 21:53 - INFO - Displaying the chat history.
+2025-03-06 21:53 - INFO - Waiting for user input...
+2025-03-06 21:53 - INFO - Page successfully configured with title: Yuvabe Care Companion AI
+2025-03-06 21:53 - INFO - User already entered a name: Velu. Displaying greeting.
+2025-03-06 21:53 - INFO - Displaying the chat history.
+2025-03-06 21:53 - INFO - Waiting for user input...
+2025-03-06 21:53 - INFO - Page successfully configured with title: Yuvabe Care Companion AI
+2025-03-06 21:53 - INFO - user_name not found in session_state, setting to None.
+2025-03-06 21:53 - INFO - user_name is None, requesting user input.
+2025-03-06 21:53 - INFO - Page successfully configured with title: Yuvabe Care Companion AI
+2025-03-06 21:53 - INFO - user_name is None, requesting user input.
+2025-03-06 21:53 - INFO - Page successfully configured with title: Yuvabe Care Companion AI
+2025-03-06 21:53 - INFO - user_name is None, requesting user input.
+2025-03-06 21:53 - INFO - Page successfully configured with title: Yuvabe Care Companion AI
+2025-03-06 21:53 - INFO - user_name is None, requesting user input.
+2025-03-06 21:53 - INFO - User entered name: Velu. Setting session_state.user_name.
+2025-03-06 21:53 - INFO - Page successfully configured with title: Yuvabe Care Companion AI
+2025-03-06 21:53 - INFO - User already entered a name: Velu. Displaying greeting.
+2025-03-06 21:53 - INFO - Displaying the chat history.
+2025-03-06 21:53 - INFO - Waiting for user input...
+2025-03-06 21:55 - INFO - Page successfully configured with title: Yuvabe Care Companion AI
+2025-03-06 21:55 - INFO - Page successfully configured with title: Yuvabe Care Companion AI
+2025-03-06 21:55 - INFO - user_name not found in session_state, setting to None.
+2025-03-06 21:55 - INFO - user_name is None, requesting user input.

requirements.txt

@@ -0,0 +1,12 @@
+streamlit
+requests
+Pillow
+pandas
+fastapi[standard]
+pinecone
+torch
+torchvision
+torchaudio
+sentence_transformers
+groq
+chromadb

src/backend/data/chroma_db.py

@@ -0,0 +1,67 @@
+import chromadb
+from utils import logger
+from chromadb.utils import embedding_functions
+default_ef = embedding_functions.DefaultEmbeddingFunction()
+
+logger = logger.get_logger()
+
+# Constants
+COLLECTION_NAME = "care_companion_ai_vectors"
+DB_PATH = "vector-db"
+
+# Initialize ChromaDB Client
+client = chromadb.PersistentClient(path="src/backend/vector-db")
+
+collection = client.get_or_create_collection(
+    name=COLLECTION_NAME,
+    embedding_function=default_ef,
+    metadata={
+    "description": "yuvabe care companion ai chroma collection",
+    "hnsw:space": "cosine",
+    "hnsw:search_ef": 100
+    })
+
+def add_data_to_vector_store(df):
+    try:
+        logger.info("Started upserting the data to database")
+        for index, row in df.iterrows():
+            input_text = row['input']
+            output_text = row['output']
+            instruction_text = row['instruction']
+            if not isinstance(input_text, str) or not input_text.strip():
+                logger.warning(f"Skipping row {index} due to empty or invalid input text.")
+                continue
+            row_dict = {
+                "question": input_text,
+                "answer" : output_text,
+                "instruction": instruction_text
+            }
+            collection.upsert(
+                documents=input_text,
+                metadatas=row_dict,
+                ids=f"id{index}"
+            )
+            logger.info(f"Successfully upserted {index} records.")
+        logger.info("Successfully upserted all the records.")
+    except Exception as e:
+        logger.exception(f"Unable to upsert the data to the database: {e}")
+
+def search_vector_store(query, n_result : int = 3):
+    try:
+        logger.info("Trying to fetch the data from database")
+        response = collection.query(
+        query_texts=[query],
+        n_results=n_result,
+        include=["metadatas","distances","documents"]
+        )
+        logger.info("Successfully fetched the data from database")
+        return response
+    except Exception as e:
+        logger.exception("Failed to fetch the data from database")
+
+def get_retrieved_context(prompt: str) -> str:
+    response = search_vector_store(prompt)
+    if response and "metadatas" in response and response["metadatas"]:
+        retrieved_contexts = [metadata["answer"] for metadata in response["metadatas"][0]]
+        return "\n".join(retrieved_contexts[:3])
+    return "No relevant information found in the database."

src/backend/data/dataset.py

@@ -0,0 +1,37 @@
+import os
+import pandas as pd
+import string
+from utils import logger
+
+logger = logger.get_logger()
+
+DATASET_PATH = "src/backend/data/dataset.csv"
+PARAQUET_DATASET_PATH = "hf://datasets/lavita/ChatDoctor-HealthCareMagic-100k/data/train-00000-of-00001-5e7cb295b9cff0bf.parquet"
+
+def get_data_set():
+
+    try:
+        if not os.path.exists(DATASET_PATH):
+            logger.info(f"{DATASET_PATH} not found. Reading from Parquet file.")
+            df = pd.read_parquet(PARAQUET_DATASET_PATH)
+            df.drop_duplicates(subset=["input", "output"], inplace=True)
+            df.dropna(subset=["input", "output"], inplace=True) # Remove NaNs first
+        
+        # This line is to remove the empty column or column with only spaces
+            df = df[(df["input"].str.strip() != "") & (df["output"].str.strip() != "")] # Remove empty strings/spaces
+
+        # This line is to remove puncuation and emjois
+            translator = str.maketrans('', '', string.punctuation)
+            df["input"] = df["input"].str.lower().str.translate(translator)
+            df["output"] = df["output"].str.lower().str.translate(translator)
+            df.to_csv(DATASET_PATH, index=False)
+            logger.info(f"CSV file created and cleaned at: {DATASET_PATH}")
+        else:
+            logger.info(f"Loading existing dataset from: {DATASET_PATH}")
+            df = pd.read_csv(DATASET_PATH)
+            logger.info("Dataset loaded successfully.")
+        return df
+
+    except Exception as e:
+        logger.error(f"Error while loading dataset: {e}", exc_info=True)
+        return None

src/backend/main.py

@@ -0,0 +1,6 @@
+from fastapi import FastAPI
+from routes.chat_api import router
+
+app = FastAPI()
+
+app.include_router(router, prefix="/chat", tags=["chat"])

src/backend/models/llm_model.py

@@ -0,0 +1,63 @@
+import os
+from groq import Groq
+from utils import logger
+from data import chroma_db
+from dotenv import load_dotenv
+
+load_dotenv()
+GROQ_KEY = os.environ.get("GROQ_API")
+logger = logger.get_logger()
+LOCAL_MODEL_PATH = "src/model/embedding_model.pkl"
+LLM_MODEL_NAME="llama-3.3-70b-versatile"
+client = Groq(api_key=GROQ_KEY)
+SYSTEM_PROMPT = """You are Yuvabe Care Companion AI, an advanced healthcare assistant designed to assist users with a wide range of health-related queries. Your role includes:
+
+- **General Medical Guidance**: Providing basic health insights (⚠️ *not a replacement for a doctor*).
+- **Physiotherapy & Rehabilitation**: Advising on recovery exercises and therapy routines.
+- **Mental Health Support**: Offering well-being tips and emotional health guidance.
+- **Lifestyle & Wellness Advice**: Helping users with diet, sleep, and fitness recommendations.
+- **Chronic Disease Management**: Educating users on managing conditions like diabetes, hypertension, etc.
+- **Emergency Guidance**: Directing users on what to do in urgent medical situations (⚠️ *always recommend calling a doctor or emergency services*).
+
+⚠️ *Important*: You are not a certified doctor. Always remind users to consult a healthcare professional for medical decisions.
+"""
+
+def get_medical_assistant_response(prompt: str):
+    try:
+        if not prompt or len(prompt.strip()) < 5:
+            return "⚠️ Your question seems too short. Please provide more details so I can assist you better."
+
+        response = chroma_db.search_vector_store(prompt)
+        
+        if response and "metadatas" in response and response["metadatas"]:
+            retrieved_contexts = [metadata['answer'] for metadata in response["metadatas"][0]]
+            context = "\n".join(retrieved_contexts[:3])
+        else:
+            context = "No relevant information found in the database."
+
+        system_prompt = f"""
+        You are a helpful medical assistant. Use the provided context to answer the question as accurately as possible.
+        If the context is not relevant, rely on your knowledge to answer.
+        
+        Context:
+        {context}
+        
+        User Question: {prompt}
+        """
+
+        chat_completion = client.chat.completions.create(
+            messages=[
+                {"role": "system", "content": SYSTEM_PROMPT},
+                {"role": "user", "content": system_prompt},
+            ],
+            model=LLM_MODEL_NAME,
+        )
+
+        assistant_response = chat_completion.choices[0].message.content
+        logger.info(f"Generated AI response for user prompt: {prompt[:50]}...")
+
+        return assistant_response
+
+    except Exception as e:
+        logger.exception("Unexpected error occurred.")
+        return f"An error occurred while processing your request: {str(e)}"

src/backend/models/schemas.py

@@ -0,0 +1,6 @@
+from pydantic import BaseModel
+from typing import List,Dict, Optional
+
+class Chat_Response(BaseModel):
+    prompt: Optional[str] = None
+    response: Optional[Dict] = None

src/backend/routes/__init__.py

@@ -0,0 +1 @@
+from routes.chat_api import router

src/backend/routes/chat_api.py

@@ -0,0 +1,18 @@
+from fastapi import APIRouter, HTTPException
+from utils import logger
+from models.schemas import Chat_Response
+from models import llm_model
+
+logger = logger.get_logger()
+router = APIRouter()
+
+@router.post("/response")
+async def get_chat_response(chat_request: Chat_Response):
+    try:
+        logger.info(f"Received user prompt: {chat_request.prompt}")
+        response_text  = llm_model.get_medical_assistant_response(chat_request.prompt)
+        logger.info(f"Generated AI response: {response_text[:100]}...")
+        return {"status": "success", "response": response_text}
+    except Exception as e:
+        logger.exception("Unexpected error occurred while processing the request.")
+        raise HTTPException(status_code=500, detail="An error occurred while processing your request.")

src/backend/utils/__init__.py

@@ -0,0 +1 @@
+from utils import *

src/backend/utils/logger.py

@@ -0,0 +1,35 @@
+import logging
+from logging.handlers import RotatingFileHandler
+import os
+
+log_file = 'yuvabe_care_companion_ai_api.log'
+log_dir = 'logs/api'
+log_level=logging.INFO
+
+def get_logger( ):
+
+    if not os.path.exists(log_dir):
+        os.makedirs(log_dir)
+
+    log_file_path = os.path.join(log_dir, log_file)
+
+    logger = logging.getLogger(__name__)
+
+    if not logger.hasHandlers():
+        logger.setLevel(log_level)
+
+        console_handler = logging.StreamHandler()
+        console_handler.setLevel(logging.DEBUG) 
+
+        file_handler = RotatingFileHandler(log_file_path, maxBytes=5*1024*1024, backupCount=3)
+        file_handler.setLevel(logging.INFO)
+
+        log_format = '%(asctime)s - %(levelname)s - %(message)s'
+        formatter = logging.Formatter(log_format, datefmt='%Y-%m-%d %H:%M')
+        console_handler.setFormatter(formatter)
+        file_handler.setFormatter(formatter)
+
+        logger.addHandler(console_handler)
+        logger.addHandler(file_handler)
+
+    return logger

src/frontend/app/common_fuctions.py

@@ -0,0 +1,26 @@
+import base64
+import requests
+from utils import logger
+
+logger = logger.get_logger()
+
+def img_to_base64(image_path):
+    """Convert image to base64."""
+    try:
+        with open(image_path, "rb") as img_file:
+            return base64.b64encode(img_file.read()).decode()
+    except Exception as e:
+        logger.error(f"Error converting image to base64: {str(e)}")
+        return None
+    
+API_URL = "http://127.0.0.1:8000"
+
+def get_api_response(endpoint:str, prompt: str):
+    try:
+        response = requests.post(f"{API_URL}/{endpoint}", json={"prompt": prompt})
+        if response.status_code == 200:
+            return response.json()["response"]
+        else:
+            return "An error occurred while processing your request."
+    except Exception as e:
+        return f"An error occurred while processing your request: {str(e)}"

src/frontend/app/homepage.py

@@ -0,0 +1,133 @@
+import streamlit as st
+from utils import logger
+from app import common_fuctions
+
+logger = logger.get_logger()
+
+PAGE_TITLE = "Yuvabe Care Companion AI"
+PAGE_LAYOUT = "wide"
+PAGE_ICON = "src/frontend/images/page_icon.jpg"
+GITHUB_LINK = "https://github.com/Vela-Test1993/yuvabe-care-companion-ai"
+ABOUT_US = "An AI-powered assistant for personalized healthcare guidance."
+
+def get_or_greet_user_name():
+    if 'user_name' not in st.session_state:
+        st.session_state.user_name = None
+        logger.info("user_name not found in session_state, setting to None.")
+
+    if st.session_state.user_name is None:
+        logger.info("user_name is None, requesting user input.")
+        user_name = st.text_input("Please let me know your name:",
+                              placeholder="Enter your name buddy")
+        if user_name:
+            st.session_state.user_name = user_name
+            logger.info(f"User entered name: {user_name}. Setting session_state.user_name.")
+            st.toast(f"Hello {st.session_state.user_name} . Welcome to {PAGE_TITLE}")
+            st.rerun()
+    else:
+        logger.info(f"User already entered a name: {st.session_state.user_name}. Displaying greeting.")
+        return st._bottom.subheader(f"Hello {st.session_state.user_name}! How can I assist you today?")
+    
+def display_chat():
+    logger.info("Displaying the chat history.")
+    if "messages" not in st.session_state:
+        st.session_state.messages = []
+    for message in st.session_state.messages:
+        with st.chat_message(message["role"]):
+            st.markdown(message["content"])
+        logger.info(f"Displayed {len(st.session_state.messages)} messages from the chat history.")
+
+def handle_user_input():
+    """
+    Handles user input for Yuvabe Care Companion AI, validates query relevance,
+    and fetches a response from the medical assistant model.
+    """
+
+    logger.info("Waiting for user input...")
+
+    prompt = st.chat_input("Ask me anything about health, physiotherapy, or medical advice!")
+
+    if prompt:
+
+        with st.chat_message("user"):
+            st.markdown(prompt)
+
+        st.session_state.messages.append({"role": "user", "content": prompt})
+
+        try:
+            with st.spinner("Processing your query..."):
+                try:
+                    response = common_fuctions.get_api_response("/chat/response", prompt)
+                except Exception as e:
+                    logger.error(f"AI response generation failed: {str(e)}")
+                    response = "⚠️ Sorry, I couldn't process your request. Please try again later."
+        except Exception as e:
+            logger.exception("Error during similarity check or response generation.")
+            response = "⚠️ Oops! Something went wrong. Please try again."
+
+        with st.chat_message("assistant"):
+            st.markdown(response)
+
+        st.session_state.messages.append({"role": "assistant", "content": response})
+
+        logger.info(f"Assistant response: {response[:100]}...")
+
+def config_homepage():
+    st.set_page_config(
+    page_title=PAGE_TITLE,
+    page_icon= PAGE_ICON,
+    layout=PAGE_LAYOUT,
+    initial_sidebar_state="auto",
+    menu_items={"Get help":GITHUB_LINK,
+                "Report a bug": GITHUB_LINK,
+                "About": ABOUT_US}
+    )
+
+    st.markdown(f"""
+        <h1 style="color: darkblue; text-align: left; font-size: 50px;">
+        <i>{PAGE_TITLE} 🏥⚕️🤖</i>
+        </h1>
+        """, unsafe_allow_html=True
+    )
+
+    logger.info(f"Page successfully configured with title: {PAGE_TITLE}")
+    st.session_state.config_status = False
+
+def setup_homepage():
+    st.set_page_config(
+        page_title=PAGE_TITLE,
+        page_icon= PAGE_ICON,
+        layout=PAGE_LAYOUT,
+        initial_sidebar_state="auto",
+        menu_items={"Get help":GITHUB_LINK,
+                    "Report a bug": GITHUB_LINK,
+                    "About": ABOUT_US}
+    )
+
+    st.markdown(f"""
+        <h1 style="color: darkblue; text-align: left; font-size: 50px;">
+        <i>{PAGE_TITLE} 🏥⚕️🤖</i>
+        </h1>
+        """, unsafe_allow_html=True
+    )
+
+    logger.info(f"Page successfully configured with title: {PAGE_TITLE}")
+    st.session_state.config_status = False
+
+    st.markdown("<hr>", unsafe_allow_html=True)  # To add a Horizontal line below title
+
+    img_base64 = common_fuctions.img_to_base64(PAGE_ICON)
+
+    if img_base64:
+        st.sidebar.markdown(
+            f'<img src="data:image/png;base64,{img_base64}" class="cover-glow">',
+            unsafe_allow_html=True,
+    )
+        
+    st.sidebar.markdown("---")
+
+    if get_or_greet_user_name():
+
+        display_chat()
+
+        handle_user_input()

src/frontend/home.py

@@ -0,0 +1,4 @@
+from app import homepage
+
+homepage.setup_homepage()
+

src/frontend/pages/database_response_page.py

@@ -0,0 +1,3 @@
+from app import homepage
+
+homepage.config_homepage()

src/frontend/utils/logger.py

@@ -0,0 +1,35 @@
+import logging
+from logging.handlers import RotatingFileHandler
+import os
+
+log_file = 'yuvabe_care_companion_ai_app.log'
+log_dir = 'logs/api'
+log_level=logging.INFO
+
+def get_logger( ):
+
+    if not os.path.exists(log_dir):
+        os.makedirs(log_dir)
+
+    log_file_path = os.path.join(log_dir, log_file)
+
+    logger = logging.getLogger(__name__)
+
+    if not logger.hasHandlers():
+        logger.setLevel(log_level)
+
+        console_handler = logging.StreamHandler()
+        console_handler.setLevel(logging.DEBUG) 
+
+        file_handler = RotatingFileHandler(log_file_path, maxBytes=5*1024*1024, backupCount=3)
+        file_handler.setLevel(logging.INFO)
+
+        log_format = '%(asctime)s - %(levelname)s - %(message)s'
+        formatter = logging.Formatter(log_format, datefmt='%Y-%m-%d %H:%M')
+        console_handler.setFormatter(formatter)
+        file_handler.setFormatter(formatter)
+
+        logger.addHandler(console_handler)
+        logger.addHandler(file_handler)
+
+    return logger