Update app.py

app.py

@@ -28,6 +28,32 @@ def get_youtube_script(url):
             api_name="/predict"
         )
         
+        # 결과가 JSON 형식이라면 파싱하여 title과 transcriptionAsText만 추출
+        if isinstance(result, dict) or (isinstance(result, str) and result.strip().startswith('{')):
+            if isinstance(result, str):
+                import json
+                try:
+                    result = json.loads(result)
+                except:
+                    pass
+            
+            # JSON 구조에서 데이터 추출
+            if isinstance(result, dict):
+                if 'data' in result and isinstance(result['data'], list) and len(result['data']) > 0:
+                    data = result['data'][0]
+                    title = data.get('title', "제목 없음")
+                    transcription_text = data.get('transcriptionAsText', "")
+                    
+                    # description은 반환하지 않음
+                    return title, transcription_text
+                elif 'title' in result and 'transcriptionAsText' in result:
+                    title = result.get('title', "제목 없음")
+                    transcription_text = result.get('transcriptionAsText', "")
+                    
+                    # description은 반환하지 않음
+                    return title, transcription_text
+        
+        # JSON이 아닌 경우 간단하게 처리
         if not isinstance(result, str):
             result = str(result)
         
@@ -35,10 +61,10 @@ def get_youtube_script(url):
             raise ValueError("추출된 스크립트가 없습니다.")
         
         title = "제목 없음"
-        description = "설명 없음"
         transcription_text = result
         
-        return title, description, transcription_text
+        # description은 반환하지 않음
+        return title, transcription_text
     except Exception as e:
         raise ValueError(f"스크립트 추출 중 오류 발생: {str(e)}")
 
@@ -181,6 +207,7 @@ def get_blog_prompt():
 10. 참고글에서 '링크를 확인해주세요'와 같은 링크 이동의 문구는 제외
 11. 참고글에 있는 작성자, 화자, 유튜버, 기자(Writer, speaker, YouTuber, reporter)의 이름, 애칭, 닉네임(Name, Nkickname)은 반드시 제외
 12. '업체로 부터 제공 받아서 작성', '쿠팡 파트너스'등의 표현을 반드시 제외하라.
+
 [여행 블로그 글 작성 필수 규칙]
 1. 블로그 작성자가 직접 경험하고 느낀 내용만 작성하세요.
 2. 오감에 기반한 생생한 묘사 및 실용 정보 포함
@@ -410,15 +437,14 @@ def save_pdf(blog_post):
     filename = save_to_pdf(content, title)
     return filename
 
-# 메인 함수
 def process_youtube_url(url, style):
     max_retries = 2
     retry_count = 0
 
     while retry_count <= max_retries:
         try:
-            # YouTube 스크립트 추출
-            title, description, script = get_youtube_script(url)
+            # YouTube 스크립트 추출 (이제 title과 script만 반환)
+            title, script = get_youtube_script(url)
             
             # 스크립트 검증
             if not script or not script.strip():
@@ -429,8 +455,8 @@ def process_youtube_url(url, style):
             # 스크립트 내용만 먼저 반환
             yield script_content, "블로그 글을 생성하는 중입니다..."
             
-            # 블로그 글 생성
-            blog_post = generate_blog_post(script, title, description, style)
+            # 블로그 글 생성 (description은 빈 문자열로 설정)
+            blog_post = generate_blog_post(script, title, "", style)
             
             # 생성된 블로그 글 반환
             yield script_content, blog_post
@@ -445,8 +471,6 @@ def process_youtube_url(url, style):
                 yield f"자막을 찾을 수 없습니다. 영상에 자막이 없거나 자막을 가져오는 데 실패했습니다. 오류: {str(e)}", ""
                 return
 
-    yield "최대 재시도 횟수를 초과했습니다. 자막을 가져오지 못했습니다.", ""
-
 # Gradio 인터페이스
 with gr.Blocks() as demo:
     gr.Markdown("## 유튜브 -> 블로그 - 여행")