Update src/streamlit_app.py

src/streamlit_app.py

@@ -199,21 +199,22 @@ if 'model' not in st.session_state:
     st.session_state.which_pic=1
 
     st.session_state.calculating=True
+    st.session_state.next=True
 
 
     
 # Button to fetch a new image
-col1, col2, col3 = st.columns([1,1,1])
-
-def refresh_start():
-    with col1: 
-        if st.button("Start",key="started",disabled=st.session_state.started):
-            if not st.session_state.started:
-                fetch_image()
-                st.session_state.started=True
-                st.session_state.calculating=False
+col1, col2, col3, col4 = st.columns([1,1,1,1])
+
+
+with col1: 
+    if st.button("Start",disabled=st.session_state.started):
+        if not st.session_state.started:
+            fetch_image()
+            st.session_state.started=True
+            st.session_state.calculating=False
             
-refresh_start()
+
 
 
 # Display the current image
@@ -291,10 +292,15 @@ if st.session_state.label_input in ['cat', 'dog']:
     st.info(f"You clicked on last picture (picture {st.session_state.which_pic}): {label_input}")
     st.session_state.which_pic=st.session_state.which_pic+1
     
-    st.session_state.started=False
-    refresh_start()
-
+    st.session_state.next=False
     
+
+ with col4: 
+    if st.button("next",disabled=st.session_state.next):
+        if not st.session_state.next:
+            fetch_image()
+            st.session_state.next=True
+            st.session_state.calculating=False