Update app.py

app.py

@@ -26,26 +26,26 @@ st.title("Gen AI Invoice Extraction")
 uploaded_files = st.file_uploader("Choose PDF files", accept_multiple_files=True, type="pdf")
 if uploaded_files:
     # st.write(f'This is {uploaded_files}')
-
-    for pdf_path in uploaded_files:
-      convert_image = pdf_to_image(pdf_path)
-      convert_image.save_image(f'{pdf_path}image')
-      print('one_pdf_converted')
-    
-    all_images = glob.glob(os.path.join(folder_name, '*.jpg'))
-    all_texts = []
-    for image_path in all_images:
-      text = image_to_text(image_path)
-      all_texts.append(text)
-      print('one text appended')
+    if st.button('Extract'):
+        for pdf_path in uploaded_files:
+          convert_image = pdf_to_image(pdf_path)
+          convert_image.save_image(f'{pdf_path}image')
+          print('one_pdf_converted')
+        
+        all_images = glob.glob(os.path.join(folder_name, '*.jpg'))
+        all_texts = []
+        for image_path in all_images:
+          text = image_to_text(image_path)
+          all_texts.append(text)
+          print('one text appended')
+        
+        empty_df = pd.DataFrame()
+        
+        for text in all_texts:
+            extracted_text = extractor(text)
+            task_details_dict = extracted_text.dict()
+            df = pd.DataFrame([task_details_dict])
+            empty_df = pd.concat([empty_df, df])
     
-    empty_df = pd.DataFrame()
     
-    for text in all_texts:
-        extracted_text = extractor(text)
-        task_details_dict = extracted_text.dict()
-        df = pd.DataFrame([task_details_dict])
-        empty_df = pd.concat([empty_df, df])
-
-
-empty_df.to_csv('extracted_data.csv')
+        empty_df.to_csv('extracted_data.csv')