Update app.py

app.py

@@ -70,8 +70,8 @@ def sidebar():
         #st.session_state.instruction = st.text_area("Istruzioni", st.session_state.instruction, help='Testo di riferimento sul quale il modello si basa per generare il testo')
         
         st.session_state.rag_enabled = st.session_state.selected_option.get('tipo', '')=='RAG'
-        if st.session_state.selected_option_key == 'Manuali':
-            st.session_state.top_k = st.slider(label="Documenti da ricercare", min_value=1, max_value=20, value=4, disabled=not st.session_state.rag_enabled)
+        if st.session_state.rag_enabled: 
+            st.session_state.top_k = st.slider(label="Documenti da considerare", min_value=1, max_value=20, value=4, disabled=not st.session_state.rag_enabled)
         st.markdown("---")
     
     def model_settings() :
@@ -127,11 +127,12 @@ def gen_augmented_prompt(prompt, top_k) :
     
     links = []
     context = ''
-    NomeCartellaOriginariaDB = 'Documenti_2\\'
+    NomeCartellaOriginariaDB = 'Documenti\'
     for doc in docs:
-        testo = doc.page_content.replace('\n', ' ')
+        testo = doc.page_content.replace('\n\n', '\n')
+        testo = doc.page_content.replace('***', '')
         context += testo + '\n\n\n'
-        reference = doc.metadata["source"].replace(NomeCartellaOriginariaDB, '') + ' (Pag. ' + str(doc.metadata["page"]) + ')'
+        reference = doc.metadata["source"].replace(NomeCartellaOriginariaDB, '') + ' (Pag. ' + str(doc.metadata["page"]+1) + ')'
         links.append((reference, testo))
     generated_prompt = f"""
     A PARTIRE DAL SEGUENTE CONTESTO: {docs},