Update app.py

app.py

@@ -191,7 +191,6 @@ def chat_response(
         temperature=temperature,
         chat_history=chat_history,
         prompt_truncation='AUTO',
-        connectors=[{"id": "web-search"}],
         preamble=system_prompt
     )
     
@@ -200,13 +199,19 @@ def chat_response(
     history = history + [[message, ""]]
     
     # Process stream
-    for event in stream:
-        if event.event_type == "text-generation":
-            buffer += event.text
-            formatted_buffer = format_text(buffer)
-            history[-1][1] = formatted_buffer
-            chat_display = format_chat_history(history)
-            yield history, chat_display
+    try:
+        for event in stream:
+            if event.event_type == "text-generation":
+                buffer += event.text
+                formatted_buffer = format_text(buffer)
+                history[-1][1] = formatted_buffer
+                chat_display = format_chat_history(history)
+                yield history, chat_display
+    except cohere.errors.CohereError as e:
+        error_message = f"Error: {str(e)}"
+        history[-1][1] = error_message
+        chat_display = format_chat_history(history)
+        yield history, chat_display
 
 def process_example(example: str) -> tuple:
     """Process example query and return empty history and updated display"""