Update app.py

app.py

@@ -19,21 +19,71 @@ import fitz
 import re
 import io
 from collections import Counter
-from streamlit_google_auth import Authenticate
+import secrets
+
+st.set_page_config(page_title="Import Fatture AI✨")
 
-secret_json_str = os.getenv("JSON_GOOGLE_SECRET")
-if secret_json_str:
-    secret_data = json.loads(secret_json_str)
-    with open("secret-account.json", "w") as f:
-        f.write(secret_json_str)
+from streamlit_google_auth import Authenticate
+google_auth_str = os.getenv("JSON_GOOGLE_SECRET")
+google_auth_data = json.loads(google_auth_str)
+with open("google_credentials.json", "w") as f:
+    json.dump(google_auth_data, f, indent=4)
 
 authenticator = Authenticate(
-    secret_credentials_path = 'secret-account.json',
-    cookie_name='my_cookie_name',
-    cookie_key='this_is_secret',
-    redirect_uri = 'https://matteoscript-importfattureai.hf.space',
+    secret_credentials_path='google_credentials.json',
+    cookie_name=f"llm_pdf_digest_{secrets.token_hex(4)}",
+    cookie_key=secrets.token_hex(32),
+    redirect_uri=os.getenv("URL_REDIRECT"),
 )
 
+st.title("Import Fatture AI ✨")
+with st.expander("Guida completa"):
+    st.write("""Questa applicazione Python, basata su Streamlit, integra servizi di intelligenza artificiale di Gemini per automatizzare l'estrazione e la validazione dei dati dalle fatture. Il sistema gestisce documenti in vari formati (PDF, immagini) e li elabora in maniera modulare per facilitare la conversione e la verifica delle informazioni.
+
+## Funzionalità Principali
+
+- **Caricamento e Gestione dei Documenti**
+- Supporta il caricamento di file PDF, JPG, JPEG e PNG tramite un’interfaccia Streamlit.
+- Se il file è un PDF con più pagine, viene suddiviso in sezioni (configurabile tramite uno slider) per una gestione più efficace. Più il numero è basso più il risultato è preciso.
+
+- **Conversione dei Dati**
+- **Upload e Inoltro a Gemini**: I file vengono caricati e inviati al rispettivo servizio AI.
+- **Estrazione dei Dati**: Il sistema invia il documento a un modello di generazione AI per ottenere una rappresentazione JSON contenente i dati (ad es. numero di documento, data, totale imponibile e articoli).
+
+- **Validazione e Verifica**
+- **Validazione JSON**: Utilizza Pydantic per verificare la correttezza della struttura e dei dati estratti. In caso di errori, il documento viene riprocessato fino a 3 volte per cercare di correggere le anomalie.
+- **Verifica Incrociata dei Contenuti**: Per i PDF, viene estratto il testo con PyPDF2 e confrontato con i codici articolo per assicurarsi che i dati siano effettivamente presenti nel documento.
+- **Filtraggio Articoli**: Vengono mantenuti solo gli articoli compatibili con i criteri specifici (codici articolo e importi non nulli).
+
+- **Visualizzazione e Highlighting**
+- I dati validati vengono mostrati in formato tabellare e in JSON.
+- Se il documento è un PDF, il sistema evidenzia graficamente (con rettangoli rossi) i testi relativi agli articoli compatibili, semplificando il controllo visivo.
+
+## Avvertenze per l'Operatore
+
+- **Controllo Manuale Obbligatorio**:  
+L'operatore **deve sempre verificare** i dati elaborati.  
+Nonostante il riprocessamento automatico in caso di errori, è fondamentale controllare la correttezza dei dati estratti e la validità degli articoli.
+
+- **Validazione dei Dati**:  
+L'operazione di validazione tramite Pydantic e la verifica incrociata sul contenuto del PDF sono cruciali.  
+Assicurarsi che non vi siano discrepanze, specialmente nei casi in cui alcuni articoli risultino non verificati.
+""")
+
+st.write("📄 **Legge più PDF:** Carica più file PDF contemporaneamente")
+st.write("🤖 **Sfrutta l'AI di Gemini:** Per ogni documento, estrae i dati in formato JSON e in formato tabellare.")
+st.write("✅ **Mostra Articoli Compatibili:** Filtra e visualizza solo gli articoli che rispettano i criteri richiesti.")
+st.write("🔍 **Anteprima Documento:** Visualizza un'anteprima del documento evidenziando gli articoli compatibili.")
+
+authenticator.check_authentification()
+authenticator.login()
+
+if not st.session_state.get('connected'):
+    with st.sidebar:
+        st.title("Login")
+        st.write("Seleziona l'account aziendale per accedere")    
+    st.stop()
+
 GENERATION_CONFIG = settings_ai.GENERATION_CONFIG
 SYSTEM_INSTRUCTION = settings_ai.SYSTEM_INSTRUCTION
 USER_MESSAGE = settings_ai.USER_MESSAGE
@@ -396,52 +446,5 @@ def main():
                             os.remove(file_path)
 
 if __name__ == "__main__":
-    st.title("Import Fatture AI ✨")
-    with st.expander("Guida completa"):
-        st.write("""Questa applicazione Python, basata su Streamlit, integra servizi di intelligenza artificiale di Gemini per automatizzare l'estrazione e la validazione dei dati dalle fatture. Il sistema gestisce documenti in vari formati (PDF, immagini) e li elabora in maniera modulare per facilitare la conversione e la verifica delle informazioni.
-
-## Funzionalità Principali
-
-- **Caricamento e Gestione dei Documenti**
-  - Supporta il caricamento di file PDF, JPG, JPEG e PNG tramite un’interfaccia Streamlit.
-  - Se il file è un PDF con più pagine, viene suddiviso in sezioni (configurabile tramite uno slider) per una gestione più efficace. Più il numero è basso più il risultato è preciso.
-
-- **Conversione dei Dati**
-  - **Upload e Inoltro a Gemini**: I file vengono caricati e inviati al rispettivo servizio AI.
-  - **Estrazione dei Dati**: Il sistema invia il documento a un modello di generazione AI per ottenere una rappresentazione JSON contenente i dati (ad es. numero di documento, data, totale imponibile e articoli).
-
-- **Validazione e Verifica**
-  - **Validazione JSON**: Utilizza Pydantic per verificare la correttezza della struttura e dei dati estratti. In caso di errori, il documento viene riprocessato fino a 3 volte per cercare di correggere le anomalie.
-  - **Verifica Incrociata dei Contenuti**: Per i PDF, viene estratto il testo con PyPDF2 e confrontato con i codici articolo per assicurarsi che i dati siano effettivamente presenti nel documento.
-  - **Filtraggio Articoli**: Vengono mantenuti solo gli articoli compatibili con i criteri specifici (codici articolo e importi non nulli).
-
-- **Visualizzazione e Highlighting**
-  - I dati validati vengono mostrati in formato tabellare e in JSON.
-  - Se il documento è un PDF, il sistema evidenzia graficamente (con rettangoli rossi) i testi relativi agli articoli compatibili, semplificando il controllo visivo.
-
-## Avvertenze per l'Operatore
-
-- **Controllo Manuale Obbligatorio**:  
-  L'operatore **deve sempre verificare** i dati elaborati.  
-  Nonostante il riprocessamento automatico in caso di errori, è fondamentale controllare la correttezza dei dati estratti e la validità degli articoli.
-
-- **Validazione dei Dati**:  
-  L'operazione di validazione tramite Pydantic e la verifica incrociata sul contenuto del PDF sono cruciali.  
-  Assicurarsi che non vi siano discrepanze, specialmente nei casi in cui alcuni articoli risultino non verificati.
-""")
-    
-    st.write("📄 **Legge più PDF:** Carica più file PDF contemporaneamente")
-    st.write("🤖 **Sfrutta l'AI di Gemini:** Per ogni documento, estrae i dati in formato JSON e in formato tabellare.")
-    st.write("✅ **Mostra Articoli Compatibili:** Filtra e visualizza solo gli articoli che rispettano i criteri richiesti.")
-    st.write("🔍 **Anteprima Documento:** Visualizza un'anteprima del documento evidenziando gli articoli compatibili.")
     st.divider()
     main()
-    #authenticator.check_authentification()
-    #authenticator.login()
-    #if st.session_state['connected']:
-    #    st.divider()
-    #    main()
-    #else:
-    #    with st.sidebar:
-    #        st.title("Login")
-    #        st.write("Seleziona l'account aziendale per accedere")