Merge pull request #63 from banodoco/green-head

Fix
banodoco · Dec 23, 2023 · 51732c6 · 51732c6
2 parents 09aea29 + c58575d
commit 51732c6
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 11 deletions.
diff --git a/ui_components/components/frame_styling_page.py b/ui_components/components/frame_styling_page.py
@@ -31,7 +31,7 @@ def frame_styling_page(shot_uuid: str, h2,data_repo,shot,timing_list, project_se
 
         st.markdown("#### There are no frames present in this shot yet.")
 
-
+        
 
 
     else:

diff --git a/ui_components/widgets/sidebar_logger.py b/ui_components/widgets/sidebar_logger.py
@@ -108,7 +108,7 @@ def sidebar_logger(shot_uuid):
                                 st.session_state['main_view_type'] = "Creative Process"
                                 st.session_state['frame_styling_view_type_index'] = 0
                                 st.session_state['frame_styling_view_type'] = "Explorer"
-                                st.session_state['change_view_type'] = False
+
                                 st.rerun()
 
 

diff --git a/ui_components/widgets/timeline_view.py b/ui_components/widgets/timeline_view.py
@@ -5,7 +5,7 @@
 from utils import st_memory
 
 
-def timeline_view(shot_uuid, stage, show_frame_upload=True):
+def timeline_view(shot_uuid, stage):
     data_repo = DataRepo()
     shot = data_repo.get_shot_from_uuid(shot_uuid)
     shot_list = data_repo.get_shot_list(shot.project.uuid)
@@ -34,11 +34,11 @@ def timeline_view(shot_uuid, stage, show_frame_upload=True):
                 shot_video_element(shot.uuid)
             if (idx + 1) % items_per_row == 0 or idx == len(shot_list) - 1:
                 st.markdown("***")
-            if show_frame_upload == True:
-                if idx == len(shot_list) - 1:
-                    with grid[(idx + 1) % items_per_row]:
-                        st.markdown("### Add new shot")
-                        add_new_shot_element(shot, data_repo)
+            # if stage isn't 
+            if idx == len(shot_list) - 1:
+                with grid[(idx + 1) % items_per_row]:
+                    st.markdown("### Add new shot")
+                    add_new_shot_element(shot, data_repo)
 
 
 

diff --git a/utils/common_utils.py b/utils/common_utils.py
@@ -15,7 +15,7 @@
 def set_default_values(timing_list, shot_uuid, data_repo):
 
     if "page" not in st.session_state:
-        st.session_state['strength'] 
+        st.session_state['page'] = "Explore"
 
     if "strength" not in st.session_state:
         st.session_state['strength'] = DefaultProjectSettingParams.batch_strength
@@ -36,8 +36,6 @@ def set_default_values(timing_list, shot_uuid, data_repo):
         st.session_state['frame_styling_view_type'] = "Generate"
         st.session_state['frame_styling_view_type_index'] = 0
 
-    if st.session_state['change_view_type'] == True:  
-        st.session_state['change_view_type'] = False
 
     if "explorer_view" not in st.session_state:
         st.session_state['explorer_view'] = "Explorations"