app.py

import gradio as gr
import os
import sys
import tempfile
from io import StringIO
import traceback
import inspect

# 각 모듈의 함수들을 저장할 딕셔너리
module_functions = {}

# 환경변수에서 각 모듈 코드를 불러와서 실행
def load_and_execute_module(module_name, module_key):
    """환경변수에서 모듈 코드를 불러와서 실행하고 함수들을 저장"""
    try:
        module_code = os.getenv(module_name)
        if not module_code:
            raise ValueError(f"환경변수 {module_name}이 설정되지 않았습니다.")
        
        # 모듈 실행 전 현재 globals 상태 저장
        before_globals = set(globals().keys())
        
        # 모듈 코드 실행
        exec(module_code, globals())
        
        # 모듈 실행 후 새로 추가된 함수들 찾기
        after_globals = set(globals().keys())
        new_functions = after_globals - before_globals
        
        # 새로 추가된 함수들을 모듈별로 저장
        module_functions[module_key] = {}
        for func_name in new_functions:
            if callable(globals()[func_name]) and not func_name.startswith('_'):
                module_functions[module_key][func_name] = globals()[func_name]
        
        print(f"{module_name} 모듈이 성공적으로 로드되었습니다.")
        print(f"로드된 함수들: {list(module_functions[module_key].keys())}")
        
    except Exception as e:
        print(f"{module_name} 모듈 로드 중 오류: {str(e)}")
        traceback.print_exc()
        module_functions[module_key] = {}

# 각 모듈 로드
print("모듈들을 로드하는 중...")

# 정보성 블로그 모듈 로드
load_and_execute_module("INFO_BLOG_CODE", "info_blog")

# 여행 블로그 모듈 로드
load_and_execute_module("TRAVEL_BLOG_CODE", "travel_blog")

# 상품후기 블로그 모듈 로드
load_and_execute_module("PRODUCT_REVIEW_BLOG_CODE", "product_review_blog")

# 핵심기능집중형 블로그 모듈 로드
load_and_execute_module("FEATURE_FOCUS_BLOG_CODE", "feature_focus_blog")

# 방문후기형 블로그 모듈 로드
load_and_execute_module("VISIT_REVIEW_BLOG_CODE", "visit_review_blog")

print("모든 모듈 로드 완료!")

# 각 모듈의 함수 확인
for module_key, functions in module_functions.items():
    print(f"{module_key}: {list(functions.keys())}")

# 래퍼 함수들 정의

# 정보성 블로그 래퍼 함수들
def generate_outline_1_wrapper(category, style, ref1, ref2, ref3):
    """정보성 블로그 아웃라인 생성 래퍼"""
    try:
        # 정보성 블로그 모듈에서 함수 찾기
        if 'generate_outline_1' in module_functions.get('info_blog', {}):
            return module_functions['info_blog']['generate_outline_1'](category, style, ref1, ref2, ref3)
        elif 'generate_outline' in module_functions.get('info_blog', {}):
            return module_functions['info_blog']['generate_outline'](category, style, ref1, ref2, ref3)
        else:
            return "정보성 블로그 아웃라인 생성 함수를 찾을 수 없습니다."
    except Exception as e:
        return f"정보성 블로그 아웃라인 생성 중 오류: {str(e)}"

def generate_blog_post_1_wrapper(category, style, ref1, ref2, ref3, outline):
    """정보성 블로그 글 생성 래퍼"""
    try:
        if 'generate_blog_post_1' in module_functions.get('info_blog', {}):
            return module_functions['info_blog']['generate_blog_post_1'](category, style, ref1, ref2, ref3, outline)
        elif 'generate_blog_post' in module_functions.get('info_blog', {}):
            return module_functions['info_blog']['generate_blog_post'](category, style, outline, ref1, ref2, ref3)
        else:
            return "정보성 블로그 글 생성 함수를 찾을 수 없습니다."
    except Exception as e:
        return f"정보성 블로그 글 생성 중 오류: {str(e)}"

# 여행 블로그 래퍼 함수들
def generate_outline_2_wrapper(category, style, ref1, ref2, ref3):
    """여행 블로그 아웃라인 생성 래퍼"""
    try:
        if 'generate_outline_2' in module_functions.get('travel_blog', {}):
            return module_functions['travel_blog']['generate_outline_2'](category, style, ref1, ref2, ref3)
        elif 'generate_outline' in module_functions.get('travel_blog', {}):
            return module_functions['travel_blog']['generate_outline'](category, style, ref1, ref2, ref3)
        else:
            return "여행 블로그 아웃라인 생성 함수를 찾을 수 없습니다."
    except Exception as e:
        return f"여행 블로그 아웃라인 생성 중 오류: {str(e)}"

def generate_blog_post_2_wrapper(category, style, ref1, ref2, ref3, outline):
    """여행 블로그 글 생성 래퍼"""
    try:
        if 'generate_blog_post_2' in module_functions.get('travel_blog', {}):
            return module_functions['travel_blog']['generate_blog_post_2'](category, style, ref1, ref2, ref3, outline)
        elif 'generate_blog_post' in module_functions.get('travel_blog', {}):
            return module_functions['travel_blog']['generate_blog_post'](category, style, outline, ref1, ref2, ref3)
        else:
            return "여행 블로그 글 생성 함수를 찾을 수 없습니다."
    except Exception as e:
        return f"여행 블로그 글 생성 중 오류: {str(e)}"

# 상품후기 블로그 래퍼 함수들
def generate_outline_3_wrapper(category, style, ref1, ref2, ref3):
    """상품후기 블로그 아웃라인 생성 래퍼"""
    try:
        if 'generate_outline_3' in module_functions.get('product_review_blog', {}):
            return module_functions['product_review_blog']['generate_outline_3'](category, style, ref1, ref2, ref3)
        elif 'generate_outline' in module_functions.get('product_review_blog', {}):
            return module_functions['product_review_blog']['generate_outline'](category, style, ref1, ref2, ref3)
        else:
            return "상품후기 블로그 아웃라인 생성 함수를 찾을 수 없습니다."
    except Exception as e:
        return f"상품후기 블로그 아웃라인 생성 중 오류: {str(e)}"

def generate_blog_post_3_wrapper(category, style, ref1, ref2, ref3, outline):
    """상품후기 블로그 글 생성 래퍼"""
    try:
        if 'generate_blog_post_3' in module_functions.get('product_review_blog', {}):
            return module_functions['product_review_blog']['generate_blog_post_3'](category, style, ref1, ref2, ref3, outline)
        elif 'generate_blog_post' in module_functions.get('product_review_blog', {}):
            return module_functions['product_review_blog']['generate_blog_post'](category, style, outline, ref1, ref2, ref3)
        else:
            return "상품후기 블로그 글 생성 함수를 찾을 수 없습니다."
    except Exception as e:
        return f"상품후기 블로그 글 생성 중 오류: {str(e)}"

# 핵심기능집중형 블로그 래퍼 함수들
def generate_outline_4_wrapper(category, style, ref1, ref2, ref3):
    """핵심기능집중형 블로그 아웃라인 생성 래퍼"""
    try:
        if 'generate_outline_4' in module_functions.get('feature_focus_blog', {}):
            result = module_functions['feature_focus_blog']['generate_outline_4'](category, style, ref1, ref2, ref3)
            if isinstance(result, tuple) and len(result) == 3:
                return result
            else:
                return (str(result), "", "")
        elif 'generate_outline' in module_functions.get('feature_focus_blog', {}):
            result = module_functions['feature_focus_blog']['generate_outline'](category, style, ref1, ref2, ref3)
            return (str(result), "", "")
        else:
            return ("핵심기능집중형 블로그 아웃라인 생성 함수를 찾을 수 없습니다.", "", "")
    except Exception as e:
        return (f"핵심기능집중형 블로그 아웃라인 생성 중 오류: {str(e)}", "", "")

def generate_blog_post_4_wrapper(category, style, ref1, ref2, ref3, outline):
    """핵심기능집중형 블로그 글 생성 래퍼"""
    try:
        if 'generate_blog_post_4' in module_functions.get('feature_focus_blog', {}):
            return module_functions['feature_focus_blog']['generate_blog_post_4'](category, style, ref1, ref2, ref3, outline)
        elif 'generate_blog_post' in module_functions.get('feature_focus_blog', {}):
            return module_functions['feature_focus_blog']['generate_blog_post'](category, style, outline, ref1, ref2, ref3)
        else:
            return "핵심기능집중형 블로그 글 생성 함수를 찾을 수 없습니다."
    except Exception as e:
        return f"핵심기능집중형 블로그 글 생성 중 오류: {str(e)}"

# 방문후기형 블로그 래퍼 함수들
def generate_outline_5_wrapper(category, style, ref1, ref2, ref3, photo_recommendations):
    """방문후기형 블로그 아웃라인 생성 래퍼"""
    try:
        if 'generate_outline_5' in module_functions.get('visit_review_blog', {}):
            return module_functions['visit_review_blog']['generate_outline_5'](category, style, ref1, ref2, ref3, photo_recommendations)
        elif 'generate_outline' in module_functions.get('visit_review_blog', {}):
            return module_functions['visit_review_blog']['generate_outline'](category, style, ref1, ref2, ref3, photo_recommendations)
        else:
            return "방문후기형 블로그 아웃라인 생성 함수를 찾을 수 없습니다."
    except Exception as e:
        return f"방문후기형 블로그 아웃라인 생성 중 오류: {str(e)}"

def generate_blog_post_5_wrapper(category, style, ref1, ref2, ref3, outline, photo_recommendations):
    """방문후기형 블로그 글 생성 래퍼"""
    try:
        if 'generate_blog_post_5' in module_functions.get('visit_review_blog', {}):
            return module_functions['visit_review_blog']['generate_blog_post_5'](category, style, ref1, ref2, ref3, outline, photo_recommendations)
        elif 'generate_blog_post' in module_functions.get('visit_review_blog', {}):
            return module_functions['visit_review_blog']['generate_blog_post'](category, style, ref1, ref2, ref3, outline, photo_recommendations)
        else:
            return "방문후기형 블로그 글 생성 함수를 찾을 수 없습니다."
    except Exception as e:
        return f"방문후기형 블로그 글 생성 중 오류: {str(e)}"

print("래퍼 함수들 정의 완료!")

# Gradio 인터페이스 설계 (Gradio 5.0 이상)
def interface():
    with gr.Blocks() as demo:
        gr.Markdown("# 블로그 포스팅")

        # 첫 번째 탭: 정보성
        with gr.Tab("정보성"):
            category_1 = gr.Radio(choices=['일반', '생활건강'], label="포스팅 카테고리", value="일반")
            style_1 = gr.Radio(choices=['친근한', '일반', '전문적인'], label="포스팅 스타일", value="친근한")
            ref1_1 = gr.Textbox(label="참고 글 1", placeholder="참고할 글을 입력하세요")
            ref2_1 = gr.Textbox(label="참고 글 2", placeholder="참고할 글을 입력하세요")
            ref3_1 = gr.Textbox(label="참고 글 3", placeholder="참고할 글을 입력하세요")
            outline_button_1 = gr.Button("아웃라인 생성")
            outline_output_1 = gr.Textbox(label="생성된 아웃라인")

            outline_1 = gr.Textbox(label="작성할 아웃라인", placeholder="생성된 아웃라인을 입력하세요")
            blog_button_1 = gr.Button("블로그 글 생성")
            blog_output_1 = gr.HTML(label="생성된 블로그 글 (HTML 형식)")

            outline_button_1.click(generate_outline_1_wrapper, inputs=[category_1, style_1, ref1_1, ref2_1, ref3_1], outputs=outline_output_1)
            blog_button_1.click(generate_blog_post_1_wrapper, inputs=[category_1, style_1, ref1_1, ref2_1, ref3_1, outline_1], outputs=blog_output_1)

        # 두 번째 탭: 여행
        with gr.Tab("여행"):
            category_2 = gr.Radio(choices=['여행 단일', '여행 코스'], label="포스팅 카테고리", value="여행 단일")
            style_2 = gr.Radio(choices=['친근한', '일반', '전문적인'], label="포스팅 스타일", value="친근한")
            ref1_2 = gr.Textbox(label="참고 글 1", placeholder="참고할 글을 입력하세요")
            ref2_2 = gr.Textbox(label="참고 글 2", placeholder="참고할 글을 입력하세요")
            ref3_2 = gr.Textbox(label="참고 글 3", placeholder="참고할 글을 입력하세요")
            outline_button_2 = gr.Button("아웃라인 생성")
            outline_output_2 = gr.Textbox(label="생성된 아웃라인")

            outline_2 = gr.Textbox(label="작성할 아웃라인", placeholder="생성된 아웃라인을 입력하세요")
            blog_button_2 = gr.Button("블로그 글 생성")
            blog_output_2 = gr.HTML(label="생성된 블로그 글 (HTML 형식)")

            outline_button_2.click(generate_outline_2_wrapper, inputs=[category_2, style_2, ref1_2, ref2_2, ref3_2], outputs=outline_output_2)
            blog_button_2.click(generate_blog_post_2_wrapper, inputs=[category_2, style_2, ref1_2, ref2_2, ref3_2, outline_2], outputs=blog_output_2)

        # 세 번째 탭: 상품후기
        with gr.Tab("상품후기"):
            category_3 = gr.Radio(choices=['일반', '기능집중형', '고객반응형'], label="포스팅 카테고리", value="일반")
            style_3 = gr.Radio(choices=['친근한', '일반', '전문적인'], label="포스팅 스타일", value="친근한")
            ref1_3 = gr.Textbox(label="참고 글 1", placeholder="참고할 글을 입력하세요")
            ref2_3 = gr.Textbox(label="참고 글 2", placeholder="참고할 글을 입력하세요")
            ref3_3 = gr.Textbox(label="참고 글 3", placeholder="참고할 글을 입력하세요")
            outline_button_3 = gr.Button("아웃라인 생성")
            outline_output_3 = gr.Textbox(label="생성된 아웃라인")

            outline_3 = gr.Textbox(label="작성할 아웃라인", placeholder="생성된 아웃라인을 입력하세요")
            blog_button_3 = gr.Button("블로그 글 생성")
            blog_output_3 = gr.HTML(label="생성된 블로그 글 (HTML 형식)")

            outline_button_3.click(generate_outline_3_wrapper, inputs=[category_3, style_3, ref1_3, ref2_3, ref3_3], outputs=outline_output_3)
            blog_button_3.click(generate_blog_post_3_wrapper, inputs=[category_3, style_3, ref1_3, ref2_3, ref3_3, outline_3], outputs=blog_output_3)

        # 네 번째 탭: 핵심기능집중형
        with gr.Tab("핵심기능집중형"):
            category_4 = gr.Radio(choices=['핵심기능집중형'], label="포스팅 카테고리", value="핵심기능집중형")
            style_4 = gr.Radio(choices=['친근한', '일반', '전문적인'], label="포스팅 스타일", value="친근한")
            ref1_4 = gr.Textbox(label="참고 글 1", placeholder="참고할 글을 입력하세요")
            ref2_4 = gr.Textbox(label="참고 글 2", placeholder="참고할 글을 입력하세요")
            ref3_4 = gr.Textbox(label="참고 글 3", placeholder="참고할 글을 입력하세요")
            outline_button_4 = gr.Button("핵심기능 선정하기")
            outline_result1 = gr.Textbox(label="핵심기능 1")
            outline_result2 = gr.Textbox(label="핵심기능 2")
            outline_result3 = gr.Textbox(label="핵심기능 3")
            outline_4 = gr.Textbox(label="작성할 핵심기능 주제 중에 1개만 입력해주세요", placeholder="핵심기능 1개만 입력하세요")
            blog_button_4 = gr.Button("블로그 글 생성")
            blog_output_4 = gr.HTML(label="생성된 블로그 글 (HTML 형식)")

            outline_button_4.click(generate_outline_4_wrapper, inputs=[category_4, style_4, ref1_4, ref2_4, ref3_4], outputs=[outline_result1, outline_result2, outline_result3])
            blog_button_4.click(generate_blog_post_4_wrapper, inputs=[category_4, style_4, ref1_4, ref2_4, ref3_4, outline_4], outputs=blog_output_4)

        # 다섯 번째 탭: 방문후기형
        with gr.Tab("방문후기형"):
            category_5 = gr.Radio(choices=['방문후기형'], label="포스팅 카테고리", value="방문후기형")
            style_5 = gr.Radio(choices=['친근한', '일반', '전문적인'], label="포스팅 스타일", value="친근한")
            ref1_5 = gr.Textbox(label="참고 글 1", placeholder="참고할 글을 입력하세요")
            ref2_5 = gr.Textbox(label="참고 글 2", placeholder="참고할 글을 입력하세요")
            ref3_5 = gr.Textbox(label="참고 글 3", placeholder="참고할 글을 입력하세요")
            photo_recommendations_5 = gr.Textbox(label="사진 형태 키워드", placeholder="사진 추천 키워드를 입력하세요")
            outline_button_5 = gr.Button("아웃라인 생성")
            outline_output_5 = gr.Textbox(label="생성된 아웃라인")

            outline_5 = gr.Textbox(label="작성할 아웃라인", placeholder="생성된 아웃라인을 입력하세요")
            blog_button_5 = gr.Button("블로그 글 생성")
            blog_output_5 = gr.HTML(label="생성된 블로그 글 (HTML 형식)")

            outline_button_5.click(
                generate_outline_5_wrapper,
                inputs=[category_5, style_5, ref1_5, ref2_5, ref3_5, photo_recommendations_5],
                outputs=outline_output_5
            )
            blog_button_5.click(
                generate_blog_post_5_wrapper,
                inputs=[category_5, style_5, ref1_5, ref2_5, ref3_5, outline_5, photo_recommendations_5],
                outputs=blog_output_5
            )

    return demo

# 인터페이스 실행
if __name__ == "__main__":
    interface().launch(share=True)