feat : 감정 조회 API 추가

Author: duehee

app/diary/router.py

@@ -550,7 +550,7 @@ async def create_diary_with_emotion_based_recommendation(
         combined_embedding = kobert.get_embedding(best_sentence)
 
         emotion_to_genres = {
-            0: ["댄스", "랩/힙합"],
+            0: ["댄스", "록/메탈"],
             1: ["R&B/Soul", "댄스"],
             2: ["인디음악", "R&B/Soul"],
             3: ["R&B/Soul", "인디음악"],

app/emotion/router.py

@@ -1,7 +1,12 @@
 import logging
 import torch
 from app.emotion.models import tokenizer, model
+from fastapi import APIRouter, Depends
+from sqlalchemy.orm import Session
+from app.database import get_db
+from app.emotion.models import EmotionType
 
+router = APIRouter()
 logger = logging.getLogger(__name__)
 
 def predict_emotion(text: str):
@@ -28,4 +33,9 @@ def predict_emotion(text: str):
             score = topk.values[0][i].item()
             logger.info(f"[감정 분석 Top-{i + 1}] ID: {idx}, 확신도: {score:.4f}")
 
-    return pred_index, probs.squeeze().tolist()
+    return pred_index, probs.squeeze().tolist()
+
+@router.get("/emotions", response_model=list[dict])
+def get_emotions(db: Session = Depends(get_db)):
+    emotions = db.query(EmotionType).all()
+    return [{"id": e.id, "name": e.name} for e in emotions]

app/genre/router.py

@@ -3,7 +3,8 @@
 from app.database import get_db, get_mongodb
 from app.genre.models import Genre
 from app.genre.schemas import GenreResponse
-
+from fastapi.responses import JSONResponse
+from collections import Counter
 router = APIRouter()
 
 @router.get("", response_model=list[GenreResponse], summary="모든 음악 장르 조회", description="저장된 모든 음악 장르를 조회합니다.")
@@ -31,3 +32,16 @@ async def add_genres_from_mongodb(mongodb=Depends(get_mongodb), db: Session = De
         db.commit()
 
     return {"message": f"{len(new_genres)}개 장르 추가 완료"}
+
+@router.get("/songs/genre-counts")
+async def get_genre_counts(db: Session = Depends(get_mongodb)):
+    genre_counter = Counter()
+
+    cursor = db.song_meta.find({}, {"genre": 1})
+
+    async for doc in cursor:
+        genre_str = doc.get("genre", "")
+        genres = [g.strip() for g in genre_str.split(",") if g.strip()]
+        genre_counter.update(genres)
+
+    return JSONResponse(content=dict(genre_counter))

app/main.py

@@ -8,6 +8,7 @@
 from app.songs.router import router as song_router
 from app.embedding.router import router as embedding_router
 from app.statistics.router import router as statistics_router
+from app.emotion.router import router as emotion_router
 
 app = FastAPI()
 
@@ -18,6 +19,7 @@
 app.include_router(song_router, prefix="/songs", tags=["songs"])
 app.include_router(embedding_router, prefix="/embedding", tags=["embedding"])
 app.include_router(statistics_router, prefix="/statistics", tags=["statistics"])
+app.include_router(emotion_router, prefix="/emotion", tags=["emotion"])
 
 app.add_middleware(
     CORSMiddleware,