[WIP]

SearchEngine.py

@@ -3,14 +3,24 @@ from chromadb.config import Settings
 import chromadb
 
 class searchengine:
+    instance = None
+    
+    def __new__(cls, model_name, collection_name):
+        if cls.instance is None:
+            cls.instance = super(searchengine, cls).__new__(cls)
+            return cls.instance
+        else:
+            return cls.instance
     def __init__(self, model_name,collection_name):
+        
         self.sentence_transformer_ef = embedding_functions.SentenceTransformerEmbeddingFunction(
-            model_name = model_name
-            )
+                model_name = model_name
+                )
         self.chroma_client = chromadb.Client(
-            )
+                )
         self.collection = self.chroma_client.get_or_create_collection(name=collection_name)
-        
+        self.initialized = True
+            
     def add(self, text , metadata,id):
         self.collection.add(
             documents = [text],

app.py

@@ -6,15 +6,14 @@ import logging
 # Set up logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
+model_name = "all-MiniLM-L6-v2" 
+collection_name = "docs"  
 
+search_engine = searchengine(model_name, collection_name)
 
 def main():
     st.title("ChromaDB Search Engine")
-    repoid = 'sentence-transformers/all-MiniLM-L6-v2'
-    model_name = "all-MiniLM-L6-v2" 
-    collection_name = "docs"  
-
-    search_engine = searchengine(model_name, collection_name)
+    
 
     st.sidebar.header("Add Document")
     text_input = st.sidebar.text_area("Text")