figures list displayed

app.py

@@ -5,6 +5,7 @@ import numpy as np
 import cv2
 import os
 import io
+from PIL import Image
 import shutil
 
 import time
@@ -112,7 +113,7 @@ if 'page_count' in st.session_state:
 
         st.session_state.color_image_list = []
         st.session_state.gray_image_np_list = []
-        pdf_figures_image_list=[]
+        st.session_state.pdf_figures_image_list=[]
         pdf_tables_image_list=[]
         st.session_state.pdf_text_list=[]
 
@@ -133,7 +134,7 @@ if 'page_count' in st.session_state:
           print("index="+str(index))
 
           figures_image_list,tables_image_list,text=utils.gray_pdf_image_np_to_text(index,gray_pdf_image_np, debug=True)
-          pdf_figures_image_list.append(figures_image_list)
+          st.session_state.pdf_figures_image_list.append(figures_image_list)
           pdf_tables_image_list.append(tables_image_list)
           st.session_state.pdf_text_list.append(text)
           st.session_state.extracted_text=st.session_state.extracted_text+f"<Page {index+1} start>\n" + text + f"\n<Page {index+1} end>\n>"
@@ -151,8 +152,17 @@ if 'page_count' in st.session_state:
                            data=string_buffer.getvalue(),
                            file_name=txt_file_path,
                            mime="text/plain")
-        for index,pdf_text in enumerate(st.session_state.pdf_text_list):
-            st.write(f"Page {index+1} \n\n {pdf_text}\n")
+        #  st.image(Image.fromarray(bgr_image))
+        # for index,pdf_text in enumerate(st.session_state.pdf_text_list):
+        for index, gray_pdf_image_np in enumerate(st.session_state.gray_image_np_list):
+            st.write(f"Page {index+1} \n\n {st.session_state.pdf_text_list[index]}\n")
+            if not st.session_state.pdf_figures_image_list[index]:
+                st.write("no figures")
+            else:
+                for pdf_figure_text_image in st.session_state.pdf_figures_image_list[index]:
+                    st.write(pdf_figure_text_image[0])
+                    st.image(Image.fromarray(pdf_figure_text_image[1]))
+
 
 # for index, gray_pdf_image_np in enumerate(st.session_state.gray_image_np_list[0:5], start=0):
 #   print("index="+str(index))