src/generate_guide.py

# generate_guide.py

import openai
import streamlit as st
import os

# 💡 OPENAI_API_KEY 환경 변수 확인 및 설정 (초기화 중복 방지)
@st.cache_resource(show_spinner="OpenAI 초기화 중...")
def init_openai():
    api_key = os.environ.get("OPENAI_API_KEY")
    if not api_key:
        raise RuntimeError(
            "❌ OPENAI_API_KEY가 환경변수로 설정되어 있지 않습니다. "
            "Hugging Face Secrets에 'OPENAI_API_KEY'를 등록해주세요."
        )
    openai.api_key = api_key
    return openai


# 📚 촬영 가이드 생성 함수
def generate_photo_guide(user_caption: str, selected_image_descriptions: list) -> str:
    prompt = f"""
    아래는 사용자가 업로드한 사진 설명입니다:
    - {user_caption}

    아래는 사용자가 참고하고 싶다고 선택한 사진 설명들입니다:
    {chr(10).join([f"- {desc}" for desc in selected_image_descriptions])}

    위 정보를 종합하여, 사용자가 사진을 다시 찍을 때 참고할 수 있도록
    아래의 내용을 자연어로 작성해주세요:

    1. 어떤 포즈를 취하면 좋을지
    2. 카메라 위치는 어디에 두면 좋은지 (높이, 각도 등)
    3. 인물 배치는 어떻게 하면 좋은지 (정면/측면, 거리 등)
    4. 전체 구도 팁이 있다면 알려주세요
    """

    try:
        init_openai()  # ✅ API 키 설정 보장
        response = openai.ChatCompletion.create(
            model="gpt-3.5-turbo",
            messages=[{"role": "user", "content": prompt}],
            temperature=0.7,
        )
        return response.choices[0].message.content.strip()
    except Exception as e:
        return f"❌ GPT 호출에 실패했습니다: {str(e)}"