Upload ai_modal_engine.py

src/services/ai_modal_engine.py

@@ -0,0 +1,119 @@
+import google.generativeai as genai
+import csv
+import io
+import traceback
+from ..config import db, get_user_keys
+
+class AiModalEngine:
+    @staticmethod
+    def context_to_csv(trades):
+        """Serializes trade list to token-efficient CSV format."""
+        if not trades or not isinstance(trades, list):
+            return str(trades)
+            
+        output = io.StringIO()
+        if len(trades) > 0:
+            # Extract headers from first record
+            keys = trades[0].keys()
+            writer = csv.DictWriter(output, fieldnames=keys)
+            writer.writeheader()
+            writer.writerows(trades)
+        
+        return output.getvalue()
+
+    @staticmethod
+    def initialize_firebase_session(uid, context):
+        """Bootstraps Auditor session and persists initial state to Firestore."""
+        try:
+            keys = get_user_keys(uid)
+            api_key = keys.get('gemini_key')
+            if not api_key: 
+                return "Error: No API Key found in Settings."
+
+            genai.configure(api_key=api_key)
+            
+            # Compression threshold: 20k characters
+            threshold = 20000
+            is_large_data = len(str(context)) > threshold
+            
+            # Toggle data format based on payload size
+            if is_large_data and isinstance(context, list):
+                clean_context = AiModalEngine.context_to_csv(context)
+                data_label = "STRUCTURED CSV DATA"
+            else:
+                clean_context = str(context)
+                data_label = "FILTERED TRADE LOGS"
+
+            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. 
+            Speak with veteran authority. Tone is blunt but constructive.
+
+            MANDATE:
+            1. Analyze the 'WHY' behind execution based on the provided logs.
+            2. STRUCTURE: 
+               - ## 📊 OVERVIEW: 2-sentence performance reality check.
+               - ## 🚩 RED FLAGS: Top 2 execution errors (FOMO, sizing, fear, etc).
+               - ## 💡 THE REMEDY: One specific, actionable rule for the next session.
+            3. FORMAT: Use bold text for emphasis. 
+            4. NO TABLES: Use bullet points only.
+            5. INTERACTION: End with a provocative question about a specific trade.
+
+            {data_label}:
+            {clean_context}
+            """
+
+            # Initialize model with embedded system instructions
+            model = genai.GenerativeModel(
+                model_name='gemini-flash-latest',
+                system_instruction=instruction
+            )
+            
+            chat = model.start_chat(history=[])
+            prompt = "Analyze my execution performance. End with: 'I have analyzed your data. Ready for audit.'"
+            response = chat.send_message(prompt)
+            
+            # Persist turn history to database
+            history = [
+                {"role": "user", "parts": [prompt]}, 
+                {"role": "model", "parts": [response.text]}
+            ]
+            db.collection('users').document(uid).set({"ai_history": history}, merge=True)
+            
+            return response.text
+        except Exception as e:
+            print(f"AI Init Error: {traceback.format_exc()}")
+            return f"System Error: {str(e)}"
+
+    @staticmethod
+    def continue_firebase_chat(uid, prompt):
+        """Resumes active session using state retrieved from Firestore."""
+        try:
+            user_doc = db.collection('users').document(uid).get()
+            data = user_doc.to_dict() if user_doc.exists else {}
+            history = data.get("ai_history", [])
+            
+            api_key = get_user_keys(uid).get('gemini_key')
+            if not api_key: 
+                return "Error: API Key missing."
+
+            genai.configure(api_key=api_key)
+            
+            # Re-apply persona constraints for session continuity
+            model = genai.GenerativeModel(
+                model_name='gemini-flash-latest',
+                system_instruction="PARSONA: QuantVAT AI Trading Journal Auditor. Senior Risk Manager and Trading Psychologist with 50 years trading experience like a Market Wizard."
+            )
+            
+            chat = model.start_chat(history=history)
+            response = chat.send_message(prompt)
+            
+            # Sync new turn to database
+            history.append({"role": "user", "parts": [prompt]})
+            history.append({"role": "model", "parts": [response.text]})
+            db.collection('users').document(uid).set({"ai_history": history}, merge=True)
+            
+            return response.text
+        except Exception as e:
+            print(f"AI Chat Error: {traceback.format_exc()}")
+            return f"Auditor Error: {str(e)}"