Primo esempio di summarization

HFBot.py

@@ -46,6 +46,12 @@ class HFBot():
         #print(prompt)
         rersponse = self.genera(prompt)
         return rersponse
+    
+    def riassumi(self, text):
+        retval =""
+        pipe = pipeline("summarization", model=self.model_name)
+        retval = pipe(text)
+        return retval
     def genera(self, prompt: str):  
         retval = ""
         messages = [

app.py

@@ -1,8 +1,8 @@
 import streamlit as st
+import os
 from transformers import pipeline
 from HFBot import HFBot
-
-
+from textutils import ParagraphDocumentProcessor, WholeTextDocumentProcessor, SmallFragmentDocumentProcessor
 def main():
     modelliLLM = [
             'Almawave/Velvet-2B',
@@ -11,6 +11,12 @@ def main():
             'sapienzanlp/Minerva-350M-base-v1.0',
             'Qwen/Qwen3-0.6B'
         ]
+    
+    UPLOAD_DIR="/tmp/"
+    if "indice_creato" not in st.session_state:
+        st.session_state["indice_creato"] = False
+
+    testi = []
     modello_scelto = st.selectbox("Seleziona un modello:", modelliLLM, index=0)
     st.write(f"Hai selezionato: {modello_scelto}")
 
@@ -23,18 +29,41 @@ def main():
     
      #res = LLM.chat(domanda=domanda, istruzioni= Istruzioni, frammenti=frammenti)
     #st.write(f"Domanda {domanda} - risposta {res}")
+    st.title("Suddivisione in paragrafi")
+    docprocessor_options = {
+        "ParagraphDocumentProcessor": ParagraphDocumentProcessor(),
+        "WholeText": WholeTextDocumentProcessor(),
+        "Small Fragments": SmallFragmentDocumentProcessor(),
+    }
+    selected_docprocessor = st.selectbox("Divisione in paragrafi", docprocessor_options.keys())
+    docprocessor = docprocessor_options[selected_docprocessor]
+    st.write(f"Hai selezionato: **{selected_docprocessor}**")
     
-
-    
-    if "atto_caricato" not in st.session_state:
+    if not st.session_state["indice_creato"]:
         st.subheader("Carica l'atto principale (Determinazione)")
 
         main_pdf = st.file_uploader("Carica 1 file PDF - Determinazione", type=["pdf"], key="main_pdf")
         if main_pdf:
-            st.session_state.main_pdf_nome = main_pdf.name
-            st.session_state["atto_caricato"] = True
-    else:
-        st.success(f'Caricato file  {st.session_state.main_pdf_nome} ')
+            st.session_state["main_pdf_nome"] = main_pdf.name
+        if st.button("Esegui analisi"):
+            if main_pdf is not None:
+                try:
+                    save_path = os.path.join(UPLOAD_DIR, main_pdf.name)
+                    with open(save_path, "wb") as f:
+                        f.write(main_pdf.read())
+                    st.success(f"✅  Determinazione caricata con successo! File salvato in: `{save_path}`")
+                    testi= docprocessor.estrai_da_pdf(save_path)
+                    st.write(testi[:50])
+                    #ßragpipeline.aggiungi_file_pdf(save_path)
+                    st.session_state["indice_creato"] = True
+                except Exception as e:
+                    st.error(f"❌ Errore nel salvataggio del file: {e}")
+            else:
+                st.warning("Nessun PDF di Determinazione caricato.")
+        if st.session_state["indice_creato"]:
+            st.write("Eseguo riassunto")
+            summary = LLM.riassumi(testi)
+            st.success(summary)
 
 if __name__ == "__main__":
     main()