Update app.py

app.py

@@ -291,7 +291,8 @@ def get_messages(user_email):
         messages = db_firestore.collection('messages').where('receiver_email', '==', user_email).order_by('timestamp').get()
         return messages  # Return the Firestore query snapshot
     except Exception as e:
-        return f"Failed to retrieve messages: {str(e)}"
+        st.error(f"Failed to retrieve messages: {str(e)}")
+        return None
 
 # Streamlit interface for sending messages
 def send_message_ui():
@@ -308,11 +309,12 @@ def view_messages_ui():
     st.title("Messages")
     user_email = st.session_state.auth_state["user"].email
     messages = get_messages(user_email)
-    for message in messages:
-        message_dict = message.to_dict()  # Convert Firestore document to dictionary
-        st.write(f"From: {message_dict['sender_email']}")
-        st.write(f"Message: {message_dict['message_content']}")
-        st.write("---")
+    if messages is not None:
+        for message in messages:
+            message_dict = message.to_dict()  # Convert Firestore document to dictionary
+            st.write(f"From: {message_dict['sender_email']}")
+            st.write(f"Message: {message_dict['message_content']}")
+            st.write("---")
 #end of messaging section 
 
 def tour_guide_ui():