sss

utilitis.py

@@ -1,6 +1,6 @@
 import streamlit as st
 from paddleocr import PaddleOCR
-from PIL import ImageDraw, ImageFont
+from PIL import ImageDraw, ImageFont,ImageEnhance
 import torch
 from transformers import AutoProcessor,LayoutLMv3ForTokenClassification
 import numpy as np
@@ -113,6 +113,7 @@ def Get_Json(true_predictions,words):
 #############################################################################
 #############################################################################   
 def Draw(image):
+    image = enhance_image(image,1.3,1.5)
     true_predictions, true_boxes,words = Run_model(image)
     draw = ImageDraw.Draw(image)
 
@@ -210,7 +211,7 @@ def Change_Image(image1,image2):
             st.session_state.current_image = 'image1'
 
         # Button to switch between images
-        if st.sidebar.button('Remove'):
+        if st.sidebar.button('Switcher'):
             if st.session_state.current_image == 'image1':
                 st.session_state.current_image = 'image2'
             else:
@@ -219,4 +220,37 @@ def Change_Image(image1,image2):
         if st.session_state.current_image == 'image1':
             st.image(image1, caption='Output', use_column_width=True)
         else:
-            st.image(image2, caption='Image initiale', use_column_width=True)
+            st.image(image2, caption='Image initiale', use_column_width=True)
+            
+#############################################################################
+#############################################################################
+def enhance_image(image,brightness_factor, contrast_factor):
+    enhancer = ImageEnhance.Brightness(image)
+    brightened_image = enhancer.enhance(brightness_factor)
+    enhancer = ImageEnhance.Contrast(brightened_image)
+    enhanced_image = enhancer.enhance(contrast_factor)
+    return enhanced_image
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+