import streamlit as st
import PyPDF2
import os
import io
import re # Per l'uso delle espressioni regolari
import uuid # Per la generazione di UUID
from datetime import datetime # Per ottenere il timestamp

# Funzione per estrarre testo dal PDF
def extract_text_from_pdf(pdf_file):
    """
    Estrae il testo dalla prima pagina di un PDF.
    """
    try:
        # Crea un oggetto BytesIO dal file caricato per permettere a PyPDF2 di leggerlo
        # Questo è importante perché Streamlit passa un UploadedFile che potrebbe non essere seekable o leggibile direttamente
        # da PyPDF2 senza essere avvolto in un buffer di byte.
        pdf_reader = PyPDF2.PdfReader(io.BytesIO(pdf_file.read()))
        if pdf_reader.pages:
            first_page = pdf_reader.pages[0]
            text = first_page.extract_text()
            return text
        else:
            return ""
    except Exception as e:
        st.error(f"Errore durante l'estrazione del testo dal PDF: {e}")
        return ""

# Funzione principale dell'app Streamlit
def app():
    st.set_page_config(layout="centered", page_title="Rinominatore PDF Intelligente")
    st.title("📄 Rinominatore di PDF Intelligente")
    st.markdown("""
        Carica uno o più file PDF. Il programma tenterà di leggere il contenuto della prima pagina
        per suggerire un nuovo nome per ciascun file. Il naming seguirà il formato:
        `1006373999_DDT[NUMERO_DOCUMENTO]_TIMESTAMP_UUID.pdf`.
        """)

    # Permette il caricamento di più file PDF
    uploaded_files = st.file_uploader("Carica i tuoi file PDF", type="pdf", accept_multiple_files=True)

    if uploaded_files:
        st.success(f"{len(uploaded_files)} file caricati con successo!")
        
        st.subheader("Suggerisci Nomi e Scarica PDF Rinominati:")
        
        for i, uploaded_file in enumerate(uploaded_files):
            st.markdown(f"---")
            st.markdown(f"**Elaborazione file: {uploaded_file.name}**")
            
            # Sposta il puntatore del file all'inizio per ogni elaborazione
            uploaded_file.seek(0) 
            
            # Estrai il testo dalla prima pagina del PDF
            extracted_text = extract_text_from_pdf(uploaded_file)
            
            sequence_number = ""
            if extracted_text:
                st.text_area(f"Anteprima testo estratto da '{uploaded_file.name}':", extracted_text[:500], height=100, key=f"text_area_{i}")
                
                # Cerca il numero di sequenza con espressione regolare "DOCUM. NR. DEL PAG. XXXXXXXXXXX"
                match = re.search(r"DOCUM\. NR\. DEL PAG\.\s*(\d+)", extracted_text)
                if match:
                    sequence_number = match.group(1)
                    st.info(f"Numero DDT rilevato: **{sequence_number}**")
                else:
                    st.warning("Numero DDT (es. 'DOCUM. NR. DEL PAG. 12345') non trovato in questo PDF. Il nome suggerito userà un placeholder per la sequenza.")
                    sequence_number = "" # Assicura che sia una stringa vuota se non trovato

            else:
                st.warning(f"Impossibile estrarre testo da '{uploaded_file.name}'. Il nome suggerito sarà basato sul nome originale.")
                sequence_number = "" # Assicura che sia una stringa vuota se non c'è testo

            # Componenti per il nuovo nome del file
            codice_ddt_fixed = "1006373999"
            
            # TIMESTAMP: AAAAMMGGhhmmssSSS (fino ai millisecondi)
            current_timestamp = datetime.now().strftime("%Y%m%d%H%M%S%f")[:-3] 
            
            # UUID: generatore univoco di 5 caratteri
            unique_uuid_5_chars = uuid.uuid4().hex[:5] 

            # Costruisci il nome suggerito con '_DDT' prima della sequenza
            if sequence_number:
                suggested_name_base = f"{codice_ddt_fixed}_DDT{sequence_number}_{current_timestamp}_{unique_uuid_5_chars}"
            else:
                # Fallback se il numero di sequenza non è trovato
                base_original_name = os.path.splitext(uploaded_file.name)[0]
                # Pulisci il nome originale da caratteri non validi per i nomi file
                clean_original_name = "".join(c if c.isalnum() or c in ('_', '-') else '_' for c in base_original_name)
                # Limita la lunghezza del fallback del nome originale per evitare nomi troppo lunghi
                suggested_name_base = f"{codice_ddt_fixed}_DDT_NOSEQ_{clean_original_name[:10]}_{current_timestamp}_{unique_uuid_5_chars}"
            
            new_filename_input = st.text_input(
                f"Nuovo nome suggerito per '{uploaded_file.name}' (senza estensione .pdf):", 
                value=suggested_name_base, 
                key=f"filename_input_{i}" # Chiave unica per ogni widget
            )

            final_new_filename = f"{new_filename_input.strip()}.pdf"

            # Prepara il contenuto del PDF originale per il download con il nuovo nome
            # Sposta il puntatore del file caricato all'inizio prima di leggerlo di nuovo
            uploaded_file.seek(0)
            pdf_bytes = uploaded_file.read()

            st.download_button(
                label=f"Scarica {final_new_filename}",
                data=pdf_bytes,
                file_name=final_new_filename,
                mime="application/pdf",
                key=f"download_btn_{i}" # Chiave unica per ogni pulsante
            )
            # Messaggio di successo più conciso e immediatamente sotto il pulsante di download
            st.success(f"PDF pronto come '{final_new_filename}'!")

    else:
        st.info("Carica uno o più file PDF per iniziare il processo di rinominazione.")


# Punto di ingresso dell'applicazione Streamlit
if __name__ == "__main__":
    app()