Added More time periods

app.py

@@ -95,14 +95,24 @@ def update_sip_calculator(*args):
     total_weight = sum(schemes.values())
 
     end_date = datetime.now().date()
+
     if period == "Custom":
         if not custom_start_date or not custom_end_date:
             return "Please provide both start and end dates for custom period.", None, None, None
         start_date = datetime.strptime(custom_start_date, "%Y-%m-%d").date()
         end_date = datetime.strptime(custom_end_date, "%Y-%m-%d").date()
+    
+    elif period == "YTD":
+        start_date = datetime(end_date.year, 1, 1)
+
     else:
-        years = int(period.split()[0])
-        start_date = end_date - timedelta(days=years*365)
+        # check if string contaiins year
+        if 'year' in period.split()[1]:
+            years = int(period.split()[0])
+            start_date = end_date - timedelta(days=years*365)
+        else:
+            months = int(period.split()[0])
+            start_date = end_date - timedelta(days=months*30)
 
     try:
         portfolio_return, final_value, total_investment, scheme_returns = calculate_portfolio_returns(schemes, sip_amount, start_date, end_date, SIP_Date,schemes_df)
@@ -206,10 +216,10 @@ def create_ui():
         gr.Markdown("# Mutual Fund SIP Returns Calculator")
 
         with gr.Row():
-            period = gr.Dropdown(choices=["1 year", "3 years", "5 years", "7 years", "10 years", "Custom"], label="Select Period")
+            period = gr.Dropdown(choices=["YTD", "1 month","3 months","6 months","1 year", "3 years", "5 years", "7 years", "10 years","15 years","20 years", "Custom"], label="Select Period")
             custom_start_date = gr.Textbox(label="Custom Start Date (YYYY-MM-DD)", visible=False)
             custom_end_date = gr.Textbox(label="Custom End Date (YYYY-MM-DD)", visible=False)
-            SIP_Date = gr.Dropdown(label="SIP Date", choices=["start","middle","end"])
+            SIP_Date = gr.Dropdown(label="Monthly SIP Date", choices=["start","middle","end"])
 
         sip_amount = gr.Number(label="SIP Amount (₹)")