Update src/generate_guide.py

src/generate_guide.py

@@ -1,42 +1,41 @@
-# generate_guide.py
-
-import openai
-
-openai.api_key = "YOUR_OPENAI_API_KEY"
-
-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. 전체 구도 팁이 있다면 알려주세요
-    """
-
-    # GPT 호출
-    response = openai.ChatCompletion.create(
-        model="gpt-3.5-turbo",
-        messages=[{"role": "user", "content": prompt}],
-        temperature=0.7,
-    )
-
-    guide = response.choices[0].message.content.strip()
-    return guide
-
-# 예시 실행
-if __name__ == "__main__":
-    user_caption = "두 사람이 벽 앞에서 그림자로 하트를 만드는 포즈"
-    selected = [
-        "해가 질 때 그림자만 보이게 벽 앞에서 하트를 만든 두 사람",
-        "측면에서 한 사람이 다른 사람을 찍는 셀카 그림자"
-    ]
-    print(generate_photo_guide(user_caption, selected))
+# generate_guide.py
+
+import openai
+import streamlit as st
+import os
+
+# 💡 API 키 설정을 캐싱해서 초기화 과정을 중복 방지
+@st.cache_resource
+def init_openai():
+    openai.api_key = os.environ["OPENAI_API_KEY"]
+    return openai
+
+openai = init_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. 전체 구도 팁이 있다면 알려주세요
+    """
+
+    # 🧠 GPT 호출
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=[{"role": "user", "content": prompt}],
+        temperature=0.7,
+    )
+
+    return response.choices[0].message.content.strip()
+