Update app.py

app.py

@@ -34,6 +34,7 @@ async def rag_pdf(pdfs: list, question: str) -> str:
     question_embeddings = np.array([get_text_embedding(question)])
     D, I = index.search(question_embeddings, k = 4)
     retrieved_chunk = [chunks[i] for i in I.tolist()[0]]
+    print(retrieved_chunk)
     text_retrieved = "\n\n".join(retrieved_chunk)
     return text_retrieved
 
@@ -57,6 +58,7 @@ async def ask_mistral(message: str, history: list):
             pdfs_extracted.append(txt)
 
         retrieved_text = rag_pdf(pdfs_extracted, message["text"])
+        print(retrieved_text)
         messages.append(ChatMessage(role = "user", content = retrieved_text + "\n\n" + message["text"]))
     else:
         messages.append(ChatMessage(role = "user", content = message["text"]))