Update knowledge_base.py

knowledge_base.py

@@ -1,12 +1,14 @@
 # Create FAISS index
-texts = text.split("\n")
-faiss_index, stored_texts = create_faiss_index(texts)
+def create_faiss_index(texts):
+    # Assuming `texts` is a list of text data
+    import faiss
+    from sentence_transformers import SentenceTransformer
 
-# Ask a question
-user_query = input("Ask a question: ")
-relevant_texts = search_faiss(faiss_index, stored_texts, user_query, top_k=3)
+    model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
+    embeddings = model.encode(texts)
 
-context = "\n".join(relevant_texts)
-answer = generate_response(context, user_query)
+    dimension = embeddings.shape[1]
+    index = faiss.IndexFlatL2(dimension)
+    index.add(embeddings)
 
-print("Chatbot Answer:", answer)
+    return index, texts