diff --git a/.DS_Store b/.DS_Store new file mode 100644 index 0000000..a5eaaa4 Binary files /dev/null and b/.DS_Store differ diff --git a/src/router/user.js b/src/router/user.js index c164a2d..8218dbc 100644 --- a/src/router/user.js +++ b/src/router/user.js @@ -172,6 +172,8 @@ module.exports = (middleware, users, plans, assets) => { const progress = await calculateProgress(plan, assets, users, req.session.userId); + const suggestions = await generateSuggestions(totalUserAssetValue, plan); + res.render('planDetail', { user: req.session.user, plan: plan, @@ -179,7 +181,7 @@ module.exports = (middleware, users, plans, assets) => { totalUserAssetValue: totalUserAssetValue, assets: userAssets, progress: progress, - suggestions: await suggestions.generateSuggestions(), + suggestions: suggestions, }); } catch (err) { diff --git a/src/util/suggestions.js b/src/util/suggestions.js index 9135a9f..d16e924 100644 --- a/src/util/suggestions.js +++ b/src/util/suggestions.js @@ -10,10 +10,14 @@ async function generateFact(factInput) { return response.text; } -async function generateSuggestions() { +async function generateSuggestions(totalUserAssetValue, plan) { const response = await ai.models.generateContent({ model: "gemini-2.0-flash", - contents: "Return a mock investment suggestion 3-5 lines only, please do not include any extra text or explanation.", + contents: `Return an investment suggestion 3-5 based on the user's total asset value and their plan. Please do not include any extra text or explanation. + + Total User Asset Value: ${totalUserAssetValue} + + Plan: ${plan}`, }); return response.text; }