Merge branch 'main' of https://huggingface.co/spaces/ATInc1/AIdea-Server

src/api/recommendation_routes.py

@@ -1,4 +1,4 @@
-from fastapi import APIRouter, Depends
+from fastapi import APIRouter, Depends, HTTPException, Query
 from typing import Dict, Any
 from src.auth.dependencies import get_current_user
 from src.db.firebase import get_firebase_db
@@ -9,20 +9,118 @@ logger = setup_logger(__name__)
 router = APIRouter(prefix="/recommendations", tags=["Recommendations"])
 recommendation_service = RecommendationService()
 
+GENERAL_VIDEOS = [
+    {
+        "videoId": "aircAruvnKk",
+        "title": "But what is a neural network?",
+        "channelTitle": "3Blue1Brown",
+        "thumbnail": "https://img.youtube.com/vi/aircAruvnKk/hqdefault.jpg",
+    },
+    {
+        "videoId": "rfscVS0vtbw",
+        "title": "Learn Python in 4 hours",
+        "channelTitle": "freeCodeCamp",
+        "thumbnail": "https://img.youtube.com/vi/rfscVS0vtbw/hqdefault.jpg",
+    },
+    {
+        "videoId": "8jLOx1hD3_o",
+        "title": "Harvard CS50 2023",
+        "channelTitle": "CS50",
+        "thumbnail": "https://img.youtube.com/vi/8jLOx1hD3_o/hqdefault.jpg",
+    },
+    {
+        "videoId": "kqtD5dpn9C8",
+        "title": "Python for Beginners – Full Course",
+        "channelTitle": "Programming with Mosh",
+        "thumbnail": "https://img.youtube.com/vi/kqtD5dpn9C8/hqdefault.jpg",
+    },
+    {
+        "videoId": "Ke90Tje7VS0",
+        "title": "React JS Crash Course",
+        "channelTitle": "Traversy Media",
+        "thumbnail": "https://img.youtube.com/vi/Ke90Tje7VS0/hqdefault.jpg",
+    },
+    {
+        "videoId": "bMknfKXIFA8",
+        "title": "Machine Learning Course for Beginners",
+        "channelTitle": "freeCodeCamp",
+        "thumbnail": "https://img.youtube.com/vi/bMknfKXIFA8/hqdefault.jpg",
+    },
+    {
+        "videoId": "ua-CiDNNj30",
+        "title": "Fullstack Web Dev in 2024",
+        "channelTitle": "Fireship",
+        "thumbnail": "https://img.youtube.com/vi/ua-CiDNNj30/hqdefault.jpg",
+    },
+    {
+        "videoId": "qiQR5rTSshw",
+        "title": "C++ Tutorial for Beginners",
+        "channelTitle": "freeCodeCamp",
+        "thumbnail": "https://img.youtube.com/vi/qiQR5rTSshw/hqdefault.jpg",
+    },
+    {
+        "videoId": "zOjov-2OZ0E",
+        "title": "100+ Linux Things You Should Know",
+        "channelTitle": "Fireship",
+        "thumbnail": "https://img.youtube.com/vi/zOjov-2OZ0E/hqdefault.jpg",
+    },
+    {
+        "videoId": "W6NZfCO5SIk",
+        "title": "JavaScript Tutorial for Beginners",
+        "channelTitle": "Programming with Mosh",
+        "thumbnail": "https://img.youtube.com/vi/W6NZfCO5SIk/hqdefault.jpg",
+    },
+]
+
+MIN_INTERACTIONS_FOR_PERSONALIZATION = 5
+
+
 @router.get("")
-async def get_general_recommendations(
+async def get_recommendations(
     current_user=Depends(get_current_user),
-    db=Depends(get_firebase_db)
+    db=Depends(get_firebase_db),
+    limit: int = Query(default=5, ge=1, le=20),
 ) -> Dict[str, Any]:
-    logger.info(f"🎯 Getting recommendations for user: {current_user['uid']}")
-    
-    recommendations = await recommendation_service.get_recommendations_for_user(
-        db=db,
-        user_id=current_user['uid'],
-        limit=5
-    )
-    
+
+    user_id = current_user.id  # ✅ your model uses "id"
+
+    if not user_id:
+        raise HTTPException(status_code=401, detail="Invalid user")
+
+    logger.info(f"🎯 Getting recommendations for user: {user_id}")
+
+    try:
+        # ✅ Use notesCount directly from user
+        notes_count = getattr(current_user, "notesCount", 0) or 0
+
+        is_personalized = notes_count >= MIN_INTERACTIONS_FOR_PERSONALIZATION
+
+        if is_personalized:
+            logger.info(f"✅ Personalized mode for {user_id} ({notes_count} notes)")
+            
+            # ⚠️ for now you can still return general videos
+            # until you build real recommender
+            recommendations = await recommendation_service.get_recommendations_for_user(
+                db=db,
+                user_id=user_id,
+                limit=limit,
+            )
+        else:
+            logger.info(f"🌱 General mode for {user_id} ({notes_count}/{MIN_INTERACTIONS_FOR_PERSONALIZATION} notes)")
+            recommendations = GENERAL_VIDEOS[:limit]
+
+    except Exception as e:
+        logger.error(f"❌ Failed to get recommendations for {user_id}: {e}")
+        raise HTTPException(
+            status_code=500,
+            detail="Failed to fetch recommendations.",
+        )
+
     return {
         "status": "success",
-        "recommendations": recommendations
+        "is_personalized": is_personalized,
+        "interactions_count": notes_count,  # ✅ mapped correctly
+        "interactions_needed": MIN_INTERACTIONS_FOR_PERSONALIZATION,
+        "count": len(recommendations),
+        "recommendations": recommendations,
     }

src/db/models.py

@@ -19,6 +19,7 @@ class User(BaseModel):
     age: Optional[int] = Field(default=None)
     gender: Optional[str] = Field(default=None)
     created_at: datetime = Field(default_factory=datetime.utcnow)
+    notesCount: Optional[int] = Field(default=0)
 
     class Config:
         from_attributes = True