image validations

app.py

@@ -400,7 +400,7 @@ def mainApp():
         avatar = C.AI_ICON if role == "assistant" else C.USER_ICON
         with st.chat_message(role, avatar=avatar):
             st.markdown(content)
-            if imagePath:
+            if imagePath and U.isValidImageUrl(imagePath):
                 st.image(imagePath)
             if buttons:
                 __showButtons(buttons)

helpers/sidebar.py

@@ -1,22 +1,10 @@
 import streamlit as st
-import requests
 import datetime as DT
 from helpers.activities import resetActivity, restoreUserActivity, ActivityLogs
 import constants as C
 import utils as U
 
 
-def isValidImageUrl(url):
-    if not url:
-        return False
-    try:
-        imageResponse = requests.head(url, timeout=5)
-        contentType = imageResponse.headers.get('Content-Type', '')
-        return imageResponse.status_code == 200 and contentType.startswith('image/')
-    except Exception:
-        return False
-
-
 def showSidebar():
     with st.sidebar:
         if not ("user" in st.session_state and "token" in st.session_state):
@@ -54,8 +42,8 @@ def showSidebar():
             </style>
             """, unsafe_allow_html=True)
 
-        userAvatar = st.session_state["user"].get("picture", "https://example.com/default-avatar.png")
-        if not isValidImageUrl(userAvatar):
+        userAvatar = st.session_state["user"].get("picture")
+        if not U.isValidImageUrl(userAvatar):
             userAvatar = C.AVATAR_ICON
         userName = st.session_state["user"].get("name", "User")
         userEmail = st.session_state["user"].get("email", "")
@@ -81,7 +69,10 @@ def showSidebar():
         if st.button("\\+ New Story", help="Your current story will be auto-saved 😊", key="save_activities_button", type="primary", use_container_width=True):
             resetActivity()
 
-        userActivityLogs: ActivityLogs = st.session_state.get("userActivitiesLog", [])
+        userActivityLogs: ActivityLogs = [
+            log for log in st.session_state.get("userActivitiesLog", [])
+            if log["id"] != st.session_state.get("activityId")
+        ]
         U.pprint(f"{userActivityLogs=}")
 
         if not userActivityLogs:
@@ -96,7 +87,5 @@ def showSidebar():
                 updatedAt = DT.datetime.fromisoformat(log["updated_at"].replace("Z", "+00:00"))
                 localTime = updatedAt.astimezone().strftime("%b %d, %I:%M %p")
                 activityId = log["id"]
-                if activityId == st.session_state.get("activityId"):
-                    continue
-                if st.button(f"Saved ⌛ {localTime}", key=f"activity_{log['id']}", use_container_width=True):
+                if st.button(f"Saved ⌛ {localTime}", key=f"activity_{activityId}", use_container_width=True):
                     restoreUserActivity(log["id"])

utils.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import datetime as DT
 import pytz
+import requests
 
 FONTS = [
     # "Poppins:ital,wght@0,100;0,200;0,300;0,400;0,500;0,600;0,700;0,800;0,900;1,100;1,200;1,300;1,400;1,500;1,600;1,700;1,800;1,900",
@@ -128,3 +129,14 @@ def applyCommonStyles():
         """,
         unsafe_allow_html=True
     )
+
+
+def isValidImageUrl(url):
+    if not url:
+        return False
+    try:
+        imageResponse = requests.head(url, timeout=5)
+        contentType = imageResponse.headers.get('Content-Type', '')
+        return imageResponse.status_code == 200 and contentType.startswith('image/')
+    except Exception:
+        return False