fix: added a first version of pdf generation (still need to be adjusted)

Author: Raffaele Crafa

app.py

@@ -22,6 +22,7 @@
 from src.clients import list_providers
 from src.core import ChatbotAgent, FinancialAdvisorAgent
 from src.models import FinancialProfile, Portfolio
+from src.ui.pdf_generator import PDFGenerator  # <-- NUOVO IMPORT
 from src.ui.settings_page import show_settings_page
 
 MONTECARLO_MIN_ASSET_VOLATILITY = float(
@@ -292,26 +293,32 @@ def initialize_financial_advisor(
         raise
 
 
+# TODO: vedere se eliminare
+# Funzione per smooth scroll della pagina
 def _scroll_to_bottom():
     """
     Force scroll to bottom by targeting the specific Streamlit main container.
     """
     js = """
     <script>
         function forceScroll() {
+            // Cerca il contenitore scrollabile principale di Streamlit
+            // I selettori coprono varie versioni di Streamlit
             var scrollable_div = window.parent.document.querySelector('section.main') ||
                                  window.parent.document.querySelector('.main') ||
                                  window.parent.document.querySelector('[data-testid="stMain"]');
 
             if (scrollable_div) {
+                // Forza lo scroll alla fine dell'altezza totale del contenuto
                 scrollable_div.scrollTop = scrollable_div.scrollHeight;
             }
         }
 
+        // Esegui più volte per "vincere" contro il rendering dinamico di Plotly e i Toast
         setTimeout(forceScroll, 100);
         setTimeout(forceScroll, 500);
         setTimeout(forceScroll, 1000);
-        setTimeout(forceScroll, 2000);
+        setTimeout(forceScroll, 2000); // Un ultimo tentativo dopo 2 secondi per sicurezza
     </script>
     """
     components.html(js, height=0, width=0)
@@ -1850,8 +1857,11 @@ def main():
         with st.chat_message(message["role"]):
             st.markdown(message["content"])
 
+    # --- INPUT SEMPRE PRESENTE ---
+    # Manteniamo la chat input renderizzata ma disabilitata se la conversazione è finita.
+    # Questo impedisce al browser di perdere il focus e "saltare" in alto.
     prompt = st.chat_input(
-        "Assessment completed. See the results above.",
+        "Ask me about your finances...",
         disabled=st.session_state.conversation_completed,
     )
 
@@ -1864,6 +1874,9 @@ def main():
                 duration="long",
             )
 
+        # TODO: vedere se elimanrla
+        _scroll_to_bottom()
+
         logger.debug("Conversation is completed")
 
         # Generate and display portfolio
@@ -1931,6 +1944,9 @@ def main():
                     "Your financial profile and PAC metrics have been extracted and analyzed."
                 )
 
+                # TODO: vedere se eliminare
+                _scroll_to_bottom()
+
                 # Financial Profile in an expanded section
                 with st.expander(
                     "📊 View Your Financial Profile & Summary", expanded=False
@@ -1943,12 +1959,44 @@ def main():
                         "- Click 'Clear Conversation' to start a new assessment or 'Change Provider' to start over"
                     )
 
+                # --- AGGIUNGO QUI IL PULSANTE DI ESPORTAZIONE PDF ---
+
+                if (
+                    st.session_state.financial_profile
+                    and st.session_state.generated_portfolio
+                ):
+                    # 1. Recupera la configurazione dell'Agente per il PDF
+                    agent_config = chatbot_agent.get_config_summary()
+
+                    # 2. Genera l'istanza del PDF
+                    pdf = PDFGenerator(
+                        agent_config=agent_config,
+                        profile_data=st.session_state.financial_profile.model_dump(),
+                        portfolio_data=st.session_state.generated_portfolio,
+                    )
+
+                    # 3. Genera il file in memoria
+                    pdf_output_bytes = pdf.generate()
+
+                    # 4. Mostra il pulsante di download Streamlit
+                    st.download_button(
+                        label="⬇️ Esporta Report Portfolio (PDF)",
+                        data=pdf_output_bytes,
+                        file_name="Report_Finanziario.pdf",
+                        mime="application/pdf",
+                        key="download_pdf_button",
+                        type="primary",
+                    )
+
+                # --- FINE PULSANTE DI ESPORTAZIONE PDF ---
+
+                # TODO: vedere se eliminare
                 _scroll_to_bottom()
 
         else:
             logger.debug("No financial profile available to display")
     else:
-        # Handle user input ONLY if there is a prompt (i.e., not disabled)
+        # Gestiamo l'input dell'utente SOLO se c'è un prompt (quindi non è disabilitato)
         if prompt:
             logger.debug("User input received: %s", prompt[:100])
 

requirements.txt

@@ -54,6 +54,7 @@ filetype==1.2.0
 Flask==3.1.1
 flatbuffers==25.9.23
 fonttools==4.58.2
+fpdf2==2.8.5
 frozendict==2.4.6
 frozenlist==1.8.0
 fsspec==2025.5.1

src/models/portfolio.py

@@ -97,12 +97,58 @@ class Portfolio(BaseModel):
         },
     )
 
+    # TODO: questa è la funzione originale: vedere se eliminarla
+    # @model_validator(mode="after")
+    # def validate_total_percentage(self):
+    #     """Validate that asset percentages sum to approximately 100."""
+    #     total = sum(asset.percentage for asset in self.assets)
+    #     if not (99 <= total <= 101):
+    #         raise ValueError(
+    #             f"Asset percentages must sum to 100%, got {total}% instead"
+    #         )
+    #     return self
+
+    # TODO: è stata aggiunta questa nuova funzione nel caso in cui non si arrivi al 100%
+    # ma va modificata perché così come è ora accetta anche i casi in cui non ho il 100%
+    # invece bisogna forzare il fatto che la somma deve essere il 100%
     @model_validator(mode="after")
-    def validate_total_percentage(self):
-        """Validate that asset percentages sum to approximately 100."""
+    def normalize_total_percentage(self):
+        """
+        Auto-correct asset percentages to ensure they sum to exactly 100%.
+        Instead of raising an error, specifically re-proportions the values.
+        """
+        if not self.assets:
+            return self
+
         total = sum(asset.percentage for asset in self.assets)
-        if not (99 <= total <= 101):
-            raise ValueError(
-                f"Asset percentages must sum to 100%, got {total}% instead"
-            )
+
+        # Se il totale è 0, non possiamo normalizzare (evitiamo divisione per zero)
+        if total == 0:
+            # In questo caso estremo, assegniamo tutto al primo asset o lanciamo errore
+            # Qui scegliamo di distribuire equamente per non rompere l'app
+            share = 100.0 / len(self.assets)
+            for asset in self.assets:
+                asset.percentage = round(share, 2)
+            return self
+
+        # Se la somma è già corretta (con tolleranza), non facciamo nulla
+        if 99.0 <= total <= 101.0:
+            return self
+
+        # LOGICA DI NORMALIZZAZIONE
+        # Esempio: Se il totale è 60%, il fattore è 100/60 = 1.666...
+        factor = 100.0 / total
+
+        for asset in self.assets:
+            # Scaliamo ogni asset
+            asset.percentage = round(asset.percentage * factor, 2)
+
+        # Controllo finale per arrotondamenti (es. somma 99.99 o 100.01)
+        new_total = sum(asset.percentage for asset in self.assets)
+        diff = 100.0 - new_total
+
+        if diff != 0:
+            # Aggiungiamo/togliamo la piccola differenza al primo asset (spesso il più grande)
+            self.assets[0].percentage = round(self.assets[0].percentage + diff, 2)
+
         return self