Update app.py

app.py

@@ -1,304 +1,124 @@
-# ============================================================
-# Personal Finance Helper
-# ============================================================
-# This app:
-# 1. Collects some simple financial information from the user
-# 2. Calculates a few basic financial indicators using Python
-# 3. Sends the information to Gemini
-# 4. Displays Gemini's response in a simple web app
-# ============================================================
-
-
-# -----------------------------
-# 1. Import the libraries we need
-# -----------------------------
-
 import os
 import gradio as gr
 import spaces
 from google import genai
 
+# Connect to Gemini
+client = genai.Client(
+    api_key=os.getenv("GEMINI_API_KEY")
+)
 
-# -----------------------------
-# 2. Connect to Gemini
-# -----------------------------
-# The API key should NOT be written directly in this file.
-#
-# Instead, add it in Hugging Face:
-#
-# Settings -> Secrets
-#
-# Name the secret:
-# GEMINI_API_KEY
-#
-# Hugging Face will then make it available securely to this app.
-
-api_key = os.getenv("GEMINI_API_KEY")
-
-if not api_key:
-    raise ValueError(
-        "GEMINI_API_KEY was not found. "
-        "Please add it under your Hugging Face Space Settings -> Secrets."
-    )
-
-client = genai.Client(api_key=api_key)
-
-
-# -----------------------------
-# 3. ZeroGPU placeholder
-# -----------------------------
-# Free Hugging Face accounts may use ZeroGPU.
-#
-# Hugging Face expects a ZeroGPU Space to contain at least one
-# function marked with @spaces.GPU.
-#
-# Our finance app does NOT actually need a GPU because Gemini
-# runs on Google's servers.
-#
-# This small placeholder function simply allows the Space to start.
-
+# Required for ZeroGPU Spaces
 @spaces.GPU(duration=5)
 def gpu_placeholder():
     return "ready"
 
 
-# -----------------------------
-# 4. Main function of the app
-# -----------------------------
-# This function runs whenever the user presses Submit.
-#
-# Gradio sends the values from the input boxes into this function.
-# The function then returns the AI response back to Gradio.
-
-def finance_helper(
-    income,
-    expenses,
-    cash,
-    investments,
-    debt,
-    goal
-):
+def finance_helper(income, expenses, cash, investments, debt, goal):
 
     try:
-
-        # -----------------------------
-        # Clean up the user's inputs
-        # -----------------------------
-        # If a user leaves a number blank, we treat it as zero.
-
         income = float(income or 0)
         expenses = float(expenses or 0)
         cash = float(cash or 0)
         investments = float(investments or 0)
         debt = float(debt or 0)
 
-
-        # -----------------------------
-        # Do some calculations ourselves
-        # -----------------------------
-        # We do NOT need an LLM to calculate simple mathematics.
-        #
-        # This is an important idea when building AI applications:
-        #
-        # Use normal code for things that are predictable.
-        # Use the LLM for interpretation, reasoning, and communication.
-
-        monthly_surplus = income - expenses
-
-        if income > 0:
-            savings_rate = monthly_surplus / income * 100
-        else:
-            savings_rate = 0
-
-        if expenses > 0:
-            emergency_fund_months = cash / expenses
-        else:
-            emergency_fund_months = 0
-
-
-        # -----------------------------
-        # Create the prompt for Gemini
-        # -----------------------------
-        # This is where much of the "intelligence" of the app comes from.
-        #
-        # Students can customise this section to create completely
-        # different applications.
+        surplus = income - expenses
+        savings_rate = surplus / income * 100 if income > 0 else 0
+        emergency_months = cash / expenses if expenses > 0 else 0
 
         prompt = f"""
 You are a friendly personal finance education assistant.
 
-Your role is to help the user understand their financial situation
-and think about sensible financial priorities.
-
 Provide general educational guidance only.
+Do not recommend specific financial products or investments.
 
-Do not recommend specific stocks, securities, investment products,
-insurance products, or financial institutions.
-
-Here is the user's financial information:
+Financial information:
 
 Monthly income: ${income:,.0f}
-
 Monthly expenses: ${expenses:,.0f}
-
-Monthly surplus: ${monthly_surplus:,.0f}
-
+Monthly surplus: ${surplus:,.0f}
 Savings rate: {savings_rate:.1f}%
-
 Cash savings: ${cash:,.0f}
-
-Emergency fund:
-{emergency_fund_months:.1f} months of current expenses
-
+Emergency fund: {emergency_months:.1f} months
 Investments: ${investments:,.0f}
-
 Debt: ${debt:,.0f}
 
-The user's main financial goal is:
-
+Main financial goal:
 {goal}
 
-
-Please provide a concise and useful response using the following format:
+Please provide:
 
 ## Financial Snapshot
-
-Summarise the user's current financial position in simple language.
-
-
 ## What Looks Good
-
-Identify two positive aspects of the user's financial position.
-
-
 ## Three Priorities
-
-Recommend three general priorities the user could consider.
-
-
 ## One Thing to Watch
-
-Identify one potential financial concern or area that deserves attention.
-
-
 ## Questions to Think About
 
-Provide three useful questions that could help the user think more carefully
-about their financial goals.
-
-
-Keep the response practical, easy to understand, and relatively concise.
+Keep the response practical and concise.
 """
 
-
-        # -----------------------------
-        # 5. Send the prompt to Gemini
-        # -----------------------------
-
         response = client.models.generate_content(
             model="gemini-3.6-flash",
             contents=prompt
         )
 
-
-        # -----------------------------
-        # 6. Return Gemini's answer
-        # -----------------------------
-        # Whatever we return here will appear in the Gradio output box.
-
         return response.text
 
-
     except Exception as e:
+        return f"## Error\n\n{str(e)}"
 
-        # -----------------------------
-        # Show errors in the app
-        # -----------------------------
-        # This is useful for learning and debugging.
-        #
-        # Instead of the app appearing to do nothing,
-        # the user will see the actual error message.
-
-        return f"""
-## Something went wrong
 
-**Error type:** {type(e).__name__}
+# Build the app
+with gr.Blocks(title="Personal Finance Helper") as demo:
 
-**Error message:** {str(e)}
-
-Please check your API key and try again.
-"""
+    gr.Markdown(
+        """
+        # 💰 Personal Finance Helper
 
+        Enter a few details below to receive an AI generated financial health check.
 
-# -----------------------------
-# 7. Build the user interface
-# -----------------------------
-# Gradio creates the web page for us.
-#
-# "inputs" defines what users can enter.
-# "outputs" defines where the result will appear.
-
-demo = gr.Interface(
-
-    fn=finance_helper,
-
-    inputs=[
-
-        gr.Number(
-            label="Monthly Income",
-            value=10000
-        ),
-
-        gr.Number(
-            label="Monthly Expenses",
-            value=6000
-        ),
-
-        gr.Number(
-            label="Cash Savings",
-            value=30000
-        ),
-
-        gr.Number(
-            label="Investments",
-            value=50000
-        ),
-
-        gr.Number(
-            label="Debt",
-            value=10000
-        ),
-
-        gr.Textbox(
-            label="What is your main financial goal?",
-            placeholder="For example: Buy a home in five years"
-        )
-    ],
+        *For educational purposes only. This is not financial advice.*
+        """
+    )
 
-    outputs=gr.Markdown(
-        label="Your AI Financial Review"
-    ),
+    with gr.Row():
+        income = gr.Number(label="Monthly Income")
+        expenses = gr.Number(label="Monthly Expenses")
 
-    title="💰 Personal Finance Helper",
+    with gr.Row():
+        cash = gr.Number(label="Cash Savings")
+        investments = gr.Number(label="Investments")
+        debt = gr.Number(label="Debt")
 
-    description="""
-Enter a few details about your finances.
+    goal = gr.Textbox(
+        label="Main Financial Goal",
+        placeholder="For example: Buy a home in five years"
+    )
 
-The app will calculate some basic financial indicators and use AI
-to provide a simple financial health check.
+    submit = gr.Button(
+        "Analyse My Finances",
+        variant="primary"
+    )
 
-This tool is for educational purposes only and is not financial advice.
-"""
-)
+    output = gr.Markdown(
+        label="AI Financial Review"
+    )
 
+    submit.click(
+        fn=finance_helper,
+        inputs=[
+            income,
+            expenses,
+            cash,
+            investments,
+            debt,
+            goal
+        ],
+        outputs=output,
+        show_progress="full"
+    )
 
-# -----------------------------
-# 8. Start the app
-# -----------------------------
-# Hugging Face automatically publishes the Gradio interface.
-#
-# We turn off SSR because it can sometimes create unnecessary
-# server issues in this simple Space.
 
 demo.launch(
     ssr_mode=False