Merge branch 'main' of https://huggingface.co/spaces/Kalpokoch/ChatbotDemo

app/app.py

@@ -2,6 +2,8 @@ from fastapi import FastAPI, Request
 from pydantic import BaseModel
 from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
 import torch
+from app.policy_vector_db import PolicyVectorDB  # Make sure this is your local DB logic
+import chromadb
 from app.policy_vector_db import PolicyVectorDB
 import chromadb # Make sure chromadb is imported if you use it directly later, though PolicyVectorDB handles it.
 
@@ -10,6 +12,7 @@ app = FastAPI()
 
 # --- REVISED: Load the vector database from the path inside the Docker container ---
 print("Loading Vector Database...")
+db = PolicyVectorDB(persist_directory="/app/policy_vector_db")
 # The path must match where you copied the DB in the Dockerfile
 DB_PERSIST_DIRECTORY = "/app/vector_database"
 db = PolicyVectorDB(persist_directory=DB_PERSIST_DIRECTORY)

app/policy_vector_db.py

@@ -101,7 +101,7 @@ def main():
     """Main function to build and verify the vector database (for local pre-building)."""
     BASE_DIR = os.path.dirname(os.path.abspath(__file__))
     INPUT_CHUNKS_PATH = os.path.join(BASE_DIR, "../processed_chunks.json")
-    PERSIST_DIRECTORY = "/tmp/policy_vector_db"
+    PERSIST_DIRECTORY = "/app/policy_vector_db"
 
     if not os.path.exists(INPUT_CHUNKS_PATH):
         print(f"FATAL ERROR: The input chunk file was not found at '{INPUT_CHUNKS_PATH}'")