Update app.py

app.py

@@ -11,8 +11,9 @@ from langchain.schema import SystemMessage, HumanMessage
 from PyPDF2 import PdfReader
 import chainlit as cl
 
-# Set up API key
-os.environ["OPENAI_API_KEY"] = "your-api-key-here"  # You'll need to handle this securely
+# Check if the API key is set
+if not os.getenv("OPENAI_API_KEY"):
+    raise ValueError("OPENAI_API_KEY environment variable is not set")
 
 # Set up prompts
 system_template = "Use the following context to answer a user's question. If you cannot find the answer in the context, say you don't know the answer."
@@ -85,7 +86,11 @@ async def main(message: cl.Message):
         return
 
     msg = cl.Message(content="")
-    async for chunk in pipeline.arun_pipeline(message.content):
-        await msg.stream_token(chunk)
+    try:
+        async for chunk in pipeline.arun_pipeline(message.content):
+            await msg.stream_token(chunk)
+    except Exception as e:
+        await cl.Message(content=f"An error occurred: {str(e)}").send()
+        return
 
     await msg.send()