Update assistant.py

assistant.py

@@ -8,65 +8,52 @@ from dotenv import load_dotenv
 load_dotenv()
 
 assistant_bp = Blueprint("assistant", __name__)
-
 GROQ_API_KEY = os.getenv("GROQ_API_KEY", "")
 
-
 def _build_financial_context(user_id: str) -> str:
     db = get_db()
     docs = list(db["transactions"].find({"user_id": user_id}).sort("date", -1).limit(20))
-
     if not docs:
         return "The user has no transactions recorded yet."
-
     income = sum(d["amount"] for d in docs if d["type"] == "income")
     expense = sum(d["amount"] for d in docs if d["type"] == "expense")
     balance = income - expense
-
     lines = [
-        f"Total income (from last 20 transactions): ₹{income:,.2f}",
-        f"Total expenses (from last 20 transactions): ₹{expense:,.2f}",
+        f"Total income: ₹{income:,.2f}",
+        f"Total expenses: ₹{expense:,.2f}",
         f"Net balance: ₹{balance:,.2f}",
         "",
-        "Recent transactions (newest first):",
+        "Recent transactions:",
     ]
     for d in docs[:10]:
         sign = "+" if d["type"] == "income" else "-"
         date_str = d["date"].strftime("%d %b %Y") if d.get("date") else "unknown date"
         lines.append(f"  {sign}₹{d['amount']:,.2f} | {d['title']} | {d['category']} | {date_str}")
-
     return "\n".join(lines)
 
-
-SYSTEM_PROMPT_TEMPLATE = """You are a smart, friendly personal finance assistant for WealthTracker — a personal finance app.
+SYSTEM_PROMPT = """You are a smart, friendly personal finance assistant for WealthTracker.
 You help users understand their spending, plan budgets, and make better financial decisions.
 Be concise, practical, and encouraging. Use Indian Rupees (₹) when referencing amounts.
 
-Here is the user's current financial snapshot:
+User's financial snapshot:
 {context}
 
-Answer the user's question based on this data when relevant. If the question is not finance-related, gently redirect."""
-
+Answer based on this data. If not finance-related, gently redirect."""
 
 @assistant_bp.route("/chat", methods=["POST"])
 @require_auth
 def chat():
     data = request.get_json(silent=True) or {}
     user_message = (data.get("message") or "").strip()
-
     if not user_message:
         return jsonify({"error": "Message is required"}), 400
-
     if not GROQ_API_KEY:
         return jsonify({"error": "Groq API key not configured"}), 503
-
     try:
         financial_context = _build_financial_context(g.user_id)
     except Exception:
         financial_context = "Could not load transaction data."
-
-    system_prompt = SYSTEM_PROMPT_TEMPLATE.format(context=financial_context)
-
+    system_prompt = SYSTEM_PROMPT.format(context=financial_context)
     try:
         client = Groq(api_key=GROQ_API_KEY)
         response = client.chat.completions.create(
@@ -79,6 +66,5 @@ def chat():
         )
         reply = response.choices[0].message.content
         return jsonify({"reply": reply}), 200
-
     except Exception as e:
         return jsonify({"error": f"AI error: {str(e)}"}), 500