Create cinema.py

apps/cinema.py

@@ -0,0 +1,58 @@
+import os
+import json
+from fastapi import APIRouter, HTTPException
+from pydantic import BaseModel
+from huggingface_hub import InferenceClient
+
+# 라우터 설정
+router = APIRouter()
+
+# HF 토큰 (Settings > Secrets에 HF_TOKEN 이 등록되어 있어야 함)
+client = InferenceClient(api_key=os.getenv("HF_TOKEN"))
+
+# 요청 파라미터 모델
+class MovieRequest(BaseModel):
+    title: str
+    director: str
+    tmdb_md: str = ""
+
+@router.post("/movie-info")
+async def get_movie_info(req: MovieRequest):
+    try:
+        context = f"TMDB 정보:\n{req.tmdb_md}" if req.tmdb_md else "TMDB 정보가 제공되지 않았습니다."
+        
+        prompt = f"""
+        당신은 영화 전문 AI 비서입니다. 아래 [제공된 정보]를 바탕으로만 답변하세요.
+
+        [제공된 정보]
+        {context}
+
+        [작성 규칙]
+        1. "summary": 영화 '{req.title}'(감독:{req.director})의 내용을 한국어로 150자 이내로 요약할 것. 내용 왜곡 방지를 위해 150자를 초과할 것 같으면 뒷부분을 '···'으로 생략할 것.
+        2. "ott": 한국에서 시청 가능한 OTT 플랫폼 이름만 쉼표로 나열할 것. (없으면 "-")
+        3. 반드시 아래 JSON 형식으로만 출력하고 다른 설명은 절대 추가하지 마세요.
+
+        [출력 형식]
+        {{
+            "summary": "...",
+            "ott": "..."
+        }}
+        """
+
+        response = client.chat_completion(
+            model="Qwen/Qwen2.5-72B-Instruct",
+            messages=[{"role": "user", "content": prompt}],
+            max_tokens=300,
+            temperature=0.1
+        )
+
+        raw_result = response.choices[0].message.content.strip()
+        
+        # 마크다운 코드 블록(```json)이 붙어 나올 경우 텍스트 전처리
+        if raw_result.startswith("```"):
+            raw_result = raw_result.strip("`").replace("json\n", "")
+
+        return json.loads(raw_result)
+
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))