Merge pull request #53 from banodoco/green-head

inference fixes
banodoco · Nov 30, 2023 · 0e0101f · 0e0101f
2 parents 747775a + 009ebbc
commit 0e0101f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/ui_components/setup.py b/ui_components/setup.py
@@ -157,6 +157,10 @@ def setup_app_ui():
                     if st.session_state['frame_styling_view_type_manual_select'] != None:
                         st.session_state['frame_styling_view_type_manual_select'] = None
 
+                    if 'page' not in st.session_state:
+                        st.session_state["page"] = CreativeProcessType.value_list()[0]
+                        st.session_state["manual_select"] = None
+
                     if st.session_state['frame_styling_view_type'] != "Explorer":
                         pages = CreativeProcessType.value_list()
                     else:
@@ -165,10 +169,6 @@ def setup_app_ui():
                         if st.session_state['page'] != "Key Frames":                            
                             st.session_state["manual_select"] = 0
                             st.session_state['page'] = "Key Frames"
-
-                    if 'page' not in st.session_state:
-                        st.session_state["page"] = pages[0]
-                        st.session_state["manual_select"] = None
 
                     if st.session_state["page"] not in pages:
                         st.session_state["page"] = pages[0]

diff --git a/utils/data_repo/data_repo.py b/utils/data_repo/data_repo.py
@@ -88,7 +88,8 @@ def get_file_from_name(self, name):
         return InternalFileObject(**file) if file else None
 
     def get_file_from_uuid(self, uuid):
-        file = self.db_repo.get_file_from_uuid(uuid).data['data']
+        res = self.db_repo.get_file_from_uuid(uuid)
+        file = res.data['data'] if res.status else None
         return InternalFileObject(**file) if file else None
 
     def get_file_list_from_log_uuid_list(self, log_uuid_list):