Update ui/components.py

ui/components.py

@@ -15,21 +15,19 @@ import hashlib
 import numpy as np
 import json
 from typing import Dict, List, Tuple, Optional
-
+import time
+import requests
 # 추가된 부분: OpenAI 모듈 임포트
 from openai import OpenAI
 
-# os.getenv("LLM_API")에서 API 키를 불러와서 사용
+# 가사 생성을 위한 OpenAI 호출 함수
 client = OpenAI(api_key=os.getenv("LLM_API"))
 
-# 가사 생성을 위한 OpenAI 호출 함수
 def openai_generate_lyrics(topic: str) -> str:
     """
-    사용자가 입력한 '주제(topic)'를 바탕으로 GPT-4.1-mini 모델을 호출하여
-    해당 언어(한국어/영어 등) 기준으로 [verse], [chorus], [bridge] 등의
-    구조를 포함한 노래 가사를 생성하여 반환합니다.
+    주제(topic)를 받아 GPT-4.1-mini로 [verse]/[chorus]/[bridge] 형식의
+    가사를 생성해 반환한다. 최대 3회 재시도한다.
     """
-    # 시스템 프롬프트(명령) : 노래 가사를 작사하는 전문가 역할
     system_prompt = (
         "너는 노래 가사를 작사하는 전문가 역할이다. 이용자가 입력하는 주제에 따라 "
         "이에 관련된 노래 가사를 작성하라. 가사의 규칙은 \"[  ]\"로 구분하여, "
@@ -45,51 +43,46 @@ def openai_generate_lyrics(topic: str) -> str:
         "skies\n\n[verse]\nKeyboard dances on the keys\nMelodies on evening breeze\nCatch "
         "the tune and hold it tight\nIn this moment we take flight\n\"\"\""
     )
-    try:
-        # GPT-4.1-mini 모델에 system / user 메시지 형식으로 호출
-        response = client.responses.create(
-            model="gpt-4.1-mini",
-            input=[
-                {
-                    "role": "system",
-                    "content": [
-                        {
-                            "type": "input_text",
-                            "text": system_prompt
-                        }
-                    ]
-                },
-                {
-                    "role": "user",
-                    "content": [
-                        {
-                            "type": "input_text",
-                            "text": topic
-                        }
-                    ]
-                }
-            ],
-            text={"format": {"type": "text"}},
-            reasoning={},
-            tools=[],
-            temperature=1,
-            max_output_tokens=2048,
-            top_p=1,
-            store=True
-        )
-        # response 구조상 마지막 assistant 응답이 input 리스트의 마지막 원소로 들어옴
-        # role이 assistant인 항목을 찾아서 'text' 필드를 추출
-        if "input" in response and len(response["input"]) > 0:
-            # 가장 마지막 메시지가 assistant일 것이라 가정
-            last_msg = response["input"][-1]
-            if last_msg.get("role") == "assistant":
-                # content 리스트 중 첫 번째에 우리가 원하는 가사가 들어있을 것
-                content = last_msg.get("content", [])
-                if len(content) > 0 and content[0].get("type") == "output_text":
-                    return content[0].get("text", "")
-        return "가사 생성 실패: 응답 형식을 파악할 수 없습니다."
-    except Exception as e:
-        return f"가사 생성 실패: {str(e)}"
+
+    retries = 3
+    for attempt in range(retries):
+        try:
+            response = client.responses.create(
+                model="gpt-4.1-mini",
+                input=[
+                    {
+                        "role": "system",
+                        "content": [{"type": "input_text", "text": system_prompt}]
+                    },
+                    {
+                        "role": "user",
+                        "content": [{"type": "input_text", "text": topic}]
+                    }
+                ],
+                text={"format": {"type": "text"}},
+                reasoning={},
+                tools=[],
+                temperature=1,
+                max_output_tokens=2048,
+                top_p=1,
+                store=True
+            )
+
+            # assistant 메시지에서 가사 텍스트 추출
+            msgs = response.get("input", [])
+            if msgs and msgs[-1].get("role") == "assistant":
+                contents = msgs[-1].get("content", [])
+                if contents and contents[0].get("type") == "output_text":
+                    return contents[0]["text"]
+
+            # 형식이 예상과 다르면 예외 발생시켜 재시도
+            raise ValueError("Unexpected response structure")
+
+        except (requests.exceptions.RequestException, Exception) as e:
+            print(f"Attempt {attempt + 1}/{retries} failed: {e}")
+            time.sleep(2)
+
+    return "가사 생성 실패: 재시도 불가"
 
 TAG_DEFAULT = "funk, pop, soul, rock, melodic, guitar, drums, bass, keyboard, percussion, 105 BPM, energetic, upbeat, groovy, vibrant, dynamic"
 LYRIC_DEFAULT = """[verse]