Update app.py

app.py

@@ -108,36 +108,62 @@ async def main(message):
 
     # --append source(s), specific rows only
     for doc, metadata in zip(docs, metadatas):
-        row_index = metadata.get("row_index", -1) 
-        source = metadata.get("source", "") 
-
-        if row_index in [2, 8, 14] and 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")
-            )
-"""
+    row_index = metadata.get("row_index", -1) 
+    source = metadata.get("source", "") 
+
+    if row_index in [2, 8, 14] and 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])}"
+        await cl.Message(content=answer, elements=source_elements).send()
+        return
         
     else:
-        answer += "\nNo sources found"
+        #answer = f"{answer}\nNo sources found"  
+        await cl.Message(content=f"No sources found").send()
+    
+    #await cl.Message(content=answer, elements=source_elements).send()
+
+
 
-    await cl.Message(content=answer, elements=source_elements).send()
-    return
 
 """
     if source_elements:
         answer += f"\nSources: {', '.join([e.content for e in source_elements])}"
-        await cl.Message(content=answer, elements=source_elements).send()
-        return
         
     else:
         answer += "\nNo sources found"
-        await cl.Message(content=answer).send()
-        return
 
-    
+    await cl.Message(content=answer, elements=source_elements).send()
+    return
+
+"""
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
 """
     for source in all_sources:
         if source in visited_sources: