Add Hugging Face deployment workflow

.github/workflows/deploy_to_hf.yml

@@ -0,0 +1,22 @@
+name: Deploy to Hugging Face Space
+
+on:
+  push:
+    branches:
+      - main
+
+jobs:
+  deploy:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+
+      - name: Push to Hugging Face Space
+        env:
+          HF_TOKEN: ${{ secrets.HF_TOKEN }}
+        run: |
+          git remote add space https://PLXR:${HF_TOKEN}@huggingface.co/spaces/PLXR/youtube_auto_image1
+          git push --force space main

app.py

@@ -28,6 +28,8 @@ if 'final_audio' not in st.session_state:
     st.session_state['final_audio'] = None
 if 'shared_script' not in st.session_state:
     st.session_state['shared_script'] = ""
+if 'thumbnail_results' not in st.session_state:
+    st.session_state['thumbnail_results'] = []
 
 # [CSS] 다크모드 + 오렌지 포인트 디자인
 st.markdown("""
@@ -310,6 +312,7 @@ with tab2:
                 seo_result = logic_seo.generate_seo_content(client, text_model_id, seo_script)
 
                 st.session_state["seo_result"] = seo_result
+                st.session_state["thumbnail_results"] = []
                 st.success("분석 완료!")
                 
                 c_seo1, c_seo2 = st.columns(2)
@@ -323,105 +326,84 @@ with tab2:
                 st.markdown("##### 📝 설명란 (Description)")
                 st.text_area("설명란 결과", seo_result['description'], height=200)
                 st.divider()
-st.subheader("🖼️ 썸네일 생성")
 
-seo_cached = st.session_state.get("seo_result")
+    st.subheader("🖼️ 썸네일 생성")
 
-if not seo_cached:
-    st.info("SEO 분석을 먼저 실행하면 썸네일 생성 버튼이 나타납니다.")
-else:
-    strat_keys = list(THUMBNAIL_STRATEGIES.keys())
-    sel_strat = st.selectbox("썸네일 전략 선택", strat_keys, index=0)
+    seo_cached = st.session_state.get("seo_result")
 
-    # 썸네일 텍스트(메인 타이틀) 후보: SEO 추천 제목 1개를 기본값으로
-    default_thumb_text = seo_cached["titles"][0] if seo_cached.get("titles") else ""
-    thumb_text = st.text_input("썸네일 메인 문구(원하면 수정)", value=default_thumb_text)
-
-    if st.button("🧠 썸네일 프롬프트 생성", key="thumb_prompt_btn"):
-        if not api_key:
-            st.error("API Key 필요")
-        else:
-            client = genai.Client(api_key=api_key)
-
-            strategy_block = THUMBNAIL_STRATEGIES[sel_strat]
-
-            prompt = f"""
-너는 유튜브 썸네일 기획자다.
-아래 '대본'과 '전략'을 참고해서, 이미지 생성 모델에 넣을 '썸네일 프롬프트'를 만든다.
-
-[대본]
-{seo_script[:8000]}
-
-[전략]
-{strategy_block}
-
-[추가 조건]
-- 썸네일 이미지 안에 텍스트를 직접 그려 넣지 마라(글자 생성 금지).
-- 대신 '텍스트를 넣을 자리'를 구도로 확보해라(상단/하단 여백, 안전영역).
-- 국가/국기/대통령/청와대/국회의사당 등 특정 국가 상징이 자동으로 나오지 않게,
-  정치 상징물은 추상적 은유(실루엣, 조명, 군중, 무대)로 처리해라.
-- 화면비는 {aspect_ratio}.
-- 출력은 "프롬프트 텍스트 1개"만. JSON/마크다운 금지.
-
-[사용자가 넣을 메인 문구(참고만)]
-{thumb_text}
-""".strip()
-
-            res = client.models.generate_content(model=text_model_id, contents=prompt)
-            thumb_prompt = (getattr(res, "text", "") or "").strip()
-            st.session_state["thumb_prompt"] = thumb_prompt
-            st.success("썸네일 프롬프트 생성 완료!")
-            st.text_area("생성된 썸네일 프롬프트", value=thumb_prompt, height=180)
+    if not seo_cached:
+        st.info("SEO 분석을 먼저 실행하면 썸네일 생성 버튼이 나타납니다.")
+    else:
+        default_thumb_text = seo_cached["titles"][0] if seo_cached.get("titles") else ""
+        thumb_text = st.text_input("썸네일 메인 문구(참고용)", value=default_thumb_text)
 
-    # 프롬프트가 있을 때만 이미지 생성 버튼 노출
-    thumb_prompt_cached = st.session_state.get("thumb_prompt", "")
-    if thumb_prompt_cached:
-        if st.button("🎨 썸네일 이미지 1장 생성", key="thumb_img_btn", type="primary"):
+        if st.button("🎨 썸네일 3종 동시 생성", key="thumb_img_btn", type="primary"):
             if not api_key:
                 st.error("API Key 필요")
+            elif not seo_script:
+                st.warning("대본 필요")
             else:
                 client = genai.Client(api_key=api_key)
-
-                # 이미지 생성 (generate_images 우선, 없으면 generate_content fallback)
-                img_bytes = None
-                try:
-                    if hasattr(client.models, "generate_images"):
-                        img_res = client.models.generate_images(
-                            model=image_model_id,
-                            prompt=thumb_prompt_cached
-                        )
-                        # logic_image에 있는 안전 추출 함수가 없다면 간단히 parts에서 뽑기
+                progress_text = "썸네일 3종 생성 중..."
+                my_bar = st.progress(0, text=progress_text)
+
+                strategy_items = list(THUMBNAIL_STRATEGIES.items())
+                temp_results = [None] * len(strategy_items)
+
+                with concurrent.futures.ThreadPoolExecutor(max_workers=3) as executor:
+                    future_to_idx = {
+                        executor.submit(
+                            logic_image.process_thumbnail_task,
+                            i,
+                            key,
+                            strategy_text,
+                            seo_script,
+                            thumb_text,
+                            client,
+                            text_model_id,
+                            image_model_id,
+                            aspect_ratio
+                        ): i
+                        for i, (key, strategy_text) in enumerate(strategy_items)
+                    }
+                    completed = 0
+                    for future in concurrent.futures.as_completed(future_to_idx):
+                        idx, prompt, img = future.result()
+                        strategy_key = strategy_items[idx][0]
+                        temp_results[idx] = (strategy_key, prompt, img)
+                        completed += 1
+                        my_bar.progress(completed / len(strategy_items), text=f"{strategy_key} 완료!")
+
+                st.session_state['thumbnail_results'] = temp_results
+                my_bar.empty()
+                st.rerun()
+
+        if st.session_state.get('thumbnail_results'):
+            cols = st.columns(3)
+            for col, item in zip(cols, st.session_state['thumbnail_results']):
+                if item is None:
+                    continue
+                strategy_key, prompt_text, img_data = item
+                with col:
+                    st.markdown(f"**{strategy_key}**")
+                    if img_data:
                         try:
-                            cand = img_res.candidates[0]
-                            part = cand.content.parts[0]
-                            img_bytes = part.inline_data.data if part.inline_data else None
+                            image = Image.open(io.BytesIO(img_data))
+                            st.image(image, use_container_width=True)
+                            st.download_button(
+                                label="⬇️ 다운로드",
+                                data=img_data,
+                                file_name=f"thumbnail_{strategy_key.split('.')[0].lower()}.png",
+                                mime="image/png",
+                                key=f"thumb_dl_{strategy_key}",
+                                use_container_width=True
+                            )
                         except:
-                            img_bytes = None
+                            st.error("이미지 오류")
                     else:
-                        img_res = client.models.generate_content(
-                            model=image_model_id,
-                            contents=thumb_prompt_cached
-                        )
-                        try:
-                            cand = img_res.candidates[0]
-                            part = cand.content.parts[0]
-                            img_bytes = part.inline_data.data if part.inline_data else None
-                        except:
-                            img_bytes = None
-                except Exception as e:
-                    st.error(f"이미지 생성 실패: {e}")
-
-                if img_bytes:
-                    st.image(Image.open(io.BytesIO(img_bytes)), use_container_width=True)
-                    st.download_button(
-                        "⬇️ 썸네일 다운로드",
-                        data=img_bytes,
-                        file_name="thumbnail.png",
-                        mime="image/png",
-                        use_container_width=True
-                    )
-                else:
-                    st.error("이미지 바이트를 추출하지 못했습니다. (모델 응답 구조 확인 필요)")
+                        st.warning("이미지 없음")
+                    with st.expander("프롬프트 보기"):
+                        st.write(prompt_text)
 
 # ----------------------------------------------------------------
 # [TAB 3] AI 성우 (TTS)
@@ -478,4 +460,4 @@ with tab3:
             if final_wav:
                 st.success("오디오 생성 완료!")
                 st.audio(final_wav, format="audio/wav")
-                st.download_button("다운로드 (WAV)", final_wav, "full_audio.wav", "audio/wav")
+                st.download_button("다운로드 (WAV)", final_wav, "full_audio.wav", "audio/wav")

logic_image.py

@@ -141,7 +141,8 @@ def process_scene_task(
             img_bytes = None
 
     return index, final_prompt, img_bytes
-    def process_thumbnail_task(
+
+def process_thumbnail_task(
     index,
     strategy_key,
     strategy_text,
@@ -151,7 +152,7 @@ def process_scene_task(
     text_model_id,
     image_model_id,
     aspect_ratio,
-    reference_image=None
+    reference_image=None,
 ):
     """
     return (idx, final_prompt, image_bytes)
@@ -177,6 +178,7 @@ def process_scene_task(
 [필수 조건]
 - 출력은 무조건 한국어만. 영어/로마자 금지.
 - 이미지 안에 글자/자막/배너/로고 절대 생성 금지.
+- 전략에 포함된 텍스트 오버레이 지시는 모두 무시하고, 텍스트가 들어가지 않게 구성.
 - 대신 텍스트를 나중에 넣을 수 있게 상단 또는 하단에 넓은 여백(안전영역)을 확보.
 - 특정 국가 상징 자동생성 금지: 태극기, 청와대, 국회의사당, 대통령, 훈장/휘장, 선거 포스터, 국기류 전부 금지.
 - 정치인을 실존 인물처럼 특정하지 말고, 익명 캐릭터/실루엣/상징적 연출로 처리.
@@ -215,4 +217,3 @@ def process_scene_task(
             img_bytes = None
 
     return index, final_prompt, img_bytes
-