Update routes/stylist.py

routes/stylist.py

@@ -357,6 +357,69 @@ def get_courtesy_consultations(user_id: str) -> Dict[str, Any]:
             "monthly_courtesy_consultations": list(monthly_data.values())
         }
 
+def get_monthly_likes(user_id: str) -> Dict[str, Any]:
+    ny_timezone = pytz.timezone('America/New_York')
+    now_ny = datetime.now(ny_timezone)
+    monthly_data = {}
+    
+    # Inicializar dados para os últimos 6 meses
+    for i in range(6):
+        target_date = now_ny - relativedelta(months=i)
+        month_num = target_date.month
+        month_name = target_date.strftime('%b')
+        year = target_date.year
+        month_key = f"{year}-{month_num}"
+        monthly_data[month_key] = {
+            "month": month_num,
+            "name": month_name,
+            "current": (month_num == now_ny.month and year == now_ny.year),
+            "likes_count": 0
+        }
+    
+    # Calcular data de início (6 meses atrás)
+    start_date = (now_ny - relativedelta(months=6)).strftime('%Y-%m-%d')
+    
+    try:
+        # Primeiro precisamos obter os itens do feed deste usuário
+        feed_url = f"{SUPABASE_URL}/rest/v1/feed?user_id=eq.{user_id}"
+        feed_response = requests.get(feed_url, headers=SUPABASE_HEADERS)
+        
+        if feed_response.status_code == 200:
+            feed_items = feed_response.json()
+            feed_item_ids = [item.get("id") for item in feed_items]
+            
+            if feed_item_ids:
+                # Agora buscar likes para esses itens de feed
+                # Usando o operador "in" do Supabase para buscar múltiplos IDs
+                likes_url = f"{SUPABASE_URL}/rest/v1/likes?feed_item_id=in.({','.join(map(str, feed_item_ids))})&created_at=gte.{start_date}"
+                likes_response = requests.get(likes_url, headers=SUPABASE_HEADERS)
+                
+                if likes_response.status_code == 200:
+                    likes = likes_response.json()
+                    
+                    # Contar likes por mês
+                    for like in likes:
+                        # Converter a data para o fuso horário de NY
+                        like_date_str = like.get("created_at")
+                        like_date = datetime.fromisoformat(like_date_str.replace('Z', '+00:00')).astimezone(ny_timezone)
+                        
+                        month_key = f"{like_date.year}-{like_date.month}"
+                        if month_key in monthly_data:
+                            monthly_data[month_key]["likes_count"] += 1
+        
+        # Convertendo para lista e ordenando por mês (do mais recente para o mais antigo)
+        result = list(monthly_data.values())
+        result.sort(key=lambda x: (now_ny.year * 12 + now_ny.month - (x["month"])) % 12)
+        
+        return {
+            "monthly_likes": result
+        }
+    except Exception as e:
+        logger.error(f"❌ Error getting monthly likes: {str(e)}")
+        return {
+            "monthly_likes": list(monthly_data.values())
+        }
+
 @router.post("/bank_account_dashboard_link")
 async def generate_bank_account_dashboard_link(data: UserIDRequest):
     try:
@@ -407,6 +470,9 @@ def get_dashboard(user_token: str = Header(None, alias="User-key"), page: int =
         # Obter dados de consultas gratuitas dos últimos 6 meses
         courtesy_data = get_courtesy_consultations(user_id)
         
+        # Obter dados de likes dos últimos 6 meses
+        likes_data = get_monthly_likes(user_id)
+        
         # Obter detalhes da conta conectada
         account_details = get_connected_account_details(stripe_id)
         
@@ -419,6 +485,7 @@ def get_dashboard(user_token: str = Header(None, alias="User-key"), page: int =
             "monthly_revenue": revenue_data["monthly_data"],
             "total_revenue_last_6_months": revenue_data["total_last_6_months"],
             "monthly_courtesy_consultations": courtesy_data["monthly_courtesy_consultations"],
+            "monthly_likes": likes_data["monthly_likes"],
             "total_followers": get_total_followers(user_id),
             "total_subscribers": get_total_subscribers(user_id),
             "account_details": account_details,