Update app.py

app.py

@@ -106,6 +106,8 @@ from chainlit.auth import create_jwt
 from dotenv import load_dotenv
 import chainlit as cl
 import uvicorn
+import asyncio
+from socket import gaierror
 # Load environment variables from .env file
 load_dotenv()
 
@@ -172,14 +174,26 @@ async def on_message(message: cl.Message):
     msg = cl.Message(content="")
     await msg.send()
 
-    stream = await client.chat.completions.create(
-        messages=message_history, stream=True, **settings
-    )
-
-    async for part in stream:
-        if token := part.choices[0].delta.content or "":
-            await msg.stream_token(token)
+    try:
+        stream = await client.chat.completions.create(
+            messages=message_history, stream=True, **settings
+        )
+
+        async for part in stream:
+            if token := part.choices[0].delta.content or "":
+                await msg.stream_token(token)
+
+    except gaierror as e:
+        logger.error(f"Network error during OpenAI API call: {e}")
+        await msg.update(content="Sorry, there was a network error.")
+    except asyncio.TimeoutError:
+        logger.error("Timeout error during OpenAI API call.")
+        await msg.update(content="Sorry, the request timed out.")
+    except Exception as e:
+        logger.error(f"Unexpected error during OpenAI API call: {e}")
+        await msg.update(content="Sorry, an unexpected error occurred.")
 
     message_history.append({"role": "assistant", "content": msg.content})
     await msg.update()
     logger.info("Message processed and response sent.")
+