Update app.py

app.py

@@ -31,9 +31,13 @@ options = {
 
 st.set_page_config(page_title="BonsiAI", page_icon="🤖")
 
-def gen_augmented_prompt(prompt, top_k) :
-    context = ""     
+def gen_augmented_prompt(prompt, top_k) :   
     links = ""
+    persist_directory1 = './DB_Decreti'
+    embedding = HuggingFaceEmbeddings(model_name="sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2")
+    db = Chroma(persist_directory=persist_directory1, embedding_function=embedding) 
+    context = db.similarity_search(prompt, k=top_k)
+    print(context)
     generated_prompt = f"""
     A PARTIRE DAL SEGUENTE CONTESTO: {context},
 
@@ -157,19 +161,10 @@ def show_source(links) :
         for i, link in enumerate(links) :
             st.info(f"{link}")
 
-def init_vector_db():
-    persist_directory1 = './DB_Decreti'
-    embedding = HuggingFaceEmbeddings(model_name="sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2")
-    db = Chroma(persist_directory=persist_directory1, embedding_function=embedding)
-    #NumeroDocumenti = 10  
-    #query = 'Come funziona la generazione delle PDA'      
-    #result = db.similarity_search(query, k=NumeroDocumenti)
-
 init_state()
 sidebar()
 header()
 chat_box()
-init_vector_db()
 
 if prompt := st.chat_input("Chatta con BonsiAI..."):
     st.chat_message("user").markdown(prompt)