Upd analytics tracker

helpers/coder.py

@@ -20,7 +20,8 @@ async def generate_code_artifacts(
     context_text: str,
     web_context: str,
     gemini_rotator,
-    nvidia_rotator
+    nvidia_rotator,
+    user_id: str = ""
 ) -> str:
     """Generate code (files-by-files) with explanations using Gemini Pro.
 
@@ -51,6 +52,27 @@ async def generate_code_artifacts(
     selection = {"provider": "gemini", "model": "gemini-2.5-pro"}
 
     logger.info(f"[CODER] Generating code for subsection {subsection_id} (task='{task[:60]}...')")
+    # Track analytics
+    try:
+        from utils.analytics import get_analytics_tracker
+        tracker = get_analytics_tracker()
+        if tracker and user_id:
+            await tracker.track_agent_usage(
+                user_id=user_id,
+                agent_name="coding",
+                action="generate_code",
+                context="report_coding",
+                metadata={"subsection_id": subsection_id}
+            )
+            await tracker.track_model_usage(
+                user_id=user_id,
+                model_name=selection["model"],
+                provider=selection["provider"],
+                context="report_coding",
+                metadata={"subsection_id": subsection_id}
+            )
+    except Exception:
+        pass
     code_md = await generate_answer_with_model(selection, system_prompt, user_prompt, gemini_rotator, nvidia_rotator)
     code_md = (code_md or "").strip()
 

helpers/diagram.py

@@ -27,7 +27,8 @@ async def generate_mermaid_diagram(
     nvidia_rotator,
     render_error: str = "",
     retry: int = 0,
-    max_retries: int = 5
+    max_retries: int = 5,
+    user_id: str = ""
 ) -> str:
     from utils.api.router import generate_answer_with_model
 
@@ -70,6 +71,27 @@ async def generate_mermaid_diagram(
     selection = {"provider": "nvidia_large", "model": "openai/gpt-oss-120b"}
 
     logger.info(f"[DIAGRAM] Generating Mermaid (retry={retry}/{max_retries})")
+    # Track analytics
+    try:
+        from utils.analytics import get_analytics_tracker
+        tracker = get_analytics_tracker()
+        if tracker and user_id:
+            await tracker.track_agent_usage(
+                user_id=user_id,
+                agent_name="diagram",
+                action="generate_mermaid",
+                context="report_diagram",
+                metadata={"retry": retry}
+            )
+            await tracker.track_model_usage(
+                user_id=user_id,
+                model_name=selection["model"],
+                provider=selection["provider"],
+                context="report_diagram",
+                metadata={"retry": retry}
+            )
+    except Exception:
+        pass
     diagram = await generate_answer_with_model(selection, sys_prompt, user_prompt, gemini_rotator, nvidia_rotator)
     diagram = (diagram or "").strip()
 

routes/reports.py

@@ -138,6 +138,19 @@ async def generate_report(
     
     # Use enhanced instructions for better CoT planning
     cot_plan = await generate_cot_plan(enhanced_instructions, file_summary, context_text, web_context_block, nvidia_rotator, gemini_rotator)
+    # Track planning agent
+    try:
+        tracker = get_analytics_tracker()
+        if tracker:
+            await tracker.track_agent_usage(
+                user_id=user_id,
+                agent_name="planning",
+                action="plan_report",
+                context="report_planning",
+                metadata={"project_id": project_id, "session_id": session_id}
+            )
+    except Exception:
+        pass
     
     # Step 2: Execute detailed subtasks based on CoT plan
     logger.info("[REPORT] Executing detailed subtasks")
@@ -151,6 +164,19 @@ async def generate_report(
     comprehensive_report = await synthesize_comprehensive_report(
         enhanced_instructions, cot_plan, detailed_analysis, eff_name, report_words, gemini_rotator, nvidia_rotator
     )
+    # Track synthesis (report agent)
+    try:
+        tracker = get_analytics_tracker()
+        if tracker:
+            await tracker.track_agent_usage(
+                user_id=user_id,
+                agent_name="report",
+                action="synthesize_report",
+                context="report_synthesis",
+                metadata={"project_id": project_id, "session_id": session_id}
+            )
+    except Exception:
+        pass
     
     # Update status: Generating answer (final report generation)
     update_report_status(session_id, "generating", "Generating answer...", 80)
@@ -482,7 +508,8 @@ async def execute_detailed_subtasks(cot_plan: Dict[str, Any], context_text: str,
                             context_text=context_text,
                             web_context=web_context,
                             gemini_rotator=gemini_rotator,
-                            nvidia_rotator=nvidia_rotator
+                            nvidia_rotator=nvidia_rotator,
+                            user_id=user_id
                         )
                         subtask_result = subtask_result + "\n\n" + code_markdown
                         try:
@@ -870,7 +897,7 @@ Create a comprehensive, authoritative report with proper hierarchical structure
         try:
             if should_generate_mermaid(instructions, report):
                 logger.info("[REPORT] Considering Mermaid diagram generation")
-                diagram = await generate_mermaid_diagram(instructions, detailed_analysis, gemini_rotator, nvidia_rotator)
+                diagram = await generate_mermaid_diagram(instructions, detailed_analysis, gemini_rotator, nvidia_rotator, user_id=user_id)
                 if diagram:
                     report = ("## Diagrams\n\n" + "```mermaid\n" + diagram.strip() + "\n```\n\n" + report)
                 else:

static/index.html

@@ -211,7 +211,7 @@
                 </div>
                 <div class="analytics-actions">
                   <button id="refresh-analytics" class="btn-primary">Refresh Data</button>
-                  <button id="debug-analytics" class="btn-secondary">Debug Load</button>
+                  <!-- <button id="debug-analytics" class="btn-secondary">Debug Load</button> -->
                 </div>
               </div>
               

utils/analytics.py

@@ -114,11 +114,12 @@ class AnalyticsTracker:
             # Daily usage trends
             daily_pipeline = [
                 {"$match": {"user_id": user_id, "timestamp": {"$gte": cutoff_time}}},
+                {"$addFields": {"date": {"$toDate": {"$multiply": ["$timestamp", 1000]}}}},
                 {"$group": {
                     "_id": {
-                        "year": {"$year": {"$dateFromTimestamp": {"$multiply": ["$timestamp", 1000]}}},
-                        "month": {"$month": {"$dateFromTimestamp": {"$multiply": ["$timestamp", 1000]}}},
-                        "day": {"$dayOfMonth": {"$dateFromTimestamp": {"$multiply": ["$timestamp", 1000]}}}
+                        "year": {"$year": "$date"},
+                        "month": {"$month": "$date"},
+                        "day": {"$dayOfMonth": "$date"}
                     },
                     "total_requests": {"$sum": 1},
                     "model_requests": {"$sum": {"$cond": [{"$eq": ["$type", "model"]}, 1, 0]}},
@@ -161,12 +162,20 @@ class AnalyticsTracker:
             model_pipeline = [
                 {"$match": {"type": "model", "timestamp": {"$gte": cutoff_time}}},
                 {"$group": {
-                    "_id": "$model_name",
+                    "_id": {
+                        "provider": "$provider",
+                        "model": "$model_name"
+                    },
                     "count": {"$sum": 1},
-                    "unique_users": {"$addToSet": "$user_id"},
-                    "provider": {"$first": "$provider"}
+                    "unique_users": {"$addToSet": "$user_id"}
+                }},
+                {"$project": {
+                    "_id": 0,
+                    "model_name": "$_id.model",
+                    "provider": "$_id.provider",
+                    "count": 1,
+                    "unique_user_count": {"$size": "$unique_users"}
                 }},
-                {"$addFields": {"unique_user_count": {"$size": "$unique_users"}}},
                 {"$sort": {"count": -1}}
             ]
             
@@ -176,12 +185,25 @@ class AnalyticsTracker:
             agent_pipeline = [
                 {"$match": {"type": "agent", "timestamp": {"$gte": cutoff_time}}},
                 {"$group": {
-                    "_id": "$agent_name",
+                    "_id": {
+                        "agent": "$agent_name",
+                        "action": "$action"
+                    },
                     "count": {"$sum": 1},
-                    "unique_users": {"$addToSet": "$user_id"},
-                    "actions": {"$addToSet": "$action"}
+                    "unique_users": {"$addToSet": "$user_id"}
+                }},
+                {"$group": {
+                    "_id": "$_id.agent",
+                    "count": {"$sum": "$count"},
+                    "unique_users": {"$addToSet": "$unique_users"},
+                    "actions": {"$addToSet": "$_id.action"}
+                }},
+                {"$project": {
+                    "_id": 1,
+                    "count": 1,
+                    "actions": 1,
+                    "unique_user_count": {"$size": {"$setUnion": "$unique_users"}}
                 }},
-                {"$addFields": {"unique_user_count": {"$size": "$unique_users"}}},
                 {"$sort": {"count": -1}}
             ]