Update app.py

app.py

@@ -86,7 +86,7 @@ def conversation(qa_chain, message, history):
     response = qa_chain({"question": message, "chat_history": formatted_chat_history})
     response_answer = response["answer"].split("Helpful Answer:")[-1]
     response_sources = response["source_documents"]
-    sources = [(source.page_content.strip(), source.metadata["page"] + 1) for source in response_sources[:3]]
+    sources = [(source.page_content.strip(), source.metadata["page"] + 1) for source in response_sources[:5]]  # Modificato da [:3] a [:5]
     new_history = history + [(message, response_answer)]
     return qa_chain, gr.update(value=""), new_history, *[item for source in sources for item in source]
 
@@ -118,21 +118,21 @@ def demo():
             language_btn = gr.Radio(["Italiano", "Inglese"], label="Lingua", value="Italiano", type="index")
             llm_progress = gr.Textbox(value="Nessuna", label="Inizializzazione catena QA")
             qachain_btn = gr.Button("Inizializza catena di Domanda e Risposta")
-        
+            
         with gr.Tab("Passo 4 - Chatbot"):
             chatbot = gr.Chatbot(height=300)
             with gr.Accordion("Opzioni avanzate - Riferimenti ai documenti", open=False):
-                doc_sources = [gr.Textbox(label=f"Riferimento {i+1}", lines=2, container=True, scale=20) for i in range(3)]
-                source_pages = [gr.Number(label="Pagina", scale=1) for _ in range(3)]
+                doc_sources = [gr.Textbox(label=f"Riferimento {i+1}", lines=2, container=True, scale=20) for i in range(5)]  # Modificato da range(3) a range(5)
+                source_pages = [gr.Number(label="Pagina", scale=1) for _ in range(5)]  # Modificato da range(3) a range(5)
             msg = gr.Textbox(placeholder="Inserisci il messaggio (es. 'Di cosa tratta questo documento?')", container=True)
             submit_btn = gr.Button("Invia messaggio")
-            clear_btn = gr.ClearButton([msg, chatbot], value="Cancella conversazione")
+            clear_btn = gr.Button("Cancella conversazione")
            
         db_btn.click(initialize_database, inputs=[document, slider_chunk_size, slider_chunk_overlap], outputs=[vector_db, collection_name, db_progress])
         qachain_btn.click(initialize_LLM, inputs=[llm_btn, slider_temperature, slider_maxtokens, slider_topk, vector_db], outputs=[qa_chain, llm_progress])
-        
+
         submit_btn.click(conversation, inputs=[qa_chain, msg, chatbot], outputs=[qa_chain, msg, chatbot] + doc_sources + source_pages)
-        msg.submit(conversation, inputs=[qa_chain, msg, chatbot], outputs=[qa_chain, msg, chatbot] + doc_sources + source_pages)
+        msg.submit(conversation, inputs=[qa_chain, msg, chatbot], outputs=[qa_chain, msg, chatbot] + doc_sources + source_pages)        
         
     demo.queue().launch(debug=True)