Update app.py

app.py

@@ -1,25 +1,30 @@
-# You can find this code for Chainlit python streaming here (https://docs.chainlit.io/concepts/streaming/python)
-
-# OpenAI Chat completion
 import os
 from openai import AsyncOpenAI  # importing openai for API usage
 import chainlit as cl  # importing chainlit for our app
 from chainlit.prompt import Prompt, PromptMessage  # importing prompt tools
 from chainlit.playground.providers import ChatOpenAI  # importing ChatOpenAI tools
 from dotenv import load_dotenv
+import asyncio
+import textract  # importing textract for document processing
 
+# Load OpenAI API key from environment variables
 api_key = os.getenv("OPENAI_API_KEY")
 
-# ChatOpenAI Templates
-system_template = """You are a helpful assistant who always speaks in a pleasant tone!
+# Templates for ChatOpenAI interaction
+system_template = """You are a helpful assistant who always speaks in a pleasant tone and answers based on the provided document!
 """
 
 user_template = """{input}
-Think through your response step by step.
+Use the document content to respond to the user query step by step.
 """
 
+# Function to extract text from uploaded documents
+async def extract_text_from_file(file):
+    # Extract text from the uploaded file using textract
+    return textract.process(file).decode('utf-8')
 
-@cl.on_chat_start  # marks a function that will be executed at the start of a user session
+# Function to start the chat session
+@cl.on_chat_start  
 async def start_chat():
     settings = {
         "model": "gpt-3.5-turbo",
@@ -32,14 +37,35 @@ async def start_chat():
 
     cl.user_session.set("settings", settings)
 
-
-@cl.on_message  # marks a function that should be run each time the chatbot receives a message from a user
+    # Welcome message with file upload option
+    await cl.Message(content="Welcome! Please upload a document to begin.").send()
+
+# Function to handle uploaded files
+@cl.on_file_upload
+async def on_file_upload(files: list):
+    # Handle the uploaded files, assuming it's the first file
+    if files:
+        file = files[0]
+        file_content = await extract_text_from_file(file.path)
+
+        # Save document content in session for later use
+        cl.user_session.set("document_content", file_content)
+        
+        # Inform the user that the document was successfully uploaded
+        await cl.Message(content=f"Document '{file.name}' uploaded successfully! You can now ask questions based on the document content.").send()
+
+# Function to handle user messages
+@cl.on_message  
 async def main(message: cl.Message):
     settings = cl.user_session.get("settings")
+    document_content = cl.user_session.get("document_content", "")
 
-    client = AsyncOpenAI()
+    if not document_content:
+        # If no document is uploaded, prompt the user to upload one
+        await cl.Message(content="Please upload a document first.").send()
+        return
 
-    print(message.content)
+    client = AsyncOpenAI()
 
     prompt = Prompt(
         provider=ChatOpenAI.id,
@@ -55,12 +81,10 @@ async def main(message: cl.Message):
                 formatted=user_template.format(input=message.content),
             ),
         ],
-        inputs={"input": message.content},
+        inputs={"input": message.content, "document": document_content},
         settings=settings,
     )
 
-    print([m.to_openai() for m in prompt.messages])
-
     msg = cl.Message(content="")
 
     # Call OpenAI
@@ -78,3 +102,4 @@ async def main(message: cl.Message):
 
     # Send and close the message stream
     await msg.send()
+