Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import pdfplumber
 import docx
 from transformers import AutoTokenizer, AutoModelForCausalLM
+import faiss
 
 st.title("JAI")
 
@@ -83,3 +84,19 @@ if document_file is not None:
 
         # Aktuális beszélgetés hozzáadása az előző beszélgetésekhez
         st.session_state.previous_conversations.append({"input_text": input_text, "response": response})
+
+        # FAISS használata az előző beszélgetések indexelésére és keresésére
+        # Először hozz létre egy üres FAISS indexet
+        index = faiss.IndexFlatL2(embedding_dimension)  # Az embedding_dimension a vektormérete az előző beszélgetések reprezentálásához
+
+        # Az előző beszélgetéseket reprezentáló vektorokat hozzáadhatod az indexhez
+        for conversation_vector in previous_conversation_vectors:
+            index.add(conversation_vector)
+
+        # Keresés egy adott lekérdezés vektor alapján
+        query_vector = generate_vector_from_text(input_text)  # A lekérdezés szövegből létrehozott vektor
+        k = 5  # A legjobb 5 hasonlóságú beszélgetés visszaadása
+        D, I = index.search(query_vector, k)  # D: távolságok, I: indexek
+
+        # Azon beszélgetések visszaadása, amelyek a legjobban illeszkednek a lekérdezéshez
+        relevant_conversations = [st.session_state.previous_conversations[i] for i in I]