add: captions input

src/streamlit_app.py

@@ -35,6 +35,8 @@ if uploaded_pdf:
         st.session_state.page_idx = 0
     if "cropped_images" not in st.session_state:
         st.session_state.cropped_images = []
+    if "captions" not in st.session_state:
+        st.session_state.captions = []
 
     total_pages = doc.page_count
     page_idx = st.session_state.page_idx % total_pages
@@ -60,8 +62,11 @@ if uploaded_pdf:
     image = Image.open(io.BytesIO(pix.tobytes("png")))
     cropped_image = st_cropper(image, realtime_update=True)
     st.image(cropped_image)
+    caption = st.text_input("Caption", key="caption")
     if st.button("Save"):
         padded_image = get_padded_image(image, cropped_image)
         st.session_state.cropped_images.append(padded_image)
+        st.session_state.captions.append(caption)
         print(f"{len(st.session_state.cropped_images)=}")
         print(f"{st.session_state.cropped_images[-1].size=}")
+        print(f"{st.session_state.captions[-1]=}")