Deploying PDF Version v6

app.py

@@ -124,27 +124,38 @@ async def on_chat_start():
 
 @cl.on_message
 async def main(message):
-    chain = cl.user_session.get("chain")  # type: ConversationalRetrievalChain
-    cb = cl.AsyncLangchainCallbackHandler()
-
-    res = await chain.acall(message.content, callbacks=[cb])
-    answer = res["answer"]
-    source_documents = res["source_documents"]  # type: List[Document]
-
-    text_elements = []  # type: List[cl.Text]
-
-    if source_documents:
-        for source_idx, source_doc in enumerate(source_documents):
-            source_name = f"source_{source_idx}"
-            # Create the text element referenced in the message
-            text_elements.append(
-                cl.Text(content=source_doc.page_content, name=source_name)
-            )
-        source_names = [text_el.name for text_el in text_elements]
-
-        if source_names:
-            answer += f"\nSources: {', '.join(source_names)}"
-        else:
-            answer += "\nNo sources found"
-
-    await cl.Message(content=answer, elements=text_elements).send()
+    try:
+        print(f"Received message type: {type(message)}")
+        print(f"Message content: {message.content}")
+        
+        chain = cl.user_session.get("chain")  # type: ConversationalRetrievalChain
+        cb = cl.AsyncLangchainCallbackHandler()
+        
+        res = await chain.acall(message.content, callbacks=[cb])
+        answer = res["answer"]
+        source_documents = res["source_documents"]  # type: List[Document]
+        
+        text_elements = []  # type: List[cl.Text]
+        if source_documents:
+            for source_idx, source_doc in enumerate(source_documents):
+                source_name = f"source_{source_idx}"
+                text_elements.append(
+                    cl.Text(content=source_doc.page_content, name=source_name)
+                )
+            
+            source_names = [text_el.name for text_el in text_elements]
+            if source_names:
+                answer += f"\nSources: {', '.join(source_names)}"
+            else:
+                answer += "\nNo sources found"
+        
+        await cl.Message(content=answer, elements=text_elements).send()
+    
+    except AttributeError as e:
+        print(f"AttributeError occurred: {e}")
+        print(f"Message type: {type(message)}")
+        print(f"Message: {message}")
+        await cl.Message(content=f"An error occurred: {e}").send()
+    except Exception as e:
+        print(f"An unexpected error occurred: {e}")
+        await cl.Message(content=f"An unexpected error occurred: {e}").send()