Update app.py

app.py

@@ -56,6 +56,7 @@ if "script" not in st.session_state: st.session_state["script"] = ""
 if "image_data" not in st.session_state: st.session_state["image_data"] = None
 if "result_card_data" not in st.session_state: st.session_state["result_card_data"] = None
 if "audio_data" not in st.session_state: st.session_state["audio_data"] = None
+if "caption" not in st.session_state: st.session_state["caption"] = ""  # [NEW] 설명글 저장용
 
 # STEP 1: 주제 생성
 st.header(f"1️⃣ {target_month}월 주제 뽑기")
@@ -91,13 +92,36 @@ if st.session_state["topics"]:
 # STEP 3: 확인 및 다운로드
 if st.session_state["script"]:
     st.markdown("---")
-    st.header("3️⃣ TTS 생성 및 다운로드")
+    st.header("3️⃣ 상세 설명글 & TTS & 다운로드")
+    
     col1, col2 = st.columns([1.3, 1])
     
     with col1:
-        st.subheader("📝 대본 & 오디오")
-        edited_script = st.text_area("대본 수정", value=st.session_state["script"], height=300)
-        
+        st.subheader("📝 대본 & 설명글")
+        # 1. 대본 수정 영역
+        edited_script = st.text_area("쇼츠 대본 수정", value=st.session_state["script"], height=200)
+
+        # 2. 설명글(캡션) 생성 버튼
+        if st.button("📝 인스타/유튜브 설명글(상세풀이+CTA) 생성"):
+             if not st.session_state["selected_topic"]:
+                 st.warning("주제가 선택되지 않았습니다.")
+             else:
+                with st.spinner("전자책 홍보 멘트가 포함된 상세 풀이를 작성 중입니다..."):
+                    caption_text = logic_saju.generate_caption(
+                        st.session_state["selected_topic"], 
+                        edited_script, 
+                        google_api_key
+                    )
+                    st.session_state["caption"] = caption_text
+                    st.success("설명글 생성 완료!")
+
+        # 3. 설명글 표시 영역 (생성된 경우에만)
+        if "caption" in st.session_state and st.session_state["caption"]:
+            st.text_area("상세 설명글 (복사해서 사용)", value=st.session_state["caption"], height=300)
+
+        st.markdown("---")
+
+        # 4. TTS 생성 버튼
         if st.button("🔊 목소리 입히기 (TTS 생성)"):
             with st.spinner("오디오 생성 중..."):
                 audio_bytes, error_msg = logic_tts.generate_audio(
@@ -110,19 +134,28 @@ if st.session_state["script"]:
                 else:
                     st.error(f"오디오 생성 실패: {error_msg}")
 
+        # 5. 다운로드 버튼
         if st.session_state["audio_data"]:
             st.markdown("### 💾 전체 파일 다운로드")
             zip_buffer = io.BytesIO()
             with zipfile.ZipFile(zip_buffer, "w") as zf:
+                # 이미지들
                 if st.session_state["image_data"]:
                     zf.writestr(f"character_{gender_select}.png", st.session_state["image_data"])
                 if st.session_state.get("result_card_data"):
                     zf.writestr("result_card.png", st.session_state["result_card_data"])
+                
+                # 오디오 & 자막 & 대본
                 zf.writestr("audio.mp3", st.session_state["audio_data"])
                 srt_content = logic_tts.create_srt_from_text(edited_script)
                 zf.writestr("subtitles.srt", srt_content)
                 zf.writestr("script.txt", edited_script)
-                # [NEW] Kling AI 프롬프트도 텍스트 파일로 같이 넣어드림
+                
+                # [NEW] 설명글도 저장
+                if "caption" in st.session_state and st.session_state["caption"]:
+                    zf.writestr("description.txt", st.session_state["caption"])
+
+                # Kling AI 프롬프트
                 zf.writestr("kling_prompt.txt", KLING_PROMPT)
 
             st.download_button(
@@ -137,9 +170,7 @@ if st.session_state["script"]:
         st.subheader("🖼️ 생성된 이미지")
         if st.session_state["image_data"]:
             st.image(st.session_state["image_data"], caption="캐릭터", use_column_width=True)
-            
-            # [NEW] 여기에 Kling AI 프롬프트 복사 기능 추가!
-            st.markdown("👇 **Kling AI 비디오화 프롬프트 (복사해서 쓰세요)**")
+            st.markdown("👇 **Kling AI 비디오화 프롬프트**")
             st.code(KLING_PROMPT, language="text")
         
         if st.session_state.get("result_card_data"):