Update app.py

app.py

@@ -25,12 +25,11 @@ async def start_chat():
 
     await cl.Message(content="Welcome! Please upload a document to begin.").send()
 
-# 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]
+# Handling user queries and expecting file upload as part of the message
+@cl.on_message
+async def main(message: cl.Message):
+    if message.files:  # Check if the message contains any files
+        uploaded_file = message.files[0]
         file_path = uploaded_file['path']
 
         # Extract text from the uploaded document
@@ -40,36 +39,33 @@ async def on_file_upload(files: list):
         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
 
-# 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()
+        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()