Update app.py

app.py

@@ -161,9 +161,9 @@ def convert_history_to_cohere_format(history: List[List[str]]) -> List[dict]:
     cohere_history = []
     for user_msg, assistant_msg in history:
         if user_msg:
-            cohere_history.append({"role": "USER", "message": user_msg})
+            cohere_history.append({"role": "User", "message": user_msg})
         if assistant_msg:
-            cohere_history.append({"role": "ASSISTANT", "message": assistant_msg})
+            cohere_history.append({"role": "Chatbot", "message": assistant_msg})
     return cohere_history
 
 def chat_response(
@@ -184,22 +184,22 @@ def chat_response(
     # Convert history to Cohere format
     chat_history = convert_history_to_cohere_format(history)
     
-    # Initialize stream
-    stream = co.chat_stream(
-        model='c4ai-aya-expanse-32b',
-        message=message,
-        temperature=temperature,
-        chat_history=chat_history,
-        prompt_truncation='AUTO',
-        preamble=system_prompt
-    )
-    
     # Initialize buffer for streaming
     buffer = ""
     history = history + [[message, ""]]
     
     # Process stream
     try:
+        # Initialize stream
+        stream = co.chat_stream(
+            model='c4ai-aya-expanse-32b',
+            message=message,
+            temperature=temperature,
+            chat_history=chat_history,
+            prompt_truncation='AUTO',
+            preamble=system_prompt
+        )
+        
         for event in stream:
             if event.event_type == "text-generation":
                 buffer += event.text
@@ -207,7 +207,8 @@ def chat_response(
                 history[-1][1] = formatted_buffer
                 chat_display = format_chat_history(history)
                 yield history, chat_display
-    except cohere.errors.CohereError as e:
+                
+    except Exception as e:
         error_message = f"Error: {str(e)}"
         history[-1][1] = error_message
         chat_display = format_chat_history(history)