Update app.py

app.py

@@ -1,4 +1,3 @@
-#!/usr/bin/env python
 
 import os
 import re
@@ -350,68 +349,115 @@ def get_emoji_for_content(text: str) -> str:
         return '▶️'
 
 ##############################################################################
-# Diagram Type Detection with Priority Score
+# FLUX Diagram Types and Prompts
 ##############################################################################
-def detect_diagram_type_with_score(title: str, content: str) -> Tuple[Optional[str], float]:
-    """슬라이드 내용을 분석하여 적절한 다이어그램 타입과 필요도 점수 결정"""
+FLUX_DIAGRAM_TYPES = {
+    "Flowchart": {
+        "keywords": ['프로세스', 'process', '절차', 'procedure', '단계', 'step', 'flow', '흐름', '워크플로우', 'workflow'],
+        "priority": 5,  # 높은 우선순위
+        "prompt_template": """A hand-drawn style flowchart, vibrant colors, minimalistic icons.
+{title}
+START [Green Button ~40px]
+├── {step1} [Relevant Icon]
+├── {step2} [Relevant Icon]
+├── {step3} [Relevant Icon]
+└── END [Checkered Flag ~40px]
+Professional business style, clear hierarchy, modern design"""
+    },
+    "Mindmap": {
+        "keywords": ['개념', 'concept', '관계', 'relationship', '마인드맵', 'mindmap', '연관', 'connection'],
+        "priority": 4,
+        "prompt_template": """A handrawn colorful mind map diagram, educational style, vibrant colors, clear hierarchy.
+{title}
+├── {branch1} [Relevant Icon ~60px]
+│ ├── {sub1_1}
+│ └── {sub1_2}
+├── {branch2} [Relevant Icon ~50px]
+│ ├── {sub2_1}
+│ └── {sub2_2}
+└── {branch3} [Relevant Icon ~45px]
+Modern business presentation style"""
+    },
+    "Infographic": {
+        "keywords": ['데이터', 'data', '통계', 'statistics', '분석', 'analysis', '차트', 'chart', '그래프', 'graph'],
+        "priority": 4,
+        "prompt_template": """A sophisticated flat-style infographic for business presentation:
+Title: "{title}"
+Visual Elements:
+- Bar charts comparing key metrics
+- Pie chart showing distribution
+- Trend lines indicating growth
+- Modern minimalist icons
+- Clean grid layout with pastel accents
+Professional corporate design, data visualization focus"""
+    },
+    "Diagram": {
+        "keywords": ['구조', 'structure', '체계', 'system', '관계도', 'diagram', '모델', 'model'],
+        "priority": 3,
+        "prompt_template": """A detailed hand-drawn business diagram:
+Title: "{title}"
+Components:
+- Clear visual hierarchy
+- Connected elements with arrows
+- Modern iconography
+- Color-coded sections
+- Professional annotations
+Clean business presentation style"""
+    },
+    "Mockup": {
+        "keywords": ['디자인', 'design', 'ui', 'ux', '화면', 'screen', '인터페이스', 'interface', '앱', 'app'],
+        "priority": 3,
+        "prompt_template": """A clean hand-drawn style wireframe:
+{title}
+- Main screen layout
+- Key UI elements
+- Navigation structure
+- User flow indicators
+Modern app design mockup style"""
+    },
+    "Product Design": {
+        "keywords": ['제품', 'product', '디자인', 'design', '컨셉', 'concept', '아이디어', 'idea'],
+        "priority": 2,
+        "prompt_template": """A sleek design concept:
+{title}
+- Modern aesthetic design
+- Key features highlighted
+- Professional sketch style
+- Innovation focused
+Clean product presentation"""
+    }
+}
+
+##############################################################################
+# Diagram Type Detection with FLUX Priority
+##############################################################################
+def detect_flux_diagram_type(title: str, content: str) -> Tuple[Optional[str], float]:
+    """FLUX 다이어그램 타입 감지 및 우선순위 점수 계산"""
     combined_text = f"{title} {content}".lower()
     
-    # 각 다이어그램 타입별 키워드와 가중치
-    diagram_scores = {
-        "Process Flow": 0,
-        "WBS Diagram": 0,
-        "Concept Map": 0,
-        "Radial Diagram": 0,
-        "Synoptic Chart": 0
-    }
+    diagram_scores = {}
+    
+    for diagram_type, config in FLUX_DIAGRAM_TYPES.items():
+        score = 0
+        # 키워드 매칭
+        for keyword in config['keywords']:
+            if keyword in combined_text:
+                score += config['priority']
+        
+        # 추가 점수 부여
+        if diagram_type == "Flowchart" and any(char in content for char in ['1.', '2.', '3.', '①', '②', '③']):
+            score += 2
+        elif diagram_type == "Infographic" and any(word in combined_text for word in ['%', '증가', 'increase', '감소', 'decrease']):
+            score += 2
+        elif diagram_type == "Mindmap" and (content.count('\n-') > 3 or content.count('\n•') > 3):
+            score += 2
+            
+        if score > 0:
+            diagram_scores[diagram_type] = score
     
-    # Process Flow keywords with weights
-    if any(word in combined_text for word in ['프로세스', 'process', '절차', 'procedure']):
-        diagram_scores["Process Flow"] += 3
-    if any(word in combined_text for word in ['단계', 'step', 'flow', '흐름']):
-        diagram_scores["Process Flow"] += 2
-    if any(word in combined_text for word in ['워크플로우', 'workflow', '순서', 'sequence']):
-        diagram_scores["Process Flow"] += 2
-    
-    # WBS keywords with weights
-    if any(word in combined_text for word in ['wbs', '작업분해', '프로젝트', 'project']):
-        diagram_scores["WBS Diagram"] += 3
-    if any(word in combined_text for word in ['업무분해', 'breakdown', '구조도', '작업구조']):
-        diagram_scores["WBS Diagram"] += 2
-    
-    # Concept Map keywords with weights
-    if any(word in combined_text for word in ['개념', 'concept', '관계', 'relationship']):
-        diagram_scores["Concept Map"] += 3
-    if any(word in combined_text for word in ['연관', 'connection', '마인드맵', 'mindmap']):
-        diagram_scores["Concept Map"] += 2
-    if any(word in combined_text for word in ['구조', 'structure', '체계', 'system']):
-        diagram_scores["Concept Map"] += 1
-    
-    # Radial Diagram keywords with weights
-    if any(word in combined_text for word in ['중심', 'central', '방사형', 'radial']):
-        diagram_scores["Radial Diagram"] += 3
-    if any(word in combined_text for word in ['핵심', 'core', '주요', 'main']):
-        diagram_scores["Radial Diagram"] += 2
-    
-    # Synoptic Chart keywords with weights
-    if any(word in combined_text for word in ['개요', 'overview', '전체', 'overall']):
-        diagram_scores["Synoptic Chart"] += 3
-    if any(word in combined_text for word in ['요약', 'summary', '시놉틱', 'synoptic']):
-        diagram_scores["Synoptic Chart"] += 2
-    
-    # 추가 점수 부여: 리스트나 구조화된 내용이 많은 경우
-    if content.count('\n-') > 3 or content.count('\n•') > 3:
-        diagram_scores["Concept Map"] += 1
-    if any(char in content for char in ['1.', '2.', '3.', '①', '②', '③']):
-        diagram_scores["Process Flow"] += 1
-    
-    # 가장 높은 점수의 다이어그램 타입 선택
-    max_score = max(diagram_scores.values())
-    if max_score > 0:
-        best_type = max(diagram_scores.items(), key=lambda x: x[1])[0]
-        # 필요도 점수 계산 (0-1 범위)
-        necessity_score = min(max_score / 5.0, 1.0)  # 최대 5점을 1.0으로 정규화
-        return best_type, necessity_score
+    if diagram_scores:
+        best_type = max(diagram_scores.items(), key=lambda x: x[1])
+        return best_type[0], min(best_type[1] / 10.0, 1.0)  # 정규화
     
     return None, 0.0
 
@@ -2392,12 +2438,13 @@ def generate_ppt(
         
         # AI 이미지 및 다이어그램 생성 알림
         visual_features = []
-        if include_ai_image and (AI_IMAGE_ENABLED or FLUX_API_ENABLED):
+        if include_ai_image and AI_IMAGE_ENABLED:
             visual_features.append("AI 3D 표지 이미지")
         if include_diagrams and DIAGRAM_GENERATORS_AVAILABLE:
-            visual_features.append("다이어그램 (최대 2개)")
+            visual_features.append("로컬 다이어그램 (최대 2개)")
         if include_flux_images and FLUX_API_ENABLED:
-            visual_features.append("AI 생성 이미지 (각 API별 최대 3개)")
+            visual_features.append("FLUX 다이어그램 (최소 5개)")
+            visual_features.append("AI 생성 이미지")
         
         if visual_features:
             yield None, f"📝 슬라이드 생성 완료!\n\n🎨 생성 중: {', '.join(visual_features)}... (시간이 소요될 수 있습니다)", llm_response
@@ -2419,12 +2466,14 @@ def generate_ppt(
         success_msg += f"🎨 디자인 테마: {DESIGN_THEMES[design_theme]['name']}\n"
         success_msg += f"📐 레이아웃 스타일: {layout_style}\n"
         
-        if include_ai_image and (AI_IMAGE_ENABLED or FLUX_API_ENABLED):
-            success_msg += f"🖼️ AI 생성 표지 이미지 포함\n"
+        if include_ai_image and AI_IMAGE_ENABLED:
+            success_msg += f"🖼️ AI 생성 3D 표지 이미지 포함\n"
+        if include_diagrams and FLUX_API_ENABLED:
+            success_msg += f"📊 FLUX 다이어그램 포함 (6가지 타입)\n"
         if include_diagrams and DIAGRAM_GENERATORS_AVAILABLE:
-            success_msg += f"📊 AI 생성 다이어그램 포함 (최대 2개)\n"
+            success_msg += f"📈 로컬 다이어그램 포함 (1-2개)\n"
         if include_flux_images and FLUX_API_ENABLED:
-            success_msg += f"🎨 AI 생성 슬라이드 이미지 포함 (API별 최대 3개)\n"
+            success_msg += f"🎨 AI 생성 슬라이드 이미지 포함\n"
             
         success_msg += f"📝 생성된 슬라이드:\n"
         
@@ -2742,7 +2791,7 @@ with gr.Blocks(css=css, title="AI PPT Generator Pro") as demo:
                 include_diagrams = gr.Checkbox(
                     label="📊 AI 다이어그램",
                     value=False,
-                    info="필요도가 높은 2개 슬���이드에 다이어그램 자동 생성"
+                    info="FLUX 다이어그램 5개 + 로컬 다이어그램 1-2개 생성"
                 )
                 
                 include_flux_images = gr.Checkbox(
@@ -2797,17 +2846,26 @@ with gr.Blocks(css=css, title="AI PPT Generator Pro") as demo:
         5. **참고 자료 업로드**: PDF, CSV, TXT 파일 지원
         6. **생성 버튼 클릭**: AI가 자동으로 PPT 생성
         
-        ### 🎨 개선된 기능 - 최적화된 시각 요소
-        - **다이어그램 개수 제한**: 전체 슬라이드 중 필요도가 가장 높은 2개 슬라이드에만 다이어그램 생성
-        - **이미지 개수 제한**: 각 API별로 최대 3개씩, 총 6개의 AI 생성 이미지
-        - **한글 폰트 지원**: NanumGothic-Regular.ttf를 사용하여 다이어그램의 한글 깨짐 문제 해결
-        - **스마트 배치**: 다이어그램과 이미지는 텍스트와 겹치지 않도록 우측에 자동 배치
+        ### 🎨 새로운 기능 - FLUX 다이어그램 6가지 타입
+        - **Flowchart**: 프로세스와 워크플로우를 시각화 (우선순위 높음)
+        - **Mindmap**: 개념과 관계를 구조화하여 표현
+        - **Infographic**: 데이터와 통계를 시각적으로 표현
+        - **Diagram**: 구조와 시스템을 다이어그램으로 표현
+        - **Mockup**: UI/UX 디자인 와이어프레임
+        - **Product Design**: 제품 컨셉 디자인
+        
+        ### 📊 시각 요소 분배 (10페이지 기준)
+        - **표지**: 3D 이미지 1개
+        - **FLUX 다이어그램**: 최소 5개 (6가지 타입 중 자동 선택)
+        - **로컬 다이어그램**: 1-2개 (WBS, 복잡한 프로세스 등)
+        - **3D/포토 이미지**: 나머지 슬라이드에 분배
+        - **모든 슬라이드**에 시각적 요소 포함
         
         ### 💡 고급 팁
-        - **다이어그램 우선순위**: 프로세스, WBS, 개념도 등의 키워드가 있는 슬라이드가 우선 선택됩니다
-        - **이미지 분배**: 3D 스타일과 포토리얼리스틱 이미지가 균형있게 분배됩니다
-        - **표지 이미지**: 주제에 맞는 프리미엄 표지 이미지가 자동 생성됩니다
-        - **한글 지원**: app.py와 같은 경로에 NanumGothic-Regular.ttf 파일을 배치하세요
+        - **한글 지원**: FLUX API를 통해 다이어그램 생성으로 한글 깨짐 문제 해결
+        - **스마트 다이어그램 선택**: 슬라이드 내용을 분석하여 가장 적합한 다이어그램 타입 자동 선택
+        - **프로세스 플로우 우선**: 프로세스 관련 키워드가 있으면 Flowchart를 우선 적용
+        - **균형잡힌 시각화**: 모든 슬라이드에 적절한 시각 요소를 배치하여 일관된 품질 유지
         """
     )
     
@@ -2848,12 +2906,12 @@ with gr.Blocks(css=css, title="AI PPT Generator Pro") as demo:
 
 # Initialize APIs on startup
 if __name__ == "__main__":
-    # 한글 폰트 확인
+    # 한글 폰트 확인 (로컬 다이어그램용)
     if os.path.exists(KOREAN_FONT_PATH):
         logger.info(f"✅ 한글 폰트 파일을 찾았습니다: {KOREAN_FONT_PATH}")
     else:
         logger.warning(f"⚠️ 한글 폰트 파일을 찾을 수 없습니다: {KOREAN_FONT_PATH}")
-        logger.warning("다이어그램에서 한글이 깨질 수 있습니다. NanumGothic-Regular.ttf 파일을 app.py와 같은 경로에 배치하세요.")
+        logger.warning("로컬 다이어그램에서 한글이 깨질 수 있습니다. NanumGothic-Regular.ttf 파일을 app.py와 같은 경로에 배치하세요.")
     
     # Try to initialize APIs in parallel
     with concurrent.futures.ThreadPoolExecutor(max_workers=2) as executor:
@@ -2872,9 +2930,13 @@ if __name__ == "__main__":
     
     # 다이어그램 생성기 모듈 상태 확인
     if DIAGRAM_GENERATORS_AVAILABLE:
-        logger.info("✅ 다이어그램 생성기 모듈이 정상적으로 로드되었습니다")
-        logger.info("지원되는 다이어그램 타입: Process Flow, Concept Map, WBS, Radial, Synoptic Chart")
+        logger.info("✅ 로컬 다이어그램 생성기 모듈이 정상적으로 로드되었습니다")
+        logger.info("로컬 다이어그램 타입: Process Flow, Concept Map, WBS, Radial, Synoptic Chart")
     else:
-        logger.warning("⚠️ 다이어그램 생성기 모듈을 찾을 수 없습니다. 다이어그램 기능이 비활성화됩니다.")
+        logger.warning("⚠️ 로컬 다이어그램 생성기 모듈을 찾을 수 없습니다.")
+    
+    if FLUX_API_ENABLED:
+        logger.info("✅ FLUX 다이어그램 생성 가능")
+        logger.info("FLUX 다이어그램 타입: Flowchart, Mindmap, Infographic, Diagram, Mockup, Product Design")
     
     demo.launch()