fix streamlit_app.py

geezacoleman · Sep 26, 2023 · 20c044c · 20c044c
1 parent 2d381dd
commit 20c044c
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/demos/streamlit_app.py b/demos/streamlit_app.py
@@ -121,7 +121,13 @@ def main():
                 }
                 config.update(updated_config)
 
-    st.session_state.current_file_idx = (len(uploaded_files) - 1) if current_file_idx >= len(uploaded_files) else current_file_idx
+    if current_file_idx >= len(uploaded_files) and current_file_idx != 0:
+        st.session_state.current_file_idx = (len(uploaded_files) - 1)
+    elif current_file_idx < 0:
+        st.session_state.current_file_idx = 0
+    else:
+        st.session_state.current_file_idx = current_file_idx
+
     if uploaded_files and 0 <= current_file_idx:
         current_file_idx = st.session_state.get("current_file_idx", 0)
         file_type = get_file_type(uploaded_files[current_file_idx])