Update process_flow_generator.py

process_flow_generator.py

@@ -2,15 +2,14 @@ import graphviz
 import json
 from tempfile import NamedTemporaryFile
 import os
+from PIL import Image
 
 def generate_process_flow_diagram(json_input: str, output_format: str) -> str:
     """
     Generates a Process Flow Diagram (Flowchart) from JSON input.
-
     Args:
         json_input (str): A JSON string describing the process flow structure.
                           It must follow the Expected JSON Format Example below.
-
     Expected JSON Format Example:
     {
       "start_node": "Start Inference Request",
@@ -80,7 +79,6 @@ def generate_process_flow_diagram(json_input: str, output_format: str) -> str:
         {"from": "log_error", "to": "end_inference_process", "label": "Error Handled"}
       ]
     }
-
     Returns:
         str: The filepath to the generated PNG image file.
     """
@@ -204,4 +202,110 @@ def generate_process_flow_diagram(json_input: str, output_format: str) -> str:
     except json.JSONDecodeError:
         return "Error: Invalid JSON format"
     except Exception as e:
-        return f"Error: {str(e)}"
+        return f"Error: {str(e)}"
+
+# PPT 통합을 위한 추가 함수
+def generate_process_flow_for_ppt(topic: str, context: str, style: str = "Business Workflow") -> Image.Image:
+    """
+    PPT 생성기에서 사용할 프로세스 플로우 다이어그램 생성
+    
+    Args:
+        topic: 프레젠테이션 주제
+        context: 슬라이드 컨텍스트
+        style: 스타일 타입
+    
+    Returns:
+        PIL Image 객체
+    """
+    # 컨텍스트에 따른 프로세스 플로우 JSON 생성
+    if "프로세스" in context or "process" in context.lower():
+        # 비즈니스 프로세스 플로우
+        flow_json = {
+            "start_node": "시작",
+            "nodes": [
+                {"id": "analyze", "label": "현황 분석", "type": "process"},
+                {"id": "plan", "label": "계획 수립", "type": "process"},
+                {"id": "review", "label": "검토 필요?", "type": "decision"},
+                {"id": "implement", "label": "실행", "type": "process"},
+                {"id": "monitor", "label": "모니터링", "type": "process"},
+                {"id": "complete", "label": "완료", "type": "end"}
+            ],
+            "connections": [
+                {"from": "시작", "to": "analyze", "label": ""},
+                {"from": "analyze", "to": "plan", "label": "분석 완료"},
+                {"from": "plan", "to": "review", "label": ""},
+                {"from": "review", "to": "implement", "label": "승인"},
+                {"from": "review", "to": "plan", "label": "재검토"},
+                {"from": "implement", "to": "monitor", "label": ""},
+                {"from": "monitor", "to": "complete", "label": "목표 달성"}
+            ]
+        }
+    elif "일정" in context or "timeline" in context.lower():
+        # 프로젝트 일정 플로우
+        flow_json = {
+            "start_node": "프로젝트 시작",
+            "nodes": [
+                {"id": "phase1", "label": "1단계: 기획", "type": "process"},
+                {"id": "milestone1", "label": "기획 검토", "type": "decision"},
+                {"id": "phase2", "label": "2단계: 개발", "type": "process"},
+                {"id": "phase3", "label": "3단계: 테스트", "type": "process"},
+                {"id": "launch", "label": "런칭", "type": "io"},
+                {"id": "end", "label": "프로젝트 종료", "type": "end"}
+            ],
+            "connections": [
+                {"from": "프로젝트 시작", "to": "phase1", "label": ""},
+                {"from": "phase1", "to": "milestone1", "label": "4주"},
+                {"from": "milestone1", "to": "phase2", "label": "승인"},
+                {"from": "milestone1", "to": "phase1", "label": "보완"},
+                {"from": "phase2", "to": "phase3", "label": "8주"},
+                {"from": "phase3", "to": "launch", "label": "2주"},
+                {"from": "launch", "to": "end", "label": ""}
+            ]
+        }
+    else:
+        # 기본 워크플로우
+        flow_json = {
+            "start_node": "시작",
+            "nodes": [
+                {"id": "input", "label": "입력 단계", "type": "io"},
+                {"id": "process", "label": "처리 단계", "type": "process"},
+                {"id": "check", "label": "검증", "type": "decision"},
+                {"id": "output", "label": "출력 단계", "type": "io"},
+                {"id": "end", "label": "종료", "type": "end"}
+            ],
+            "connections": [
+                {"from": "시작", "to": "input", "label": ""},
+                {"from": "input", "to": "process", "label": "데이터"},
+                {"from": "process", "to": "check", "label": ""},
+                {"from": "check", "to": "output", "label": "통과"},
+                {"from": "check", "to": "process", "label": "재처리"},
+                {"from": "output", "to": "end", "label": ""}
+            ]
+        }
+    
+    # JSON을 문자열로 변환
+    json_str = json.dumps(flow_json, ensure_ascii=False)
+    
+    # 프로세스 플로우 다이어그램 생성
+    png_path = generate_process_flow_diagram(json_str, 'png')
+    
+    if png_path.startswith("Error:"):
+        # 에러 발생 시 기본 이미지 반환
+        from PIL import Image, ImageDraw, ImageFont
+        img = Image.new('RGB', (800, 600), 'white')
+        draw = ImageDraw.Draw(img)
+        draw.text((400, 300), png_path, fill='red', anchor='mm')
+        return img
+    
+    # PNG 파일을 PIL Image로 변환
+    with Image.open(png_path) as img:
+        # 이미지 복사본 생성 (파일 닫기 위해)
+        img_copy = img.copy()
+    
+    # 임시 파일 삭제
+    try:
+        os.unlink(png_path)
+    except:
+        pass
+    
+    return img_copy