Update app.py

app.py

@@ -120,12 +120,15 @@ def formattaPrompt(prompt, systemRole, systemStyle, instruction):
 
 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)
+    db = Chroma(persist_directory='./DB_Decreti', embedding_function=embedding) 
+    print('Due')
+    query = "Bullismo"
+    docs = db.similarity_search(query, k=10)
+    print(docs)
+    #context = db.similarity_search(prompt, k=top_k)
     generated_prompt = f"""
-    A PARTIRE DAL SEGUENTE CONTESTO: {context},
+    A PARTIRE DAL SEGUENTE CONTESTO: {docs},
 
     ----
     RISPONDI ALLA SEGUENTE RICHIESTA: {prompt}