Update app.py

app.py

@@ -12,6 +12,7 @@ import zipfile
 from datetime import datetime
 import time
 import traceback
+import base64
 
 # 환경 변수에서 토큰 가져오기
 REPLICATE_API_TOKEN = os.getenv("RAPI_TOKEN")
@@ -159,6 +160,136 @@ PPT_TEMPLATES = {
     }
 }
 
+def generate_slide_content(topic: str, slide_title: str, slide_context: str) -> Dict[str, str]:
+    """각 슬라이드의 텍스트 내용 생성"""
+    print(f"[슬라이드 내용] {slide_title} 텍스트 생성 중...")
+    
+    url = "https://api.friendli.ai/dedicated/v1/chat/completions"
+    headers = {
+        "Authorization": f"Bearer {FRIENDLI_TOKEN}",
+        "Content-Type": "application/json"
+    }
+    
+    system_prompt = """You are a professional presentation content writer specializing in creating concise, impactful slide content.
+
+Your task is to create:
+1. A compelling subtitle (max 10 words)
+2. Exactly 5 bullet points, each being a complete, concise sentence
+3. Each bullet point should be 10-15 words
+
+Guidelines:
+- Be specific and actionable
+- Use professional business language
+- Include relevant data points or metrics when appropriate
+- Ensure content aligns with the slide's purpose
+- Make each point distinct and valuable
+- Use active voice and strong verbs
+
+Output format:
+Subtitle: [subtitle here]
+• [Point 1]
+• [Point 2]
+• [Point 3]
+• [Point 4]
+• [Point 5]"""
+    
+    user_message = f"""Topic: {topic}
+Slide Title: {slide_title}
+Context: {slide_context}
+
+Create compelling content for this presentation slide."""
+    
+    payload = {
+        "model": "dep89a2fld32mcm",
+        "messages": [
+            {
+                "role": "system",
+                "content": system_prompt
+            },
+            {
+                "role": "user",
+                "content": user_message
+            }
+        ],
+        "max_tokens": 300,
+        "top_p": 0.8,
+        "temperature": 0.7,
+        "stream": False
+    }
+    
+    try:
+        response = requests.post(url, json=payload, headers=headers, timeout=30)
+        if response.status_code == 200:
+            result = response.json()
+            content = result['choices'][0]['message']['content'].strip()
+            
+            # Parse content
+            lines = content.split('\n')
+            subtitle = ""
+            bullet_points = []
+            
+            for line in lines:
+                if line.startswith("Subtitle:"):
+                    subtitle = line.replace("Subtitle:", "").strip()
+                elif line.strip().startswith("•"):
+                    bullet_points.append(line.strip())
+            
+            # 한글로 번역이 필요한 경우
+            if any(ord('가') <= ord(char) <= ord('힣') for char in topic):
+                subtitle = translate_content_to_korean(subtitle)
+                bullet_points = [translate_content_to_korean(point) for point in bullet_points]
+            
+            return {
+                "subtitle": subtitle,
+                "bullet_points": bullet_points[:5]  # 최대 5개
+            }
+        else:
+            return {
+                "subtitle": slide_title,
+                "bullet_points": ["내용을 생성할 수 없습니다."] * 5
+            }
+    except Exception as e:
+        print(f"[슬라이드 내용] 오류: {str(e)}")
+        return {
+            "subtitle": slide_title,
+            "bullet_points": ["내용을 생성할 수 없습니다."] * 5
+        }
+
+def translate_content_to_korean(text: str) -> str:
+    """영어 텍스트를 한글로 번역"""
+    url = "https://api.friendli.ai/dedicated/v1/chat/completions"
+    headers = {
+        "Authorization": f"Bearer {FRIENDLI_TOKEN}",
+        "Content-Type": "application/json"
+    }
+    
+    payload = {
+        "model": "dep89a2fld32mcm",
+        "messages": [
+            {
+                "role": "system",
+                "content": "You are a translator. Translate the given English text to Korean. Maintain professional business tone. Only return the translation without any explanation."
+            },
+            {
+                "role": "user",
+                "content": text
+            }
+        ],
+        "max_tokens": 200,
+        "top_p": 0.8,
+        "stream": False
+    }
+    
+    try:
+        response = requests.post(url, json=payload, headers=headers, timeout=30)
+        if response.status_code == 200:
+            result = response.json()
+            return result['choices'][0]['message']['content'].strip()
+        else:
+            return text
+    except Exception as e:
+        return text
+
 def generate_prompt_with_llm(topic: str, style_example: str = None, slide_context: str = None) -> str:
     """주제와 스타일 예제를 받아서 LLM을 사용해 이미지 프롬프트를 생성"""
     print(f"[LLM] 프롬프트 생성 시작: {slide_context}")
@@ -290,7 +421,7 @@ def generate_image(prompt: str, seed: int = 10, slide_info: str = "") -> Tuple[I
             "seed": seed,
             "prompt": english_prompt,
             "speed_mode": "Extra Juiced 🚀 (even more speed)",
-            "output_quality": 80
+            "output_quality": 100
         }
         
         start_time = time.time()
@@ -325,10 +456,146 @@ def generate_image(prompt: str, seed: int = 10, slide_info: str = "") -> Tuple[I
         print(f"[이미지 생성] 상세 오류:\n{traceback.format_exc()}")
         return None, error_msg
 
-def generate_ppt_images_sequential(topic: str, template_name: str, custom_slides: List[Dict], seed: int, progress=gr.Progress()):
-    """PPT 템플릿에 따라 이미지를 순차적으로 생성하며 진행 상황 표시"""
+def create_slide_preview_html(slide_data: Dict) -> str:
+    """16:9 비율의 슬라이드 프리뷰 HTML 생성"""
+    
+    # 이미지를 base64로 인코딩
+    img_base64 = ""
+    if slide_data.get("image"):
+        buffered = BytesIO()
+        slide_data["image"].save(buffered, format="PNG")
+        img_base64 = base64.b64encode(buffered.getvalue()).decode()
+    
+    # 텍스트 내용 가져오기
+    subtitle = slide_data.get("subtitle", "")
+    bullet_points = slide_data.get("bullet_points", [])
+    
+    # HTML 생성
+    html = f"""
+    <div class="slide-container" style="
+        width: 100%;
+        max-width: 1200px;
+        margin: 20px auto;
+        background: white;
+        border-radius: 8px;
+        box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+        overflow: hidden;
+    ">
+        <div class="slide-header" style="
+            background: #2c3e50;
+            color: white;
+            padding: 15px 30px;
+            font-size: 18px;
+            font-weight: bold;
+        ">
+            슬라이드 {slide_data.get('slide_number', '')}: {slide_data.get('title', '')}
+        </div>
+        
+        <div class="slide-content" style="
+            display: flex;
+            height: 0;
+            padding-bottom: 56.25%; /* 16:9 비율 */
+            position: relative;
+        ">
+            <div class="slide-inner" style="
+                position: absolute;
+                top: 0;
+                left: 0;
+                width: 100%;
+                height: 100%;
+                display: flex;
+            ">
+                <!-- 텍스트 영역 (좌측) -->
+                <div class="text-area" style="
+                    flex: 1;
+                    padding: 40px;
+                    display: flex;
+                    flex-direction: column;
+                    justify-content: center;
+                    background: #f8f9fa;
+                ">
+                    <h2 style="
+                        color: #2c3e50;
+                        font-size: 28px;
+                        margin-bottom: 30px;
+                        font-weight: 600;
+                    ">{subtitle}</h2>
+                    
+                    <ul style="
+                        list-style: none;
+                        padding: 0;
+                        margin: 0;
+                    ">
+    """
+    
+    for point in bullet_points:
+        html += f"""
+                        <li style="
+                            margin-bottom: 15px;
+                            padding-left: 25px;
+                            position: relative;
+                            color: #34495e;
+                            font-size: 16px;
+                            line-height: 1.6;
+                        ">
+                            <span style="
+                                position: absolute;
+                                left: 0;
+                                color: #3498db;
+                            ">▶</span>
+                            {point.replace('•', '').strip()}
+                        </li>
+        """
+    
+    html += f"""
+                    </ul>
+                </div>
+                
+                <!-- 이미지 영역 (우측) -->
+                <div class="image-area" style="
+                    flex: 1;
+                    background: #e9ecef;
+                    display: flex;
+                    align-items: center;
+                    justify-content: center;
+                    padding: 20px;
+                ">
+    """
+    
+    if img_base64:
+        html += f"""
+                    <img src="data:image/png;base64,{img_base64}" style="
+                        max-width: 100%;
+                        max-height: 100%;
+                        object-fit: contain;
+                        border-radius: 4px;
+                        box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);
+                    " alt="Slide Image">
+        """
+    else:
+        html += """
+                    <div style="
+                        color: #6c757d;
+                        text-align: center;
+                    ">
+                        <div style="font-size: 48px;">🖼️</div>
+                        <p>이미지 생성 중...</p>
+                    </div>
+        """
+    
+    html += """
+                </div>
+            </div>
+        </div>
+    </div>
+    """
+    
+    return html
+
+def generate_ppt_with_content(topic: str, template_name: str, custom_slides: List[Dict], seed: int, progress=gr.Progress()):
+    """PPT 이미지와 텍스트 내용을 함께 생성"""
     results = []
-    images_for_gallery = []
+    preview_html = ""
     
     # 템플릿 선택
     if template_name == "사용자 정의" and custom_slides:
@@ -337,7 +604,7 @@ def generate_ppt_images_sequential(topic: str, template_name: str, custom_slides
         slides = PPT_TEMPLATES[template_name]["slides"]
     
     if not slides:
-        yield [], "슬라이드가 정의되지 않았습니다."
+        yield "", "슬라이드가 정의되지 않았습니다."
         return
     
     total_slides = len(slides)
@@ -345,75 +612,66 @@ def generate_ppt_images_sequential(topic: str, template_name: str, custom_slides
     print(f"[PPT 생성] 주제: {topic}")
     print(f"[PPT 생성] 템플릿: {template_name}")
     
+    # CSS 스타일 추가
+    preview_html = """
+    <style>
+        .slides-container {
+            width: 100%;
+            max-width: 1400px;
+            margin: 0 auto;
+        }
+    </style>
+    <div class="slides-container">
+    """
+    
     # 각 슬라이드 순차 처리
     for i, slide in enumerate(slides):
         progress((i + 1) / (total_slides + 1), f"슬라이드 {i+1}/{total_slides} 처리 중...")
         
         slide_info = f"슬라이드 {i+1}: {slide['title']}"
-        status_msg = f"\n### 🔄 {slide_info} 생성 중...\n"
         
-        # 현재까지의 상태 업데이트
-        yield images_for_gallery, status_msg + format_results_status(results)
+        # 텍스트 내용 생성
+        slide_context = f"{slide['title']} - {slide.get('prompt_hint', '')}"
+        content = generate_slide_content(topic, slide['title'], slide_context)
         
+        # 프롬프트 생성 및 이미지 생성
         style_key = slide["style"]
         if style_key in STYLE_TEMPLATES:
             style_info = STYLE_TEMPLATES[style_key]
-            slide_context = f"{slide['title']} - {slide.get('prompt_hint', '')}"
-            
-            # 프롬프트 생성
             prompt = generate_prompt_with_llm(topic, style_info["example"], slide_context)
             
             # 이미지 생성
             slide_seed = seed + i
             img, used_prompt = generate_image(prompt, slide_seed, slide_info)
             
-            # 결과 저장
-            result = {
-                "slide_title": slide["title"],
-                "style": slide["style"],
+            # 슬라이드 데이터 구성
+            slide_data = {
+                "slide_number": i + 1,
+                "title": slide["title"],
+                "subtitle": content["subtitle"],
+                "bullet_points": content["bullet_points"],
                 "image": img,
-                "prompt": prompt,
-                "used_prompt": used_prompt,
-                "success": img is not None
+                "style": style_info["name"]
             }
-            results.append(result)
             
-            # 성공한 이미지는 갤러리에 추가
-            if img is not None:
-                caption = f"{i+1}. {slide['title']} ({style_info['name']})"
-                images_for_gallery.append((img, caption))
-                
-                status_msg = f"\n### ✅ {slide_info} 완료!\n"
-            else:
-                status_msg = f"\n### ❌ {slide_info} 실패: {used_prompt}\n"
+            # 프리뷰 HTML 생성
+            preview_html += create_slide_preview_html(slide_data)
             
-            # 즉시 업데이트
-            yield images_for_gallery, status_msg + format_results_status(results)
+            # 현재까지의 상태 업데이트
+            yield preview_html + "</div>", f"### 🔄 {slide_info} 생성 중..."
+            
+            results.append({
+                "slide_data": slide_data,
+                "success": img is not None
+            })
     
     # 최종 결과
+    preview_html += "</div>"
     progress(1.0, "완료!")
     successful = sum(1 for r in results if r["success"])
-    final_status = f"\n## 🎉 생성 완료!\n총 {total_slides}개 슬라이드 중 {successful}개 성공\n\n"
-    final_status += format_results_status(results)
-    
-    yield images_for_gallery, final_status
-
-def format_results_status(results: List[Dict]) -> str:
-    """결과를 상태 메시지로 포맷"""
-    if not results:
-        return ""
-    
-    status_lines = ["### 📊 생성 결과:\n"]
-    for i, result in enumerate(results):
-        if result["success"]:
-            status_lines.append(f"✅ **슬라이드 {i+1}: {result['slide_title']}**")
-            status_lines.append(f"   - 스타일: {result['style'].split('(')[0].strip()}")
-            status_lines.append(f"   - 프롬프트: {result['prompt'][:60]}...\n")
-        else:
-            status_lines.append(f"❌ **슬라이드 {i+1}: {result['slide_title']}** - 실패")
-            status_lines.append(f"   - 오류: {result['used_prompt']}\n")
+    final_status = f"### 🎉 생성 완료! 총 {total_slides}개 슬라이드 중 {successful}개 성공"
     
-    return "\n".join(status_lines)
+    yield preview_html, final_status
 
 def create_custom_slides_ui():
     """사용자 정의 슬라이드 구성 UI"""
@@ -442,11 +700,13 @@ def create_custom_slides_ui():
     return slides
 
 # Gradio 인터페이스 생성
-with gr.Blocks(title="PPT 이미지 생성기", theme=gr.themes.Soft()) as demo:
+with gr.Blocks(title="PPT 이미지 생성기", theme=gr.themes.Soft(), css="""
+    .preview-container { max-width: 1400px; margin: 0 auto; }
+""") as demo:
     gr.Markdown("""
-    # 🎯 AI 기반 PPT 이미지 생성기
+    # 🎯 AI 기반 PPT 통합 생성기
     
-    ### 전문적인 프레젠테이션을 위한 맞춤형 이미지를 한 번에 생성하세요!
+    ### 텍스트와 이미지가 완벽하게 조화된 프레젠테이션을 자동으로 생성합니다!
     """)
     
     # API 토큰 상태 확인
@@ -484,48 +744,38 @@ with gr.Blocks(title="PPT 이미지 생성기", theme=gr.themes.Soft()) as demo:
                 label="시드 값"
             )
             
-            generate_btn = gr.Button("🚀 PPT 이미지 세트 생성", variant="primary", size="lg")
+            generate_btn = gr.Button("🚀 PPT 전체 생성 (텍스트 + 이미지)", variant="primary", size="lg")
             
             # 사용자 정의 섹션
             with gr.Accordion("📝 사용자 정의 슬라이드 구성", open=False) as custom_accordion:
                 gr.Markdown("템플릿을 사용하지 않고 직접 슬라이드를 구성하세요.")
                 custom_slides_components = create_custom_slides_ui()
-            
-        with gr.Column(scale=2):
-            # 진행 상황 표시
-            status_output = gr.Markdown(
-                value="### 👆 템플릿을 선택하고 생성 버튼을 클릭하세요!\n\n생성 진행 상황이 실시간으로 표시됩니다."
-            )
-            
-            # 결과 갤러리
-            output_gallery = gr.Gallery(
-                label="생성된 PPT 이미지",
-                show_label=True,
-                elem_id="gallery",
-                columns=3,
-                rows=3,
-                object_fit="contain",
-                height="auto"
-            )
     
-    # 스타일 가이드
-    with gr.Accordion("📚 스타일별 활용 가이드", open=False):
-        style_guide = "### PPT 제작을 위한 스타일 가이드\n\n"
-        for style_name, style_info in STYLE_TEMPLATES.items():
-            style_guide += f"**{style_name}**\n"
-            style_guide += f"- 용도: {style_info['use_case']}\n"
-            style_guide += f"- 특징: {style_info['description']}\n\n"
-        gr.Markdown(style_guide)
+    # 상태 표시
+    status_output = gr.Markdown(
+        value="### 👆 템플릿을 선택하고 생성 버튼을 클릭하세요!"
+    )
+    
+    # 프리뷰 영역
+    preview_output = gr.HTML(
+        label="PPT 프리뷰 (16:9)",
+        elem_classes="preview-container"
+    )
     
     # 활용 팁
     gr.Markdown("""
     ---
-    ### 💡 PPT 제작 팁:
+    ### 💡 새로운 기능:
+    
+    1. **자동 텍스트 생성**: 각 슬라이드마다 적절한 제목과 5개의 핵심 포인트 자동 생성
+    2. **16:9 프리뷰**: 실제 PPT와 동일한 비율로 미리보기
+    3. **좌우 레이아웃**: 좌측 텍스트, 우측 이미지로 깔끔한 구성
+    4. **실시간 업데이트**: 각 슬라이드가 생성될 때마다 즉시 확인
     
-    1. **템플릿 활용**: 목적에 맞는 템플릿을 선택하면 최적화된 슬라이드 구성을 제공합니다
-    2. **일관성**: 하나의 주제로 전체 슬라이드를 생성하여 시각적 일관성을 유지합니다
-    3. **실시간 확인**: 각 슬라이드가 생성될 때마다 진행 상황을 확인할 수 있습니다
-    4. **고화질**: 모든 이미지는 프레젠테이션에 적합한 고해상도로 생성됩니다
+    ### 📌 활용 팁:
+    - 생성된 내용을 복사하여 실제 PPT에 바로 사용 가능
+    - 각 슬라이드의 텍스트와 이미지가 주제에 맞게 자동 조율
+    - 한글 주제 입력 시 한글로 텍스트 생성
     """)
     
     # 이벤트 핸들러
@@ -538,9 +788,9 @@ with gr.Blocks(title="PPT 이미지 생성기", theme=gr.themes.Soft()) as demo:
             return info
         return ""
     
-    def generate_ppt_images_handler(topic, template_name, seed, progress=gr.Progress(), *custom_inputs):
+    def generate_ppt_handler(topic, template_name, seed, progress=gr.Progress(), *custom_inputs):
         if not topic.strip():
-            yield [], "❌ 주제를 입력해주세요."
+            yield "", "❌ 주제를 입력해주세요."
             return
         
         # 사용자 정의 슬라이드 처리
@@ -558,9 +808,9 @@ with gr.Blocks(title="PPT 이미지 생성기", theme=gr.themes.Soft()) as demo:
                         "prompt_hint": hint
                     })
         
-        # 순차적으로 이미지 생성하며 진행 상황 표시
-        for gallery, status in generate_ppt_images_sequential(topic, template_name, custom_slides, seed, progress):
-            yield gallery, status
+        # PPT 생성
+        for preview, status in generate_ppt_with_content(topic, template_name, custom_slides, seed, progress):
+            yield preview, status
     
     # 이벤트 연결
     template_select.change(
@@ -579,9 +829,9 @@ with gr.Blocks(title="PPT 이미지 생성기", theme=gr.themes.Soft()) as demo:
         ])
     
     generate_btn.click(
-        fn=generate_ppt_images_handler,
+        fn=generate_ppt_handler,
         inputs=[topic_input, template_select, seed_input] + all_custom_inputs,
-        outputs=[output_gallery, status_output]
+        outputs=[preview_output, status_output]
     )
     
     # 초기 템플릿 정보 표시
@@ -594,7 +844,7 @@ with gr.Blocks(title="PPT 이미지 생성기", theme=gr.themes.Soft()) as demo:
 # 앱 실행
 if __name__ == "__main__":
     print("\n" + "="*50)
-    print("🚀 PPT 이미지 생성기 시작!")
+    print("🚀 PPT 통합 생성기 시작!")
     print("="*50)
     
     # 환경 변수 확인