Update app.py

app.py

@@ -93,7 +93,7 @@ if prompt := st.chat_input("Message..."):
     if not api_key:
         st.error("API key required")
         st.stop()
-    
+
     st.session_state.messages.append({"role": "user", "content": prompt})
     with st.chat_message("user", avatar=USER_PFP):
         st.markdown(prompt)
@@ -101,10 +101,18 @@ if prompt := st.chat_input("Message..."):
     try:
         co = cohere.ClientV2(api_key)
         with st.chat_message("assistant", avatar=AI_PFP):
-            response = co.chat(model=selected_model, messages=st.session_state.messages)
-            reply = response.text if hasattr(response, 'text') else "Error processing response"
+            response = co.chat(
+                model=selected_model,
+                messages=st.session_state.messages
+            )
+            if hasattr(response, "message") and hasattr(response.message, "content") and hasattr(response.message.content, "text"):
+                reply = response.message.content.text
+            else:
+                st.write(response)
+                reply = "❗️Couldn't find response.message.content.text in the Cohere response."
             st.markdown(reply)
+
         st.session_state.messages.append({"role": "assistant", "content": reply})
-    
+
     except Exception as e:
-        st.error(f"Error: {str(e)}")
+        st.error(f"Error: {str(e)}")