Update app.py

app.py

@@ -2,17 +2,20 @@ import streamlit as st
 import cv2
 import Scraper
 import Manager
+import os
 
 
 
 loc_link = st.text_input('LOC Link')
-Scraper.download_loc('https://tile.loc.gov/storage-services/service/ll/usrep/usrep200/usrep200361/usrep200361.pdf')
-Manager.run('200_361')
+run = st.button("Run")
+if run:
+    with st.spinner("Downloading"):
+        Scraper.download_loc('https://tile.loc.gov/storage-services/service/ll/usrep/usrep200/usrep200361/usrep200361.pdf')
+        Manager.run('200_361')
 
-col1, col2, col3, col4 = st.columns(4)
-image = cv2.imread('PDF Cases/19-896_2135/10-processed.png')
-with col1:
-    st.image(image)
-image = cv2.imread('PDF Cases/200_361/0-processed.png')
-with col3:
-    st.image(image)
+cols = st.columns(4)
+image_filenames = [f for f in os.listdir('PDF Cases/200_361') if "procsessed" in f]
+for (i,f) in enumerate(image_filenames):
+    image = cv2.imread('PDF Cases/200_361/' + f)
+    with cols[i%4]:
+        st.image(image)