Update app.py

app.py

@@ -86,24 +86,34 @@ def calculate_differentiated_payment(loan_amount, interest_rate, loan_term):
 
     return total_payment / loan_term
 
-def save_to_file(income, loan_payments, has_guarantors, loan_term, interest_rate, preferential_rate, 
-                 monthly_payment, start_date, end_date, desired_loan_amount, 
-                 max_monthly_payment, max_loan_amount, payment_type):
-    with open("output.txt", "w") as f:
-        f.write(f"Доход: {income} BYN\n")
-        f.write(f"Платежи по кредитам: {loan_payments} BYN\n")
-        f.write(f"Наличие поручителей: {'Да' if has_guarantors else 'Нет'}\n")
-        f.write(f"Срок кредита, мес: {loan_term}\n")
-        f.write(f"Процентная ставка: {interest_rate} %\n")
-        f.write(f"Льготная ставка: {preferential_rate} %\n")
-        f.write(f"Желаемая сумма кредита: {desired_loan_amount} BYN\n")
-        f.write(f"Максимальная доступная сумма кредита: {max_loan_amount:.2f} BYN\n")
-        f.write(f"Размер ежемесячного платежа: {monthly_payment:.2f} BYN\n")
-        f.write(f"Дата получения кредита: {start_date.strftime('%Y-%m-%d')}\n")
-        f.write(f"Дата окончания кредита: {end_date.strftime('%Y-%m-%d')}\n")
-        f.write(f"Максимальный платеж, возможный для клиента: {max_monthly_payment:.2f} BYN\n")
-
-    st.success("Данные успешно сохранены в файл.")
+def save_to_file(income, loan_payments, has_guarantors, loan_term, interest_rate, 
+                  preferential_rate, monthly_payment, start_date, end_date, 
+                  desired_loan_amount, max_monthly_payment, max_loan_amount, payment_type):
+    # Создание DataFrame с результатами расчета
+    data = {
+        "Сумма кредита": [desired_loan_amount],
+        "Срок (мес.)": [loan_term],
+        "Процентная ставка (%)": [interest_rate],
+        "Ежемесячный платеж": [monthly_payment],
+        "Дата начала": [start_date],
+        "Дата окончания": [end_date],
+        "Максимальный ежемесячный платеж": [max_monthly_payment],
+        "Максимальная сумма кредита": [max_loan_amount],
+        "Тип платежа": [payment_type],
+    }
+    
+    df = pd.DataFrame(data)
+    
+    # Запись в CSV
+    csv = df.to_csv(index=False)
+    
+    # Сохранение файла в память
+    st.download_button(
+        label="Скачать результаты",
+        data=csv,
+        file_name='loan_calculation_results.csv',
+        mime='text/csv'
+    )
 
 if __name__ == "__main__":
     main()