Update app.py

app.py

@@ -26,6 +26,7 @@ st.title("Import Fatture AI ✨")
 # Gestionione LOGIN
 if "logged" not in st.session_state:
     st.session_state.logged = False
+    st.session_state.model = "gemini-2.0-flash"
 if st.session_state.logged == False:
     login_placeholder = st.empty()
     with login_placeholder.container():
@@ -87,7 +88,7 @@ API_KEY_GEMINI = settings_ai.API_KEY_GEMINI
 # Configura il modello Gemini
 genai.configure(api_key=API_KEY_GEMINI)
 model = genai.GenerativeModel(
-    model_name="gemini-2.0-flash",
+    model_name=st.session_state.model,
     generation_config=GENERATION_CONFIG,
     system_instruction=SYSTEM_INSTRUCTION
 )
@@ -291,7 +292,7 @@ def process_document(path_file: str, number_pages_split: int, use_azure: bool =
     if mime_type is None:
         mime_type = "application/octet-stream"
     if use_azure:
-        number_pages_split = 2
+        number_pages_split = 1
     if not path_file.lower().endswith(".pdf"):
         print("File non PDF: elaborazione come immagine.")
         documento_finale = process_document_splitted(path_file, chunk_label="(immagine)", use_azure=use_azure)
@@ -327,14 +328,31 @@ def process_document(path_file: str, number_pages_split: int, use_azure: bool =
     if documento_finale is None:
         raise RuntimeError("Nessun documento elaborato.")
 
-    # Controlli aggiuntivi: Se esiste un AVE non possono esistere altri articoli non ave. Se articoli DOPPI segnalo!
+    # Controlli aggiuntivi: Se esiste un AVE non possono esistere altri articoli non ave. 
     if any(articolo.CodiceArticolo.startswith("AVE") for articolo in documento_finale.Articoli):
         documento_finale.Articoli = [articolo for articolo in documento_finale.Articoli if articolo.CodiceArticolo.startswith("AVE")]
-    combinazioni = [(articolo.CodiceArticolo, articolo.TotaleNonIvato) for articolo in documento_finale.Articoli]
-    conta_combinazioni = Counter(combinazioni)
-    for articolo in documento_finale.Articoli:
-        if conta_combinazioni[(articolo.CodiceArticolo, articolo.TotaleNonIvato)] > 1:
-            articolo.Verificato = False    
+    # Controllo occorrenze di doppioni 
+    if path_file.lower().endswith(".pdf"):
+        pdf_text = pdf_to_text(path_file)
+        pdf_text = pdf_text.replace(" ", "")
+        occorrenze = {}
+        for articolo in documento_finale.Articoli:
+            codice_clean = articolo.CodiceArticolo.replace(" ", "")
+            if codice_clean not in occorrenze:
+                occorrenze[codice_clean] = pdf_text.count(codice_clean)
+        articoli_contati = {}
+        for articolo in documento_finale.Articoli:
+            codice_clean = articolo.CodiceArticolo.replace(" ", "")
+            if codice_clean in pdf_text:
+                print(codice_clean)
+                print(occorrenze[codice_clean])
+                articoli_contati[codice_clean] = articoli_contati.get(codice_clean, 0) + 1
+                if articoli_contati[codice_clean] <= occorrenze.get(codice_clean, 0):
+                    articolo.Verificato = True
+                else:
+                    articolo.Verificato = False
+            else:
+                articolo.Verificato = False
     return documento_finale
 
 # Analizza Fattura con AZURE 
@@ -369,11 +387,15 @@ def parse_invoice_to_documento_azure(result) -> Documento:
     if items_field and items_field.value_array:
         for item in items_field.value_array:
             product_code_field = item.value_object.get("ProductCode")
+            description_field = str(item.value_object.get("Description").get("content"))
+            if not description_field:
+                description_field = ""
             codice_articolo = product_code_field.value_string if product_code_field and product_code_field.value_string else ""
             amount_field = item.value_object.get("Amount")
             totale_non_ivato = amount_field.value_currency.amount if amount_field and amount_field.value_currency else 0.0
             articolo = Articolo(
                 CodiceArticolo=codice_articolo,
+                DescrizioneArticolo=description_field,
                 TotaleNonIvato=totale_non_ivato,
                 Verificato=None
             )
@@ -393,9 +415,11 @@ def main():
     #st.set_page_config(page_title="Import Fatture AI", page_icon="✨")
     st.sidebar.title("Caricamento File")
     uploaded_files = st.sidebar.file_uploader("Seleziona uno o più PDF", type=["pdf", "jpg", "jpeg", "png"], accept_multiple_files=True)
-    model_ai = st.sidebar.selectbox("Modello", ['Gemini Flash 2.0']) # 'Azure Intelligence'])
+    model_ai = st.sidebar.selectbox("Modello", ['Gemini Flash 2.0', 'Gemini 2.5 Pro', 'Azure Intelligence'])
+    if model_ai == 'Gemini 2.5 Pro':
+        st.session_state.model = "gemini-2.5-pro-exp-03-25"
     use_azure = True if model_ai == 'Azure Intelligence' else False
-    number_pages_split = st.sidebar.slider('Split Pagine', 1, 30, 2, help="Numero suddivisione pagine del PDF. Più il numero è basso e più il modello AI è preciso, più è alto più è veloce")
+    number_pages_split = st.sidebar.slider('Split Pagine', 1, 30, 1, help="Numero suddivisione pagine del PDF. Più il numero è basso e più il modello AI è preciso, più è alto più è veloce")
     if st.sidebar.button("Importa", type="primary", use_container_width=True):
         if not uploaded_files:
             st.warning("Nessun file caricato!")
@@ -419,15 +443,18 @@ def main():
                             f"- **Articoli Compatibili**: {len(doc.Articoli)}\n"
                             f"- **Totale Documento**: {format_euro(doc.TotaleImponibile)}\n"
                         )
+                        if totale_non_ivato > doc.TotaleImponibile and doc.TotaleImponibile > 0:
+                            st.warning("Totale Ave maggiore di Totale Merce")
                         if totale_non_ivato_non_verificato > 0:
-                            st.error(f"Totale Ave Non Verificato: {format_euro(totale_non_ivato_verificato)}")
-                        elif totale_non_ivato != 0:
+                            st.error(f"Totale Ave Non Verificato: {format_euro(totale_non_ivato_non_verificato)}")
+                        if totale_non_ivato > 0:
                             st.success(f"Totale Ave Verificato: {format_euro(totale_non_ivato_verificato)}")
                         df = pd.DataFrame([{k: v for k, v in Articolo.model_dump().items() if k != ""} for Articolo in doc.Articoli])                                       
                         if 'Verificato' in df.columns:
                             df['Verificato'] = df['Verificato'].apply(lambda x: "✅" if x == 1 else "❌" if x == 0 else "❓" if x == 2 else x)
                         if totale_non_ivato > 0:
-                            st.dataframe(df, use_container_width=True ,column_config={"TotaleNonIvato": st.column_config.NumberColumn("Totale non Ivato",format="€ %.2f")})
+                            df["TotaleNonIvato"] = df["TotaleNonIvato"].apply(format_euro)
+                            st.dataframe(df, use_container_width=True)
                         st.json(doc.model_dump(), expanded=False)
                         if totale_non_ivato == 0:
                             st.info(f"Non sono presenti articoli 'AVE'")