Update app.py

app.py

@@ -83,21 +83,39 @@ async def init():
 async def main(message):
     chain = cl.user_session.get("chain")
     cb = cl.AsyncLangchainCallbackHandler(
-        stream_final_answer=True, answer_prefix_tokens=["FINAL", "ANSWER"]
+        stream_final_answer=True, 
+        answer_prefix_tokens=["FINAL", "ANSWER"]
     )
     cb.answer_reached = True
-    res = await chain.acall(message, callbacks=[cb], )
+    res = await chain.acall(message, callbacks=[cb])
 
     answer = res["result"]
     source_elements = []
     visited_sources = set()
-    #return
 
     # --documents, user session
     docs = res["source_documents"]
     metadatas = [doc.metadata for doc in docs]
     all_sources = [m["source"] for m in metadatas]
     
+    for doc, metadata in zip(docs, metadatas):
+        row_index = metadata.get("row_index", -1)  # --when row_index@metadata
+        if row_index in [2, 8, 14]:  # --incl. only rows 2, 8, and 14
+            source = metadata.get("source", "")
+            if source and source not in visited_sources:
+                visited_sources.add(source)
+                source_elements.append(
+                    cl.Text(content="https://www.daysoff.no" + source, name="Info_Url")
+                )
+
+    if source_elements:
+        answer += f"\nSources: {', '.join([e.content for e in source_elements])}"
+    else:
+        answer += "\nNo sources found"
+
+    await cl.Message(content=answer, elements=source_elements).send()
+
+    
 """
     for source in all_sources:
         if source in visited_sources:
@@ -133,4 +151,3 @@ async def main(message):
         answer += "\nNo sources found"
 
     await cl.Message(content=answer, elements=source_elements).send()
-