Update app.py

app.py

@@ -122,27 +122,32 @@ def gen_augmented_prompt(prompt, top_k) :
     links = ""
     embedding = HuggingFaceEmbeddings(model_name="sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2")
     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)
+    docs = db.similarity_search(prompt, k=top_k)
+    
+    links = []
+    context = ''
+    NomeCartellaOriginariaDB = 'Documenti_2\\'
+    for doc in docs:
+        testo = doc.page_content.replace('\n', ' ')
+        context += testo + '\n\n\n'
+        reference = doc.metadata["source"].replace(NomeCartellaOriginariaDB, '') + ' (Pag. ' + str(doc.metadata["page"]) + ')'
+        links.append((reference, testo))
     generated_prompt = f"""
     A PARTIRE DAL SEGUENTE CONTESTO: {docs},
 
     ----
     RISPONDI ALLA SEGUENTE RICHIESTA: {prompt}
     """
-    return generated_prompt, links
+    return context, links
 
 def generate_chat_stream(prompt) :
     links = []
     prompt_originale = prompt
-    prompt = formattaPrompt(prompt, st.session_state.systemRole, st.session_state.systemStyle, st.session_state.instruction)    
     if st.session_state.rag_enabled :
         with st.spinner("Ricerca nei documenti...."):
             time.sleep(1)
-            prompt, links = gen_augmented_prompt(prompt=prompt_originale, top_k=st.session_state.top_k)        
+            st.session_state.instruction, links = gen_augmented_prompt(prompt=prompt_originale, top_k=st.session_state.top_k)        
+    prompt = formattaPrompt(prompt, st.session_state.systemRole, st.session_state.systemStyle, st.session_state.instruction) 
     print(prompt)
     with st.spinner("Generazione in corso...") :
         time.sleep(1)
@@ -162,8 +167,9 @@ def stream_handler(chat_stream, placeholder) :
 
 def show_source(links) :
     with st.expander("Mostra fonti") :
-        for i, link in enumerate(links) :
-            st.info(f"{link}")
+        for link in links:
+            reference, testo = link
+            st.info(reference + '\n\n'+ testo)
 
 init_state()
 sidebar()