Fixed datetime format

GooeyAI · Feb 12, 2024 · e985052 · e985052
1 parent cd991b5
commit e985052
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/daras_ai_v2/base.py b/daras_ai_v2/base.py
@@ -1519,14 +1519,12 @@ def _render_after_output(self):
         if "seed" in self.RequestModel.schema_json():
             seed = st.session_state.get("seed")
             caption += f' with seed <span style="color: black;">{seed}</span> '
-        try:
-            format_created_at = st.session_state.get(
-                "created_at", datetime.datetime.today()
-            ).strftime("%d %b %Y %-I:%M%p")
-        except:
-            format_created_at = format_created_at = st.session_state.get(
-                "created_at", datetime.datetime.today()
-            )
+        created_at = st.session_state.get(
+            StateKeys.created_at, datetime.datetime.today()
+        )
+        if not isinstance(created_at, datetime.datetime):
+            created_at = datetime.datetime.fromisoformat(created_at)
+        format_created_at = created_at.strftime("%d %b %Y %-I:%M%p")
         caption += f' at <span style="color: black;">{format_created_at}</span>'
         st.caption(caption, unsafe_allow_html=True)