import streamlit as st
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

def calculate_investment(initial_deposit, years, rate, contribution, freq):
    months = years * 12
    rate_monthly = (rate / 100) / 12
    balance = [initial_deposit]
    
    for i in range(1, months + 1):
        new_balance = balance[-1] * (1 + rate_monthly) + (contribution if freq == 'Monthly' else 0)
        if i % 12 == 0 and freq == 'Annually':
            new_balance += contribution
        balance.append(new_balance)
    
    return balance

# Streamlit UI
st.title("Kalkulator Investasi")

# User Inputs
initial_deposit = st.number_input("Deposit Awal (Rp)", min_value=1, value=1000000)
years = st.number_input("Lama Investasi (Tahun)", min_value=1, max_value=50, value=10)
estimated_rate = st.number_input("Estimasi Tingkat Pengembalian Tahunan (%)", min_value=0.0, value=5.0)
contribution = st.number_input("Jumlah Kontribusi (Rp)", min_value=0, value=100000)
contribution_freq = st.radio("Frekuensi Kontribusi", ("Bulanan", "Tahunan"))

if st.button("Hitung"):
    balances = calculate_investment(initial_deposit, years, estimated_rate, contribution, contribution_freq)
    
    # Plot results
    plt.figure(figsize=(8,5))
    plt.plot(range(len(balances)), balances, label="Total Saldo", color='green')
    plt.xlabel("Bulan")
    plt.ylabel("Saldo (Rp)")
    plt.title("Pertumbuhan Investasi dari Waktu ke Waktu")
    plt.legend()
    st.pyplot(plt)
    
    st.success(f"Saldo Akhir: Rp{balances[-1]:,.2f}")


#Sekarang kalkulator investasi menggunakan mata uang Rupiah (Rp) dengan bahasa Indonesia. Jika ada yang ingin diubah lagi, beri tahu saya!