addui

ui.py

@@ -18,6 +18,7 @@ class OCRUI:
         self.current_page = None
         self.API_URL = "http://s15.serv00.com:9081/compareAnalyze"
         self.stripe_remover = StripeRemover()  # Initialize here
+        self.processed_image = None
         
     async def send_to_external_api(self, file_bytes):
         async with aiohttp.ClientSession() as session:
@@ -123,9 +124,11 @@ class OCRUI:
                     with st.spinner('处理图像中...'):
                         processed_img = self.process_image(self.current_page, preprocess_method)
                         if processed_img is not None:
+                            self.processed_image = processed_img
                             # Show preview
                             st.sidebar.image(processed_img, caption=f"预处理结果 - {preprocess_method}", 
                                         width=200)
+                            self.cols[0].image(processed_img)
                             
                             # Encode and send
                             encode_params = [cv2.IMWRITE_JPEG_QUALITY, 50]