Update src/modules/loans_engine.py

src/modules/loans_engine.py

@@ -696,7 +696,6 @@ def show_loans_engine(client, sheet_name):
                 
                 # Calcul de l'ID sous forme PRT-2026-xxxx
                 current_year = datetime.now().year
-                # len(ws_prets.get_all_values()) compte les lignes existantes pour l'incrémentation
                 new_id = f"PRT-{current_year}-{len(ws_prets.get_all_values()):04d}"
                 
                 # ORDRE STRICT DEMANDÉ POUR Prets_Master
@@ -715,7 +714,7 @@ def show_loans_engine(client, sheet_name):
                     nb_versements,                             # Nb_Versements
                     ";".join([d.strftime("%d/%m/%Y") for d in dates_versements]) if dates_versements else "", # Dates_Versements
                     date_debut.strftime("%d/%m/%Y"),           # Date_Deblocage
-                    date_fin.strftime("%d/%m/%Y"),   # Date_Fin (Maintenant garantie)
+                    date_fin.strftime("%d/%m/%Y"),             # Date_Fin (Maintenant garantie)
                     moyen_transfert,                           # Moyen_Transfert
                     "ACTIF",                                   # Statut
                     garant_id,                                 # ID_Garant
@@ -723,10 +722,11 @@ def show_loans_engine(client, sheet_name):
                 ]
                 ws_prets.append_row(row_data)
                 time.sleep(1)
-                st.success(f"Prêt {new_id} enregistré !")
-    
-                # 7b. GÉNÉRATION PDF (USINE A DOCS)
-                loan_data_dict = {
+                
+                # ✅ SAUVEGARDE DANS SESSION STATE POUR PERSISTANCE
+                st.session_state.loan_validated = True
+                st.session_state.loan_id = new_id
+                st.session_state.loan_data = {
                     "ID_Pret": new_id, 
                     "Montant_Capital": montant, 
                     "Montant_Total": montant_total,
@@ -734,19 +734,74 @@ def show_loans_engine(client, sheet_name):
                     "Duree_Semaines": duree_semaines, 
                     "Motif": motif,
                     "Date_Deblocage": date_debut.strftime("%d/%m/%Y"),
-                    "Date_Fin": date_fin.strftime("%d/%m/%Y") # Ajout ici
+                    "Date_Fin": date_fin.strftime("%d/%m/%Y")
                 }
+                st.session_state.client_data = client_info.to_dict()
+                st.session_state.garant_data = selected_garant.to_dict() if selected_garant is not None else None
+                st.session_state.df_amort = df_amort.copy()
+                
+                st.success(f" Le Prêt {new_id} enregistré avec succès !")
+                st.rerun()  # Force le rechargement pour afficher les boutons
+            
+            # ✅ AFFICHAGE BOUTONS EN DEHORS DU FORMULAIRE (persistants)
+            if st.session_state.get('loan_validated', False):
+                st.markdown("---")
+                st.markdown(f"### Documents du prêt **{st.session_state.loan_id}**")
+                
+                # Génération des PDFs (une seule fois, conservés en mémoire)
+                loan_data = st.session_state.loan_data
+                client_data = st.session_state.client_data
+                garant_data = st.session_state.garant_data
+                df_amort_saved = st.session_state.df_amort
+                
+                # Affichage des boutons en colonnes
+                col_pdf1, col_pdf2, col_pdf3, col_reset = st.columns(4)
                 
                 # PDF 1 : CONTRAT
-                pdf_contrat = generer_contrat_pret(loan_data_dict, client_info, df_amort)
-                st.download_button("Contrat de Prêt", pdf_contrat, f"Contrat_{new_id}.pdf", "application/pdf")
+                with col_pdf1:
+                    pdf_contrat = generer_contrat_pret(loan_data, client_data, df_amort_saved)
+                    st.download_button(
+                        "Contrat de Prêt", 
+                        pdf_contrat, 
+                        f"Contrat_{st.session_state.loan_id}.pdf", 
+                        "application/pdf",
+                        use_container_width=True
+                    )
                 
                 # PDF 2 : RECONNAISSANCE
-                pdf_dette = generer_reconnaissance_dette(loan_data_dict, client_info)
-                st.download_button("Reconnaissance de Dette", pdf_dette, f"Dette_{new_id}.pdf", "application/pdf")
+                with col_pdf2:
+                    pdf_dette = generer_reconnaissance_dette(loan_data, client_data)
+                    st.download_button(
+                        "Reconnaissance de Dette", 
+                        pdf_dette, 
+                        f"Dette_{st.session_state.loan_id}.pdf", 
+                        "application/pdf",
+                        use_container_width=True
+                    )
                 
                 # PDF 3 : CAUTION (SI GARANT)
-                if selected_garant is not None:
-                    pdf_caution = generer_contrat_caution(loan_data_dict, selected_garant)
-                    st.download_button("Contrat de Caution", pdf_caution, f"Caution_{new_id}.pdf", "application/pdf")
+                with col_pdf3:
+                    if garant_data is not None:
+                        pdf_caution = generer_contrat_caution(loan_data, garant_data)
+                        st.download_button(
+                            "Contrat de Caution", 
+                            pdf_caution, 
+                            f"Caution_{st.session_state.loan_id}.pdf", 
+                            "application/pdf",
+                            use_container_width=True
+                        )
+                    else:
+                        st.info("Pas de garant")
+                
+                # Bouton pour réinitialiser et créer un nouveau prêt
+                with col_reset:
+                    if st.button("Nouveau Prêt", use_container_width=True, type="primary"):
+                        # Nettoyage du session state
+                        st.session_state.loan_validated = False
+                        st.session_state.pop('loan_id', None)
+                        st.session_state.pop('loan_data', None)
+                        st.session_state.pop('client_data', None)
+                        st.session_state.pop('garant_data', None)
+                        st.session_state.pop('df_amort', None)
+                        st.rerun()
     st.markdown('</div>', unsafe_allow_html=True)