Update app.py

app.py

@@ -3,6 +3,7 @@ import aiohttp
 import os
 import json
 from collections import deque
+import asyncio
 
 TOKEN = os.getenv("HUGGINGFACE_API_TOKEN")
 
@@ -84,9 +85,13 @@ async def respond(
 
 async def chat(message, history, system_message, max_tokens, temperature, top_p):
     response = ""
-    async for chunk in respond(message, history, system_message, max_tokens, temperature, top_p):
-        response = chunk
-        yield response
+    try:
+        async for chunk in respond(message, history, system_message, max_tokens, temperature, top_p):
+            response = chunk
+            yield response
+    except Exception as e:
+        print(f"Chat function error: {str(e)}")
+        yield f"An error occurred in the chat function: {str(e)}"
 
 theme = "Nymbo/Nymbo_Theme"
 
@@ -109,6 +114,5 @@ demo = gr.ChatInterface(
 )
 
 if __name__ == "__main__":
-    import asyncio
     asyncio.run(test_api())  # Run API test
-    demo.queue().launch(max_threads=20)
+    demo.queue().launch(max_threads=20, show_error=True)