Update app.py

app.py

@@ -1,24 +1,23 @@
 import gradio as gr
 import openai
-from transformers import pipeline
-import environ
 
-env = environ.Env()
-environ.Env.read_env()
-API_KEY = env("apikey")
-openai.api_key = API_KEY
-
-def financial_planning(prompt):
-    response = openai.chat.completions.create(
-        model="gpt-3.5-turbo-16k",
-        messages=prompt,
+def financial_planning(prompt, api_key):
+    
+    openai.api_key = api_key
+    
+    message=[
+    {"role": "system", "content": "You are a knowledgeable and insightful Financial Planner dedicated to helping users make informed investment decisions based on their unique financial profiles."},
+    {"role": "user", "content": prompt}
+  ]
+    response = openai.ChatCompletion.create(
+        model="gpt-4o",
+        messages= message,
         max_tokens=500,
         temperature=0.7,
     )
-    return response.choices[0].text.strip()
-
+    return response.choices[0]["message"]["content"]
 
-def generate_advice(name, age, income, expenses, savings, debts, investment_preferences, risk_tolerance, retirement_age, retirement_savings_goal, short_term_goals, long_term_goals, dependents, health_expenses, education_expenses, insurance_policies, major_purchases, emergency_fund, annual_income_growth, annual_expense_growth, currency):
+def generate_advice(name, age, income, expenses, savings, debts, investment_preferences, risk_tolerance, retirement_age, retirement_savings_goal, short_term_goals, long_term_goals, dependents, health_expenses, education_expenses, insurance_policies, major_purchases, emergency_fund, annual_income_growth, annual_expense_growth, currency, api_key):
     prompt = (
         f"Name: {name}\n"
         f"Age: {age}\n"
@@ -42,30 +41,60 @@ def generate_advice(name, age, income, expenses, savings, debts, investment_pref
         f"Annual Expense Growth: {annual_expense_growth}%\n"
         f"Provide a detailed financial plan based on the above information."
     )
-    advice = financial_planning(prompt)
+    advice = financial_planning(prompt, api_key)
     return advice
 
-iface = gr.Interface(
-    fn=generate_advice,
-    inputs=["text", "number", "number", "number", "number", "number", "text",
-        gr.Radio(["Low", "Medium", "High"],label="Risk Tolerance"),
-        gr.Number(label="Retirement Age"),
-        gr.Number(label="Retirement Savings Goal"),
-        gr.Textbox(label="Short-term Goals (comma separated)"),
-        gr.Textbox(label="Long-term Goals (comma separated)"),
-        gr.Number(label="Dependents"),
-        gr.Number(label="Monthly Health Expenses"),
-        gr.Number(label="Monthly Education Expenses"),
-        gr.Textbox(label="Insurance Policies (comma separated)"),
-        gr.Textbox(label="Major Purchases (with estimated costs, comma separated)"),
-        gr.Number(label="Emergency Fund"),
-        gr.Number(label="Annual Income Growth (%)"),
-        gr.Number(label="Annual Expense Growth (%)"),
-        gr.Dropdown(label="Currency", choices=["USD", "EUR", "GBP", "JPY", "INR", "AUD", "CAD", "CHF", "CNY", "Other"]),
-    ],
-    outputs="text",
-    title="Smart Finance Planner: Your Personalized Financial Guide",
-    description="Unlock your financial potential with Smart Finance Planner. Input your financial details and receive a customized, comprehensive financial plan tailored to your goals and lifestyle. Whether you're planning for retirement, saving for a major purchase, or just looking to optimize your investments, our intelligent tool provides expert advice to help you achieve financial success. Get started today and take control of your financial future!"
-)
+# Function to clear inputs
+def clear_inputs():
+    return "", 0, 0, 0, 0, 0, "", "Low", 0, 0, "", "", 0, 0, 0, "", "", 0, 0, 0, "HKD"
+
+
+with gr.Blocks(gr.themes.Base()) as demo:
+    gr.Markdown("<h1 style='text-align: center'>Smart Finance Planner: Your Personalized Financial Guide</h1>")
+    gr.Markdown("<p style='text-align: center'>Unlock your financial potential with Smart Finance Planner. Input your financial details and receive a customized, comprehensive financial plan tailored to your goals and lifestyle. Whether you're planning for retirement, saving for a major purchase, or just looking to optimize your investments, our intelligent tool provides expert advice to help you achieve financial success. Get started today and take control of your financial future!</p>")
+
+    
+    with gr.Row():
+            name = gr.Textbox(label="Name")
+            age = gr.Number(label="Age")
+            income = gr.Textbox(label="Monthly Income")
+            expenses = gr.Textbox(label="Monthly Expenses")
+            savings = gr.Textbox(label="Savings", placeholder = "Overall Savings")
+            debts = gr.Textbox(label="Debts", placeholder="Current Debts")
+            investment_preferences = gr.Textbox(label="Investment Preferences")
+            risk_tolerance = gr.Dropdown(label="Risk Tolerance", choices=["Low", "Medium", "High"])
+            retirement_age = gr.Number(label="Retirement Age")
+            retirement_savings_goal = gr.Textbox(label="Retirement Savings Goal")
+            short_term_goals = gr.Textbox(label="Short-term Goals", placeholder="Comma separated")
+            long_term_goals = gr.Textbox(label="Long-term Goals", placeholder="Comma separated")
+            dependents = gr.Number(label="Dependents")
+            health_expenses = gr.Textbox(label="Monthly Health Expenses")
+            education_expenses = gr.Textbox(label="Education Expenses", placeholder= "Monthly Expense")
+            insurance_policies = gr.Textbox(label="Insurance Policies", placeholder="Comma separated")
+            major_purchases = gr.Textbox(label="Major Purchases", placeholder="Enter estimated costs")
+            emergency_fund = gr.Textbox(label="Emergency Fund")
+            annual_income_growth = gr.Textbox(label="Annual Income Growth", placeholder="In %")
+            annual_expense_growth = gr.Textbox(label="Annual Expense Growth", placeholder="In %")
+            currency = gr.Dropdown(label="Currency", choices=["HKD", "USD", "EUR", "GBP", "JPY", "INR", "AUD", "CAD", "CHF", "CNY", "Other"])
+
+    with gr.Row():
+         api_key = gr.Textbox(label='OpenAI API Key', type='password')
+
+    with gr.Row():
+        submit_btn = gr.Button("Generate Financial Plan")
+        clear_btn = gr.Button("Clear")
+
+    with gr.Row():
+        output = gr.Markdown("<style='text-align: center'>",label="Financial Plan")  
+    
+    submit_btn.click(
+        generate_advice, 
+        inputs=[name, age, income, expenses, savings, debts, investment_preferences, risk_tolerance, retirement_age, retirement_savings_goal, short_term_goals, long_term_goals, dependents, health_expenses, education_expenses, insurance_policies, major_purchases, emergency_fund, annual_income_growth, annual_expense_growth, currency, api_key], 
+        outputs=output
+    )
+    clear_btn.click(
+        clear_inputs, 
+        outputs=[name, age, income, expenses, savings, debts, investment_preferences, risk_tolerance, retirement_age, retirement_savings_goal, short_term_goals, long_term_goals, dependents, health_expenses, education_expenses, insurance_policies, major_purchases, emergency_fund, annual_income_growth, annual_expense_growth, currency]
+    )
 
-iface.launch()
+demo.launch()