Update app.py

app.py

@@ -199,41 +199,35 @@ def init():
     msg.send()
 
 @cl.on_message
-def main(message):
-       
-    # msg = cl.Message(content=f"Thinking...")
-    # await msg.send()
-    
+async def main(message):
     chain = cl.user_session.get("chain")
-    cb = cl.LangchainCallbackHandler(
+    cb = cl.AsyncLangchainCallbackHandler(
         stream_final_answer=False, answer_prefix_tokens=["FINAL", "ANSWER"]
     )
     cb.answer_reached = True
-    res = chain.__call__(message, callbacks=[cb], )
-    
-    # print(res.keys()) # keys are "input" and "output"
-    
-    answer = res["output"]
+    res = await chain.acall(message, callbacks=[cb], )
+
+    answer = res["result"]
     source_elements = []
-    # visited_sources = set()
+    visited_sources = set()
 
-    # # Get the documents from the user session
-    # docs = res["source_documents"]
-    # metadatas = [doc.metadata for doc in docs]
-    # all_sources = [m["source"] for m in metadatas]
+    # Get the documents from the 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 the text element referenced in the message
-    #     source_elements.append(
-    #         cl.Text(content="https://www.imdb.com" + source, name="Review URL")
-    #     )
+    for source in all_sources:
+        if source in visited_sources:
+            continue
+        visited_sources.add(source)
+        # Create the text element referenced in the message
+        source_elements.append(
+            cl.Text(content="https://www.imdb.com" + source, name="Review URL")
+        )
 
-    # if source_elements:
-    #     answer += f"\nSources: {', '.join([e.content.decode('utf-8') for e in source_elements])}"
-    # else:
-    #     answer += "\nNo sources found"
+    if source_elements:
+        answer += f"\nSources: {', '.join([e.content.decode('utf-8') for e in source_elements])}"
+    else:
+        answer += "\nNo sources found"
 
-    cl.Message(content=answer, elements=source_elements).send()
+    await cl.Message(content=answer, elements=source_elements).send()