Update src/services/ai_modal_engine.py

src/services/ai_modal_engine.py

@@ -14,7 +14,7 @@ class AiModalEngine:
 
             client = genai.Client(api_key=api_key)
             
-            # 100% RETAINED ORIGINAL PARSONA
+            # PARSONA
             instruction = f"""
             PARSONA:
             You are the QuantVAT AI Trading Journal Auditor, a senior Risk Manager and Trading Psychologist with 50 years trading experience like a Market Wizard. 
@@ -34,12 +34,18 @@ class AiModalEngine:
             {context}
             """
 
-            # Switched to Flash-Preview for quota safety & token efficiency
             response = client.models.generate_content(
                 model='gemini-3-flash-preview',
                 contents=f"Analyze this Ledger: {context}",
                 config=types.GenerateContentConfig(system_instruction=instruction)
             )
+
+            history = [
+                {"role": "user", "parts": [{"text": f"Analyze this Ledger: {context}"}]},
+                {"role": "model", "parts": [{"text": response.text}]}
+            ]
+            db.collection('users').document(uid).set({"ai_history": history}, merge=True)
+
             return response.text
             
         except Exception as e:
@@ -56,7 +62,7 @@ class AiModalEngine:
             api_key = get_user_keys(uid).get('gemini_key')
             client = genai.Client(api_key=api_key)
             
-            # Use Flash for continuous chat to prevent "Token Drain"
+            # Use Flash for continuous chat
             chat = client.chats.create(
                 model='gemini-3-flash-preview',
                 history=history,
@@ -67,7 +73,7 @@ class AiModalEngine:
             
             response = chat.send_message(prompt)
             
-            # Sync back to Firestore (Standardizing the format for google-genai)
+            # Sync back to Firestore
             history.append({"role": "user", "parts": [{"text": prompt}]})
             history.append({"role": "model", "parts": [{"text": response.text}]})
             db.collection('users').document(uid).set({"ai_history": history}, merge=True)