Update app.py

app.py

@@ -64,7 +64,7 @@ def main(message: str):
 
         # Create embeddings and vector store
         embeddings = OpenAIEmbeddings()
-        vectorstore = FAISS.from_texts(chunks, embeddings)
+        vectorstore = FAISS.from_texts(chunks, embeddings, metadatas=[{"title": selected_paper.title, "link": selected_paper.entry_id}] * len(chunks))
 
         # Create the conversational chain
         memory = ConversationBufferMemory(
@@ -95,8 +95,8 @@ def main(message: str):
             response = qa_chain({"question": message})
             answer = response["answer"]
 
-            # Handling the sources and formatting the response
-            sources = "\n".join([f"- {doc.metadata['source']}" for doc in response.get("source_documents", [])])
+            # Check if source metadata is present, otherwise handle gracefully
+            sources = "\n".join([f"- {doc.metadata.get('title', 'Unknown title')} ({doc.metadata.get('link', 'No link')})" for doc in response.get("source_documents", [])])
             if sources:
                 answer += f"\n\nSources:\n{sources}"