sd

App.py

@@ -1,14 +1,11 @@
 import streamlit as st
-from utilitis import Draw, Add_Results, Change_Image
+from utilitis import Draw,Change_Image,check_if_changed,Update
 from PIL import Image
 import time 
 
-def check_if_changed(original_values, updated_values):
-    for key, value in original_values.items():
-        if updated_values[key] != value:
-            return True
-    return False
+    
 
+st.set_page_config(layout='wide')
 st.title("Bienvenue à Textra Web App")
 st.markdown("### Drag and Drop votre facture ici:")
 st.write("(PNG, JPG, JPEG)")
@@ -26,35 +23,20 @@ if uploaded_file is not None:
     # Process the image and retrieve results
     image, Results = process_image(uploaded_file)
     Change_Image(image,image_initiale)
+    # Some Initializations
+    sauvgarder_button = st.sidebar.empty()
+    success_message = st.sidebar.empty()
     st.sidebar.title('Results')
-    st.write(Results)
-    # Define text inputs with initial values
-    text_fourni = st.sidebar.text_input("Fournisseur", value=Results["Fournisseur"])
-    text_InvDate = st.sidebar.text_input("Date Facture", value=Results["Date Facture"])
-    text_InvNum = st.sidebar.text_input("Numéro de facture", value=Results["Numéro de facture"])
-    text_TT = st.sidebar.text_input("Total HT", value=Results["Total HT"])
-    text_TTC = st.sidebar.text_input("TTC", value=Results["Total TTC"])
-    text_TVA = st.sidebar.text_input("TVA", value=Results["TVA"])
-    
-    st.write("Predicted Text:", text_fourni, text_InvDate, text_InvNum, text_TT, text_TVA, text_TTC)
-    
-    New_results = {
-        "Fournisseur": text_fourni,
-        "Date Facture": text_InvDate,
-        "Numéro de facture": text_InvNum,
-        "Total HT": text_TT,
-        "TVA": text_TVA,
-        "Total TTC": text_TTC
-    }
-    
+    # Get Track of User Modeifications :
+    New_results = Update(Results)
     # Check if any input has been changed
     if check_if_changed(Results,New_results):
-        st.write(check_if_changed(Results,New_results))
-        # Add a button to save changes
-        if st.sidebar.button("Sauvegarder"):
-            # Get updated values from text inputs
-            updated_results = New_results
-            success_message = st.sidebar.empty()
+        st.write(check_if_changed(Results,New_results))        
+        if sauvgarder_button.button("Sauvegarder"):
             success_message.success("Les résultats ont été sauvegardés avec succès !")
             time.sleep(1)
             success_message.empty()
+            st.write(New_results)
+ 
+            
+            

utilitis.py

@@ -65,7 +65,8 @@ def Encode(image):
     encoding = processor(image, words, boxes=boxes,return_offsets_mapping=True,truncation=True, max_length=512, padding="max_length", return_tensors="pt")
     offset_mapping = encoding.pop('offset_mapping')
     return encoding, offset_mapping,words
-
+#############################################################################
+#############################################################################
 def unnormalize_box(bbox, width, height):
      return [
          width * (bbox[0] / 1000),
@@ -73,7 +74,8 @@ def unnormalize_box(bbox, width, height):
          width * (bbox[2] / 1000),
          height * (bbox[3] / 1000),
      ]
-    
+#############################################################################
+#############################################################################  
 def Run_model(image):
     encoding,offset_mapping,words = Encode(image)
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
@@ -94,7 +96,8 @@ def Run_model(image):
     true_boxes = [unnormalize_box(box, width, height) for idx, box in enumerate(token_boxes) if not is_subword[idx]]
     return true_predictions,true_boxes,words
 
-
+#############################################################################
+#############################################################################
 def Get_Json(true_predictions,words):
     Results = {}
     i = 0
@@ -107,7 +110,8 @@ def Get_Json(true_predictions,words):
     Results = {key_mapping.get(key, key): value for key, value in Results.items()}
     return Results
       
-    
+#############################################################################
+#############################################################################   
 def Draw(image):
     true_predictions, true_boxes,words = Run_model(image)
     draw = ImageDraw.Draw(image)
@@ -152,11 +156,51 @@ def Draw(image):
     
     return image,Results
 
+#############################################################################
+#############################################################################
 
 def Add_Results(data):
     # Render the table
     for key, value in data.items():
         data[key] = st.sidebar.text_input(key, value)
+#############################################################################
+#############################################################################
+      
+def check_if_changed(original_values, updated_values):
+    for key, value in original_values.items():
+        if updated_values[key] != value:
+            return True
+    return False
+#############################################################################
+#############################################################################
+
+def Update(Results):
+    New_results = {}
+
+    if "Fournisseur" in Results.keys():
+        text_fourni = st.sidebar.text_input("Fournisseur", value=Results["Fournisseur"])
+        New_results["Fournisseur"] = text_fourni
+        
+    if "Date Facture" in Results.keys():
+        text_InvDate = st.sidebar.text_input("Date Facture", value=Results["Date Facture"])
+        New_results["Date Facture"] = text_InvDate
+        
+    if "Numéro de facture" in Results.keys():
+        text_InvNum = st.sidebar.text_input("Numéro de facture", value=Results["Numéro de facture"])
+        New_results["Numéro de facture"] = text_InvNum
+        
+    if "Total HT" in Results.keys():
+        text_TT = st.sidebar.text_input("Total HT", value=Results["Total HT"])
+        New_results["Total HT"] = text_TT
+        
+    if "TVA" in Results.keys():
+        text_TVA = st.sidebar.text_input("TVA", value=Results["TVA"])
+        New_results["TVA"] = text_TVA
+        
+    if "Total TTC" in Results.keys():
+        text_TTC = st.sidebar.text_input("TTC", value=Results["Total TTC"])
+        New_results["Total TTC"] = text_TTC
+    return New_results
 
 #############################################################################
 #############################################################################