Update app.py

app.py

@@ -4,6 +4,7 @@ import os
 import re
 import uuid
 from functools import partial
+from urllib.error import HTTPError
 
 
 
@@ -28,21 +29,23 @@ def generate_response(user_message, history=None):
     stream = co.chat_stream(message=user_message, conversation_id=cid, model='command-r', connectors=[], temperature=0.3)
     
     output = ""
+    
     for idx, response in enumerate(stream):
-        if response.event_type == "text-generation":
-            output += response.text
-        
-        if idx == 0:
-            history.append(" " + output)
+        if response.status_code == HTTPStatus.OK:
+            if response.event_type == "text-generation":
+                output += response.text
+            if idx == 0:
+                history.append(" " + output)
+            else:
+                history[-1] = output
+            chat = [
+                (history[i].strip(), history[i + 1].strip())
+                for i in range(0, len(history) - 1, 2)
+            ] 
+            yield chat, history
         else:
-            history[-1] = output
-
-        chat = [
-            (history[i].strip(), history[i + 1].strip())
-            for i in range(0, len(history) - 1, 2)
-        ]
-        
-        yield chat, history
+            raise HTTPError(f"Request id: {response.request_id}, Status code: {response.status_code}, 
+                            error code: {response.code}, error message: {response.message}")
 
     return chat, history