Update app.py

app.py

@@ -53,8 +53,8 @@ async def init():
     msg = cl.Message(content=f"Building Index...")
     await msg.send()
 
-    # --build FAISS index from csv
-    loader = CSVLoader(file_path="./data/total_faq.csv", source_column="Answer") # columns in csv: answer, question, info_url
+    # --builds FAISS index from csv
+    loader = CSVLoader(file_path="./data/total_faq.csv", source_column="Answer") # columns in csv: Answer, Question, Info_Url
     data = loader.load()
     documents = text_splitter.transform_documents(data)
     store = LocalFileStore("./cache/")
@@ -83,7 +83,7 @@ async def init():
 async def main(message):
     chain = cl.user_session.get("chain")
     cb = cl.AsyncLangchainCallbackHandler(
-        stream_final_answer=False, 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], )
@@ -93,23 +93,24 @@ async def main(message):
     visited_sources = set()
     return
 
-    # --get documents from user session
+    # --documents, user session
     docs = res["source_documents"]
     metadatas = [doc.metadata for doc in docs]
     all_sources = [m["source"] for m in metadatas]
 
-    for source in all_sources:
-        if source in visited_sources:
-            continue
-        visited_sources.add(source)
-        # --create text element referenced in message
-        source_elements.append(
-            cl.Text(content="https://www.daysoff.no" + source, name="Info_Url")
-            #cl.Text(content="https://www.imdb.com" + source, name="Review URL")
-        )
+    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.decode('utf-8') for e in source_elements])}"
+        answer += f"\nSources: {', '.join([e.content for e in source_elements])}"
     else:
         answer += "\nNo sources found"