Update app.py

app.py

@@ -86,9 +86,16 @@ 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[:5]]  # Modificato da [:3] a [:5]
+    sources = [(source.page_content.strip(), source.metadata["page"] + 1) for source in response_sources[:5]]
     new_history = history + [(message, response_answer)]
-    return qa_chain, gr.update(value=""), new_history, *[item for source in sources for item in source]
+    
+    # Ensure we always return 5 sources and 5 pages
+    source_texts = [source[0] for source in sources] + [''] * (5 - len(sources))
+    source_pages = [source[1] for source in sources] + [0] * (5 - len(sources))
+    
+    return (qa_chain, gr.update(value=""), new_history, 
+            *source_texts[:5],  # Unpack exactly 5 source texts
+            *source_pages[:5])  # Unpack exactly 5 source pages
 
 def demo():
     with gr.Blocks(theme="base") as demo: