Update app.py

app.py

@@ -34,6 +34,19 @@ except ImportError:
     PPTX_AVAILABLE = False
     logger.warning("python-pptx 라이브러리가 설치되지 않았습니다. pip install python-pptx")
 
+# 다이어그램 생성기 모듈 import 추가
+try:
+    from concept_map_generator import generate_concept_map
+    from synoptic_chart_generator import generate_synoptic_chart
+    from radial_diagram_generator import generate_radial_diagram
+    from process_flow_generator import generate_process_flow_diagram
+    from wbs_diagram_generator import generate_wbs_diagram
+    DIAGRAM_GENERATORS_AVAILABLE = True
+    logger.info("다이어그램 생성기 모듈 로드 성공")
+except ImportError as e:
+    DIAGRAM_GENERATORS_AVAILABLE = False
+    logger.warning(f"다이어그램 생성기 모듈을 찾을 수 없습니다: {e}")
+
 ##############################################################################
 # API Configuration
 ##############################################################################
@@ -94,26 +107,9 @@ def initialize_flux_api():
         return False
 
 ##############################################################################
-# Diagram Generation API Configuration
+# Diagram Generation API Configuration - REMOVED (using local generators)
 ##############################################################################
-DIAGRAM_API_URL = "http://211.233.58.201:7860"  # ChartGPT API URL
-DIAGRAM_API_ENABLED = False
-diagram_api_client = None
-
-def initialize_diagram_api():
-    """다이어그램 생성 API 초기화"""
-    global DIAGRAM_API_ENABLED, diagram_api_client
-    
-    try:
-        logger.info("Connecting to Diagram Generation API...")
-        diagram_api_client = Client(DIAGRAM_API_URL)
-        DIAGRAM_API_ENABLED = True
-        logger.info("Diagram API connected successfully")
-        return True
-    except Exception as e:
-        logger.error(f"Failed to connect to Diagram API: {e}")
-        DIAGRAM_API_ENABLED = False
-        return False
+# 기존 Diagram API 관련 코드 제거하고 로컬 생성기 사용
 
 ##############################################################################
 # Design Themes and Color Schemes
@@ -354,32 +350,40 @@ def get_emoji_for_content(text: str) -> str:
         return '▶️'
 
 ##############################################################################
-# Diagram Type Detection
+# Diagram Type Detection - UPDATED for 6 types
 ##############################################################################
 def detect_diagram_type(title: str, content: str) -> Optional[str]:
     """슬라이드 내용을 분석하여 적절한 다이어그램 타입 결정"""
     combined_text = f"{title} {content}".lower()
     
     # Process Flow keywords
-    if any(word in combined_text for word in ['프로세스', 'process', '절차', 'procedure', '단계', 'step', 'flow', '흐름', '워크플로우', 'workflow']):
+    if any(word in combined_text for word in ['프로세스', 'process', '절차', 'procedure', '단계', 'step', 'flow', '흐름', '워크플로우', 'workflow', '순서', 'sequence']):
         return "Process Flow"
     
     # WBS keywords
-    elif any(word in combined_text for word in ['wbs', '작업분해', '프로젝트', 'project', '업무분해', 'breakdown', '구조도']):
+    elif any(word in combined_text for word in ['wbs', '작업분해', '프로젝트', 'project', '업무분해', 'breakdown', '구조도', '작업구조', 'work structure']):
         return "WBS Diagram"
     
     # Concept Map keywords
-    elif any(word in combined_text for word in ['개념', 'concept', '관계', 'relationship', '연관', 'connection', '마인드맵', 'mindmap']):
+    elif any(word in combined_text for word in ['개념', 'concept', '관계', 'relationship', '연관', 'connection', '마인드맵', 'mindmap', '구조', 'structure', '체계', 'system']):
         return "Concept Map"
     
     # Radial Diagram keywords
-    elif any(word in combined_text for word in ['중심', 'central', '방사형', 'radial', '핵심', 'core', '주요', 'main']):
+    elif any(word in combined_text for word in ['중심', 'central', '방사형', 'radial', '핵심', 'core', '주요', 'main', '중앙', 'center']):
         return "Radial Diagram"
     
     # Synoptic Chart keywords
-    elif any(word in combined_text for word in ['개요', 'overview', '전체', 'overall', '요약', 'summary', '시놉틱', 'synoptic']):
+    elif any(word in combined_text for word in ['개요', 'overview', '전체', 'overall', '요약', 'summary', '시놉틱', 'synoptic', '총괄', 'comprehensive']):
         return "Synoptic Chart"
     
+    # Default: 키워드가 없으면 내용의 특성에 따라 결정
+    # 리스트나 항목이 많으면 Concept Map
+    if content.count('\n-') > 3 or content.count('\n•') > 3:
+        return "Concept Map"
+    # 숫자가 있으면 Process Flow
+    elif any(char in content for char in ['1.', '2.', '3.', '①', '②', '③']):
+        return "Process Flow"
+    
     return None
 
 ##############################################################################
@@ -390,7 +394,7 @@ def generate_diagram_json(title: str, content: str, diagram_type: str) -> Option
     if not FRIENDLI_TOKEN:
         return None
     
-    # 다이어그램 타입별 JSON 구조 가이드
+    # 다이어그램 타입별 JSON 구조 가이드 - paste-2.txt의 구조와 동일하게
     json_guides = {
         "Concept Map": """Generate a JSON for a concept map with the EXACT following structure:
 {
@@ -476,7 +480,9 @@ Important rules:
 2. The JSON must follow the EXACT structure shown above
 3. Create content based on the provided title and content
 4. Use the user's language (Korean or English) for the content values
-5. Keep it simple with 3-5 main nodes/steps"""
+5. Keep it simple with 3-5 main nodes/steps
+6. For Process Flow: 'type' can be "process", "decision", "start", "end"
+7. Ensure all connections reference existing node IDs"""
 
     messages = [
         {"role": "system", "content": system_prompt},
@@ -520,37 +526,40 @@ Important rules:
     return None
 
 ##############################################################################
-# Generate Diagram using API
+# Generate Diagram using Local Generators - NEW
 ##############################################################################
-def generate_diagram_via_api(json_data: str, diagram_type: str) -> Optional[str]:
-    """다이어그램 API를 통해 다이어그램 생성"""
-    if not DIAGRAM_API_ENABLED or not diagram_api_client:
+def generate_diagram_locally(json_data: str, diagram_type: str, output_format: str = "png") -> Optional[str]:
+    """로컬 생성기를 사용하여 다이어그램 생성"""
+    if not DIAGRAM_GENERATORS_AVAILABLE:
+        logger.error("다이어그램 생성기 모듈을 사용할 수 없습니다")
         return None
     
     try:
-        # API 호출
-        result = diagram_api_client.predict(
-            prompt_input=f"Generate {diagram_type}",  # 프롬프트
-            diagram_type_select=diagram_type,  # 다이어그램 타입
-            design_type_select="None",  # 디자인 타입은 None
-            output_format_radio="png",  # PNG 형식
-            use_search_checkbox=False,  # 검색 사용 안함
-            api_name="/generate_with_llm"
-        )
+        # 적절한 생성기를 사용하여 다이어그램 생성
+        if diagram_type == "Concept Map":
+            result = generate_concept_map(json_data, output_format)
+        elif diagram_type == "Synoptic Chart":
+            result = generate_synoptic_chart(json_data, output_format)
+        elif diagram_type == "Radial Diagram":
+            result = generate_radial_diagram(json_data, output_format)
+        elif diagram_type == "Process Flow":
+            result = generate_process_flow_diagram(json_data, output_format)
+        elif diagram_type == "WBS Diagram":
+            result = generate_wbs_diagram(json_data, output_format)
+        else:
+            logger.error(f"Unknown diagram type: {diagram_type}")
+            return None
         
-        # 결과에서 이미지 경로 추출
-        if isinstance(result, tuple) and len(result) > 0:
-            image_path = result[0]
-            if image_path and os.path.exists(image_path):
-                # 임시 파일로 복사
-                with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp:
-                    shutil.copy2(image_path, tmp.name)
-                    return tmp.name
+        # 결과가 문자열이고 에러 메시지인 경우
+        if isinstance(result, str) and result.startswith("Error:"):
+            logger.error(f"Diagram generation error: {result}")
+            return None
         
-        return None
+        # 성공적으로 생성된 경우 파일 경로 반환
+        return result
         
     except Exception as e:
-        logger.error(f"Failed to generate diagram via API: {e}")
+        logger.error(f"Failed to generate diagram locally: {e}")
         return None
 
 ##############################################################################
@@ -1042,43 +1051,6 @@ def generate_ai_image_via_3d_api(prompt: str) -> Optional[str]:
         logger.error(f"Failed to generate 3D image: {e}")
         return None
 
-def generate_flux_image_via_api(prompt: str) -> Optional[str]:
-    """FLUX API를 통해 포토리얼리스틱 이미지 생성"""
-    if not FLUX_API_ENABLED or not flux_api_client:
-        return None
-    
-    try:
-        logger.info(f"Generating FLUX photorealistic image with prompt: {prompt[:100]}...")
-        
-        result = flux_api_client.predict(
-            prompt=prompt,
-            width=768,
-            height=768,
-            guidance=3.5,
-            inference_steps=8,
-            seed=random.randint(1, 1000000),
-            do_img2img=False,
-            init_image=None,
-            image2image_strength=0.8,
-            resize_img=True,
-            api_name="/generate_image"
-        )
-        
-        if isinstance(result, tuple) and len(result) > 0:
-            image_path = result[0]
-            if image_path and os.path.exists(image_path):
-                with Image.open(image_path) as img:
-                    png_tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
-                    img.save(png_tmp.name, format="PNG")
-                    logger.info(f"FLUX image generated and saved to {png_tmp.name}")
-                    return png_tmp.name
-        
-        return None
-        
-    except Exception as e:
-        logger.error(f"Failed to generate FLUX image: {e}")
-        return None
-
 def generate_images_parallel(prompt_3d: str, prompt_photo: str) -> Tuple[Optional[str], Optional[str]]:
     """두 API를 병렬로 호출하여 이미지 생성"""
     import concurrent.futures
@@ -1738,13 +1710,13 @@ def create_advanced_ppt_from_content(
             # 우측에 시각적 요소 추가
             visual_added = False
             
-            if visual_type[0] == 'diagram':
+            if visual_type[0] == 'diagram' and DIAGRAM_GENERATORS_AVAILABLE:
                 # 다이어그램 생성
                 logger.info(f"Generating {visual_type[1]} for slide {i+1}")
                 diagram_json = generate_diagram_json(slide_title, slide_content, visual_type[1])
                 
                 if diagram_json:
-                    diagram_path = generate_diagram_via_api(diagram_json, visual_type[1])
+                    diagram_path = generate_diagram_locally(diagram_json, visual_type[1], "png")
                     if diagram_path and os.path.exists(diagram_path):
                         try:
                             # 다이어그램 이미지 추가
@@ -1755,8 +1727,22 @@ def create_advanced_ppt_from_content(
                             )
                             visual_added = True
                             
+                            # 다이어그램 캡션 추가
+                            caption_box = slide.shapes.add_textbox(
+                                Inches(5.2), Inches(4.6), Inches(4.3), Inches(0.3)
+                            )
+                            caption_tf = caption_box.text_frame
+                            caption_tf.text = f"{visual_type[1]} Diagram"
+                            caption_p = caption_tf.paragraphs[0]
+                            caption_p.font.size = Pt(10)
+                            caption_p.font.color.rgb = theme['colors']['secondary']
+                            caption_p.alignment = PP_ALIGN.CENTER
+                            
                             # 임시 파일 삭제
-                            os.unlink(diagram_path)
+                            try:
+                                os.unlink(diagram_path)
+                            except:
+                                pass
                         except Exception as e:
                             logger.error(f"Failed to add diagram: {e}")
             
@@ -2264,14 +2250,10 @@ def generate_ppt(
                 include_flux_images = False
                 yield None, "⚠️ FLUX API 연결 실패. 포토리얼리스틱 이미지 없이 진행합니다.", ""
         
-        # 다이어그램 API 초기화
-        if include_diagrams and not DIAGRAM_API_ENABLED:
-            yield None, "🔄 다이어그램 생성 API에 연결하는 중...", ""
-            if initialize_diagram_api():
-                yield None, "✅ 다이어그램 API 연결 성공!", ""
-            else:
-                include_diagrams = False
-                yield None, "⚠️ 다이어그램 API 연결 실패. 다이어그램 없이 진행합니다.", ""
+        # 다이어그램 생성기 확인
+        if include_diagrams and not DIAGRAM_GENERATORS_AVAILABLE:
+            yield None, "⚠️ 다이어그램 생성기 모듈을 찾을 수 없습니다. 다이어그램 없이 진행합니다.", ""
+            include_diagrams = False
         
         # Process reference files if provided
         additional_context = ""
@@ -2347,7 +2329,7 @@ def generate_ppt(
         visual_features = []
         if include_ai_image and FLUX_API_ENABLED:
             visual_features.append("AI 3D 표지 이미지")
-        if include_diagrams and DIAGRAM_API_ENABLED:
+        if include_diagrams and DIAGRAM_GENERATORS_AVAILABLE:
             visual_features.append("다이어그램")
         if include_flux_images and FLUX_API_ENABLED:
             visual_features.append("AI 생성 이미지")
@@ -2374,7 +2356,7 @@ def generate_ppt(
         
         if include_ai_image and FLUX_API_ENABLED:
             success_msg += f"🖼️ AI 생성 표지 이미지 포함\n"
-        if include_diagrams and DIAGRAM_API_ENABLED:
+        if include_diagrams and DIAGRAM_GENERATORS_AVAILABLE:
             success_msg += f"📊 AI 생성 다이어그램 포함\n"
         if include_flux_images and FLUX_API_ENABLED:
             success_msg += f"🎨 AI 생성 슬라이드 이미지 포함\n"
@@ -2574,6 +2556,23 @@ input[type="checkbox"] {
     max-height: 500px;
     overflow-y: auto;
 }
+
+/* 반응형 디자인 */
+@media (max-width: 768px) {
+    .main-container {
+        padding: 15px;
+        margin: 10px;
+    }
+    
+    .header-section h1 {
+        font-size: 2em;
+    }
+    
+    .gr-tab-item {
+        padding: 10px 15px;
+        font-size: 1em;
+    }
+}
 """
 
 with gr.Blocks(css=css, title="AI PPT Generator Pro") as demo:
@@ -2582,7 +2581,7 @@ with gr.Blocks(css=css, title="AI PPT Generator Pro") as demo:
         # 🎯 AI 기반 PPT 자동 생성 시스템 Pro
         
         고급 디자인 테마와 레이아웃을 활용한 전문적인 프레젠테이션을 자동으로 생성합니다.
-        FLUX AI로 생성한 고품질 이미지와 다이어그램을 포함하여 시각적으로 풍부한 PPT를 만듭니다.
+        6가지 다이어그램 타입과 AI 생성 이미지를 포함하여 시각적으로 풍부한 PPT를 만듭니다.
         """
     )
     
@@ -2678,7 +2677,7 @@ with gr.Blocks(css=css, title="AI PPT Generator Pro") as demo:
                 include_diagrams = gr.Checkbox(
                     label="📊 AI 다이어그램",
                     value=False,
-                    info="내용에 맞는 다이어그램 자동 생성"
+                    info="6가지 타입의 다이어그램 자동 생성"
                 )
                 
                 include_flux_images = gr.Checkbox(
@@ -2733,18 +2732,19 @@ with gr.Blocks(css=css, title="AI PPT Generator Pro") as demo:
         5. **참고 자료 업로드**: PDF, CSV, TXT 파일 지원
         6. **생성 버튼 클릭**: AI가 자동으로 PPT 생성
         
-        ### 🎨 새로운 기능
-        - **병렬 AI 이미지 생성**: 3D 스타일과 포토리얼리스틱 API를 동시에 사용하여 생성 시간 단축
-        - **다양한 이미지 스타일**: 각 슬라이드마다 다른 스타일과 메타포로 이미지 생성
-        - **AI 표지 이미지**: 우측 하단 배치로 텍스트와 겹치지 않음
-        - **결론 슬라이드 강조**: 결론/요약 슬라이드에 특별한 이미지 추가
-        - **좌-우 레이아웃**: 텍스트는 좌측, 시각적 요소는 우측 배치
+        ### 🎨 새로운 기능 - 6가지 다이어그램 타입
+        - **Process Flow**: 프로세스와 워크플로우를 시각화
+        - **Concept Map**: 개념과 관계를 구조화하여 표현
+        - **WBS Diagram**: 프로젝트 작업 분해 구조도
+        - **Radial Diagram**: 중심 개념과 관련 요소들의 방사형 표현
+        - **Synoptic Chart**: 전체 개요와 요약을 한눈에 보여주는 차트
+        - **자동 감지**: 슬라이드 내용을 분석하여 적절한 다이어그램 타입 자동 선택
         
         ### 💡 고급 팁
-        - 표지 이미지는 우측 하단에 배치되어 제목/부제와 겹치지 않습니다
-        - 3D 스타일과 포토리얼리스틱 이미지가 병렬로 생성되어 시간이 절약됩니다
-        - 각 슬라이드는 내용에 맞는 고유한 시각 메타포와 스타일로 이미지가 생성됩니다
-        - 결론/요약 슬라이드는 자동으로 감지되어 특별한 이미지가 추가됩니다
+        - **다이어그램 자동 생성**: 슬라이드 내용에 따라 6가지 다이어그램 중 가장 적합한 타입이 자동으로 선택됩니다
+        - **병렬 이미지 생성**: 3D 스타일과 포토리얼리스틱 이미지가 동시에 생성되어 시간이 절약됩니다
+        - **스마트 배치**: 다이어그램과 이미지는 텍스트와 겹치지 않도록 우측에 자동 배치됩니다
+        - **키워드 기반 감지**: 프로세스, 개념, 프로젝트, 중심, 개요 등의 키워드를 감지하여 다이어그램 타입을 결정합니다
         """
     )
     
@@ -2755,6 +2755,8 @@ with gr.Blocks(css=css, title="AI PPT Generator Pro") as demo:
             ["2024년 디지털 마케팅 트렌드", 12, True, True, [], "modern", "modern", "consistent", False, True, True, True],
             ["기후변화와 지속가능한 발전", 15, True, True, [], "nature", "classic", "consistent", False, True, True, True],
             ["스타트업 사업계획서", 8, False, True, [], "creative", "modern", "varied", False, True, True, True],
+            ["프로젝트 관리 방법론", 10, False, True, [], "professional", "modern", "consistent", False, False, True, False],
+            ["머신러닝 프로세스 가이드", 12, False, True, [], "modern", "modern", "consistent", False, False, True, False],
         ],
         inputs=[topic_input, num_slides, use_web_search, use_korean, reference_files, 
                 design_theme, font_style, layout_style, include_charts, include_ai_image,
@@ -2784,14 +2786,12 @@ with gr.Blocks(css=css, title="AI PPT Generator Pro") as demo:
 # Initialize APIs on startup
 if __name__ == "__main__":
     # Try to initialize APIs in parallel
-    with concurrent.futures.ThreadPoolExecutor(max_workers=3) as executor:
+    with concurrent.futures.ThreadPoolExecutor(max_workers=2) as executor:
         futures = []
         if AI_IMAGE_API_URL:
             futures.append(executor.submit(initialize_ai_image_api))
         if FLUX_API_URL:
             futures.append(executor.submit(initialize_flux_api))
-        if DIAGRAM_API_URL:
-            futures.append(executor.submit(initialize_diagram_api))
         
         # Wait for all to complete
         for future in concurrent.futures.as_completed(futures):
@@ -2800,4 +2800,11 @@ if __name__ == "__main__":
             except Exception as e:
                 logger.error(f"API initialization failed: {e}")
     
-    demo.launch()
+    # 다이어그램 생성기 모듈 상태 확인
+    if DIAGRAM_GENERATORS_AVAILABLE:
+        logger.info("✅ 다이어그램 생성기 모듈이 정상적으로 로드되었습니다")
+        logger.info("지원되는 다이어그램 타입: Process Flow, Concept Map, WBS, Radial, Synoptic Chart")
+    else:
+        logger.warning("⚠️ 다이어그램 생성기 모듈을 찾을 수 없습니다. 다이어그램 기능이 비활성화됩니다.")
+    
+    demo.launch()