Update app.py

app.py

@@ -25,47 +25,51 @@ async def start_chat():
 
     await cl.Message(content="Welcome! Please upload a document to begin.").send()
 
-# This handles incoming files manually (instead of on_file_upload)
-@cl.on_message
-async def main(message: cl.Message):
-    if message.attachments:  # Check if the message contains file attachments
-        uploaded_file = message.attachments[0]
+# Handling file upload events
+@cl.on_file_upload  # Using the proper decorator for file upload handling
+async def on_file_upload(files: list):
+    if files:
+        # Get the first uploaded file
+        uploaded_file = files[0]
         file_path = uploaded_file['path']
 
         # Extract text from the uploaded document
         file_content = await extract_text_from_file(file_path)
-        
+
         # Store document content in user session
         cl.user_session.set("document_content", file_content)
 
         await cl.Message(content=f"Document '{uploaded_file['name']}' uploaded successfully! You can now ask questions based on the document content.").send()
-    else:
-        document_content = cl.user_session.get("document_content", "")
-        
-        if not document_content:
-            await cl.Message(content="Please upload a document first.").send()
-            return
 
-        settings = cl.user_session.get("settings")
-        client = AsyncOpenAI()
+# Handling user queries
+@cl.on_message
+async def main(message: cl.Message):
+    document_content = cl.user_session.get("document_content", "")
+    
+    if not document_content:
+        await cl.Message(content="Please upload a document first.").send()
+        return
+
+    settings = cl.user_session.get("settings")
+    client = AsyncOpenAI()
 
-        # Create the prompt for OpenAI based on user message and document content
-        prompt = f"Document Content: {document_content}\n\nUser Query: {message.content}"
-        
-        msg = cl.Message(content="")
+    # Create the prompt for OpenAI based on user message and document content
+    prompt = f"Document Content: {document_content}\n\nUser Query: {message.content}"
+    
+    msg = cl.Message(content="")
 
-        # Send prompt to OpenAI and stream response
-        async for stream_resp in await client.chat.completions.create(
-            model=settings["model"],
-            messages=[{"role": "system", "content": "Answer based on the provided document."},
-                      {"role": "user", "content": prompt}],
-            stream=True,
-            **settings,
-        ):
-            token = stream_resp.choices[0].delta.content
-            if not token:
-                token = ""
-            await msg.stream_token(token)
+    # Send prompt to OpenAI and stream response
+    async for stream_resp in await client.chat.completions.create(
+        model=settings["model"],
+        messages=[{"role": "system", "content": "Answer based on the provided document."},
+                  {"role": "user", "content": prompt}],
+        stream=True,
+        **settings,
+    ):
+        token = stream_resp.choices[0].delta.content
+        if not token:
+            token = ""
+        await msg.stream_token(token)
 
-        # Send final response
-        await msg.send()
+    # Send final response
+    await msg.send()