bug fixes + adding goal tracker

app.py

@@ -1,4 +1,5 @@
 import os
+import datetime
 import gradio as gr
 from sentence_transformers import SentenceTransformer
 import torch
@@ -99,20 +100,6 @@ body {
     background-color: #52528C !important;
 }
 """
-
-with gr.Blocks(css=custom_css) as demo:
-    gr.Image("banner_image.png", show_label=False, container=False)
-
-    gr.ChatInterface(
-        respond,
-        title="Student Formula Bot 🔬",
-        description='Welcome to the core component of \"The Student Formula\": the RAG chatbot! With the ability to act as a finance tutor, accountability buddy, and goal-setting partner all in one, it\'s designed to best suit your needs on the way to productivity and success. To get started, ask about the basic principles of creating a budget!',
-        examples=[
-            "How do I build and balance a budget?",
-            "Why do people write SMART goals?",
-            "What is educational investment?"
-        ]
-    )
     
     gr.HTML(
         """
@@ -143,11 +130,72 @@ def budgetCalculator(income):
         f"Based on a monthly income of ${income:,.2f}."
     )
 
-budgetTab = gr.Interface(
-    base = budgetCalculator,
-    inputs = gr.Number(label = "Monthly Income ($))", minimum = 0),
-    outputs = gr.Markdown(label = "Monthly Budget"),
-    title = "50/30/20 Budget Calculator",
-    description = "Want to see what your monthly budget might look like when following financial literacy principles? Enter your monthly income to see the suggested breakdown!"
+def goalTracker(goal, savings, targetDate):
+    if goal is None or savings is None  or not targetDate:
+        return "Please fill in the tracker to get started!"
+    if goal <= 0:
+        return "Set your goal for higher than 0!"
+    if savings < 0:
+        return "Savings can't be negative; start at 0 if needed!"
+
+    try:
+        target = datetime.datetime.strptime(targetDate, "%Y-%m-%d").date()
+    except ValueError:
+        return "Please enter the date in YYYY-MM-DD format! (ex: 2026-06-12)"
+
+    today = datetime.date.today()
+    daysLeft = (targetDate - today).days()
+
+    if daysLeft <= 0:
+        return "Try to set your goal for sometime in the future!"
+
+    remaining = goal - savings
+    if remaining <= 0:
+        return "Congratulations on hitting your goal!"
+
+    weeklyAmount = remaining / (daysLeft / 7)
+    monthlyAmount = remaining / (daysLeft / 30.44)
+
+    return (
+        f"### Goal Tracker Results\n\n"
+        f"- **Amount Remaining:** ${remaining:,.2f}\n"
+        f"- **Days Remaining:** ${daysLeft:,.2f}\n"
+        f"- **Save this much per week:** ${weeklyAmount:,.2f}\n"
+        f"- **Save this much per month:** ${monthlyAmount:,.2f}\n"
+    )
+
+with gr.Blocks(css=custom_css) as demo:
+    gr.Image("banner_image.png", show_label=False, container=False)
+
+    with gr.Tabs():
+        with gr.Tab("Chatbot"):
+            gr.ChatInterface(
+                respond,
+                title="Student Formula Bot 🔬",
+                description='Welcome to the core component of \"The Student Formula\": the RAG chatbot! With the ability to act as a finance tutor, accountability buddy, and goal-setting partner all in one, it\'s designed to best suit your needs on the way to productivity and success. To get started, ask about the basic principles of creating a budget!',
+                examples=[
+                    "How do I build and balance a budget?",
+                    "Why do people write SMART goals?",
+                    "What is educational investment?"
+                ]
+            )
+
+        with gr.Tab("Budget Calculator"):
+            gr.Markdown("## 50/30/20 Budget Calculator")
+            gr.Markdown("Want to see what your monthly budget might look like when following financial literacy principles? Enter your monthly income to see the suggested breakdown!")
+            incomeInput = gr.Number(label = "Monthly Income ($))", minimum = 0)
+            budgetOutput = gr.Markdown(label = "Monthly Budget"),
+            budgetButton = gr.Button("Calculate")
+            budgetButton.click(fn = budgetCalculator, inputs = incomeInput, outputs = budgetOutput)
+
+        with gr.Tab("Goal Tracker"):
+            gr.Markdown("## Savings Goal Tracker")
+            gr.Markdown("Saving up to buy something special? Want to figure out how long it'll take to reach that goal? Enter your goal amount, current savings, and target date to see the suggested timeline!")
+            goalInput = gr.Number(label = "Savings Goal ($)", minimum = 0),
+            savingsInput = gr.Number(label = "Current Savings ($)", minimum = 0),
+            targetInput = gr.Textbox(label = "Target Date (YYYY-MM-DD", placeholder = "2026-12-31")
+            goalOutput =  gr.Markdown(label = "Your Savings Plan")
+            goalButton = gr.Button("Calculate")
+            goalButton.click(fn = goalTracker, inputs = [goalInput, savingsInput, targetInput], outputs = goalOutput)
 
 demo.launch()