similarity search

app.py

@@ -28,19 +28,19 @@ if files:
         text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=150)
         chunks = text_splitter.split_text(full_text)
         embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
-        vectorstore = FAISS.from_texts(chunks, embeddings)
+        db = FAISS.from_texts(chunks, embeddings)
         memory = ConversationBufferMemory(memory_key='chat_history', return_messages=True,)
-        llm = AutoModelForCausalLM.from_pretrained("red1xe/Llama-2-7B-codeGPT")
-        chain = RetrievalQA.from_llm(
-            llm=llm,
-            retriever=vectorstore.as_retriever(),
-            memory=memory,  
-        )
-        st.success("Done!")
-        st.header("Start Chat")
-        st.subheader("Ask a question")
-        question = st.text_input("Question")
-        if st.button("Ask"):
-            with st.spinner("Thinking..."):
-                answer = chain.query(question)
-                st.success(answer)
+
+def retrieve_info(query):
+    similar_response = db.similarity_search(query, k=3)
+    page_contents_array = [doc.page_contents for doc in similar_response]
+    page_contents = " ".join(page_contents_array)
+    return page_contents
+
+st.header("Chatbot")
+st.subheader("Ask a question")
+question = st.text_input("Question")
+if question:
+    st.subheader("Answer")
+    answer = retrieve_info(question)
+    st.write(answer)