Update app.py

app.py

@@ -12,10 +12,14 @@ import chromadb
 
 app = FastAPI()
 client = chromadb.PersistentClient(path="/data/chroma_db")
-collection = client.get_collection(name="knowledge_base")
+collection = client.get_or_create_collection(name="knowledge_base")
 pdf_file="Sutures and Suturing techniques.pdf"
 pptx_file="impalnt 1.pptx"
 process_and_store(pdf_path=pdf_file, pptx_path=pptx_file)
+collections = client.list_collections()
+
+print("Existing Collections:", [c.name for c in collections])
+collection = client.get_collection(name="knowledge_base")
 
 # Initialize models
 text_model = SentenceTransformer('all-MiniLM-L6-v2')