Update app.py

app.py

@@ -76,7 +76,8 @@ def _create_embeddings_and_save(user: User, chunks: any) -> tuple[int, FAISS]:
 def ask_question(user: User, question: str, vector_store : FAISS) -> tuple[str, int]:
 
     docs = vector_store.similarity_search(question,k = 3)
-    retrieved_chunks = docs[0].page_content + docs[1].page_content + docs[2].page_content
+    x = min(len(docs), 3)
+    retrieved_chunks = "".join([docs[i].page_content for i in range(x)])
     prompt = f'Question: "{question}"\nContext: "{retrieved_chunks}"'
 
     try: