Create app.py

app.py

@@ -0,0 +1,41 @@
+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("Investment Calculator")
+
+# User Inputs
+initial_deposit = st.number_input("Initial Deposit ($)", min_value=1, value=1000)
+years = st.number_input("Years of Growth", min_value=1, max_value=50, value=10)
+estimated_rate = st.number_input("Estimated Annual Rate of Return (%)", min_value=0.0, value=5.0)
+contribution = st.number_input("Contribution Amount ($)", min_value=0, value=100)
+contribution_freq = st.radio("Contribution Frequency", ("Monthly", "Annually"))
+
+if st.button("Calculate"):
+    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 Balance", color='green')
+    plt.xlabel("Months")
+    plt.ylabel("Balance ($)")
+    plt.title("Investment Growth Over Time")
+    plt.legend()
+    st.pyplot(plt)
+    
+    st.success(f"Final Balance: ${balances[-1]:,.2f}")