app.py

import os
import gradio as gr
from dotenv import load_dotenv
import logging
import time
import uuid
import google.generativeai as genai

# 로깅 설정 - INFO 레벨로 변경
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

# 환경 변수 로드
load_dotenv()

# API 키 설정
GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
logger.info("API 키 설정 완료")

# Google Gemini 초기화
genai.configure(api_key=GOOGLE_API_KEY)

# 모델별 프롬프트 정의
GEMINI_PROMPTS = {
    "맞춤법 검사기": """
# 목적: 한국어 문법 교정 및 개선
## 지침
1. 주어진 텍스트의 문법을 전문적으로 검토하고 교정하세요.
2. 원본 텍스트의 의미와 의도를 철저히 유지하면서 문법적인 부분만 수정하세요.
3. 다음 요소들을 한국어 문법 규칙에 맞게 정확하게 수정하세요:
   - 맞춤법 오류
   - 띄어쓰기 오류
   - 문장 부호 사용
   - 조사 사용의 정확성
   - 어미 활용의 정확성
   - 문장 성분의 호응 관계
4. 절대 원본 텍스트에 없는 내용을 추가하거나 의미를 변경하지 마세요.
5. 부가 설명이나 메타 정보(예: "수정된 텍스트:", "개선된 텍스트:" 등)는 출력하지 마세요.
6. 오직 교정된 텍스트만 출력하세요.
""",

    "글 다듬기": """
# 목적: 문장 구조와 표현 개선
## 지침
1. 문장 구조 최적화
   - 불필요한 문구와 중복 표현을 제거하여 간결성 확보
   - 주어-서술어 관계를 명확히 하여 의미 전달력 강화
   - 과도하게 긴 문장은 적절히 분리하여 가독성 향상
   - 문장 간의 논리적 연결성 확보

2. 단락 구성 최적화
   - 각 단락이 명확한 핵심 주제를 가지도록 조정
   - 단락 간 자연스러운 흐름을 위한 연결어 사용
   - 단락 길이의 균형 유지

3. 어휘 및 표현 개선
   - 불필요한 외래어는 적절한 한국어로 대체
   - 추상적이거나 모호한 표현은 구체적이고 명확한 용어로 대체
   - 중복되는 표현 제거로 문장 효율성 증대
   - 관용적 표현과 적절한 비유를 활용하여 표현력 강화

4. 전달력 강화
   - 핵심 메시지가 돋보이도록 문장 구조 재배치
   - 논리적 흐름과 일관성 확보
   - 독자의 이해도를 고려한 표현 선택

5. 일관성 유지
   - 문체와 톤의 일관성 유지
   - 전문 용어와 표현의 통일성 확보
   - 시제의 일관된 사용

## 출력 형식
- 원본 텍스트와 동일한 말투와 어조를 유지하되, 위 기준에 따라 개선된 텍스트만 출력
- 부가 설명이나 메타 정보는 포함하지 않음
""",

    "명언 인용하기": """
# 목적: 명언을 활용한 설득력 있는 텍스트 개선
## 지침
1. 문맥 분석
   - 텍스트의 핵심 주제와 의도 파악
   - 텍스트의 분위기와 톤 분석
   - 독자층과 목적 고려

2. 적절한 명언 선택
   - 텍스트의 주제와 직접적으로 연관된 명언 선택
   - 다양한 분야(철학, 문학, 역사, 과학 등)의 명언 고려
   - 고전적인 명언뿐만 아니라 현대 유명인(연예인, 기업인, 운동선수, 정치인 등)의 인상적인 말도 적극 활용
   - 한국 및 세계의 저명한 인물의 명언 활용
   - 텍스트의 톤과 분위기에 어울리는 명언 선택
   - 가능하면 시의적절하고 최신 트렌드를 반영한 현대적 명언 우선 고려

3. 명언 통합
   - 텍스트 내 가장 효과적인 위치에 명언 배치 (시작, 중간, 또는 결론)
   - 명언을 자연스럽게 도입하여 기존 텍스트와 조화롭게 통합
   - 필요시 명언의 출처(인물 이름) 포함
   - 딱 한 번만 명언 인용하기
   - 현대 유명인의 명언 사용 시 간단한 맥락 설명 고려(필요한 경우)

4. 전체 텍스트 조정
   - 명언 도입 후 필요시 전후 문장 조정하여 자연스러운 흐름 유지
   - 원본 텍스트의 핵심 메시지와 의도 보존
   - 전체적인 일관성과 응집력 유지
   - 명언이 텍스트에 신선함과 권위를 부여하는지 확인

## 출력 형식
- 명언이 자연스럽게 통합된 개선된 텍스트만 출력
- 부가 설명이나 메타 정보 없이 순수하게 개선된 텍스트만 제공
""",

    "맞춤형 변환": """
# 목적: 사용자 정의 페르소나와 목적에 맞는 텍스트 변환
## 지침
1. 페르소나 분석 및 적용
   - 사용자가 제공한 페르소나의 특성 분석 (말투, 어휘 선택, 문장 구조 등)
   - 해당 페르소나가 실제로 작성했을 법한 스타일로 텍스트 재구성
   - 페르소나의 고유한 표현 방식과 관점 반영

2. 목적 분석 및 적용
   - 사용자가 명시한 목적에 맞게 텍스트의 전반적인 방향성 조정
   - 목적에 맞는 구조, 논조, 강조점 설정
   - 목적 달성에 필요한 요소 강화 및 불필요한 요소 제거

3. 텍스트 맞춤 변환
   - 페르소나와 목적을 종합적으로 고려한 텍스트 변환
   - 원본 텍스트의 핵심 정보와 주요 논점 유지
   - 어휘, 문장 구조, 비유, 예시 등을 페르소나와 목적에 맞게 조정
   - 전체적인 일관성과 진정성 유지

4. 품질 및 효과 최적화
   - 명확성과 이해도 향상
   - 설득력과 호소력 강화
   - 독창성과 참신함 추구
   - 목표 독자에게 효과적으로 전달될 수 있는 형태로 최적화

## 출력 형식
- 페르소나와 목적에 맞게 변형된 텍스트만 출력
- 부가 설명이나 메타 정보 없이 순수하게 변환된 텍스트만 제공
- 사용자가 명시한 목적과 페르소나의 특성이 명확히 드러나는 결과물 제공
""",
}

def process_text(input_text, improvement_type, custom_purpose, persona, temperature, top_p):
    """Gemini 모델용 텍스트 처리 함수"""
    try:
        print("텍스트 처리 시작")
        request_id = str(uuid.uuid4())[:8]
        timestamp_micro = int(time.time() * 1000000) % 1000

        if improvement_type == "맞춤형 변환":
            purpose = f"다음 페르소나를 가지고 텍스트를 다듬으세요: {persona}\n\n목적: {custom_purpose}"
        else:
            purpose = GEMINI_PROMPTS[improvement_type]

        model = genai.GenerativeModel(
            model_name="gemini-2.0-flash",
            generation_config={
                "temperature": temperature,
                "top_p": top_p,
                "max_output_tokens": 2000,
            }
        )

        # 시스템 지시사항을 첫 번째 사용자 메시지에 포함
        prompt = f"REQ-{request_id}-{timestamp_micro}\n\n다음 텍스트를 목적에 맞게 다듬어주세요.\n\n텍스트: {input_text}\n\n목적: {purpose}"
        
        response = model.generate_content(prompt)
        print("텍스트 처리 완료")
        return response.text

    except Exception as e:
        print("텍스트 처리 실패")
        logger.error(f"처리 중 오류 발생: {str(e)}")
        return f"오류 발생: {str(e)}"

def create_interface():
    # FontAwesome 아이콘 포함
    fontawesome = """
    <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.0.0/css/all.min.css" crossorigin="anonymous" referrerpolicy="no-referrer" />
    """
    
    # 커스텀 CSS 스타일
    css = """
    :root {
        --primary-color: #FF7F00;
        --secondary-color: #ff9a8b;
        --accent-color: #FF6B6B;
        --background-color: #FFF8F0;
        --card-bg: #ffffff;
        --text-color: #334155;
        --border-radius: 18px;
        --shadow: 0 8px 30px rgba(251, 127, 13, 0.08);
    }
    
    body {
        font-family: 'Pretendard', 'Noto Sans KR', -apple-system, BlinkMacSystemFont, sans-serif;
        background-color: var(--background-color);
        color: var(--text-color);
        line-height: 1.6;
    }
    
    footer {
        visibility: hidden;
    }
    
    .container {
        max-width: 1200px;
        margin: 0 auto;
    }
    
    .header {
        background: linear-gradient(135deg, #FF7F00, #FF9A5B);
        padding: 2rem;
        border-radius: 15px;
        margin-bottom: 20px;
        box-shadow: var(--shadow);
        text-align: center;
        color: white;
    }
    
    .header h1 {
        margin: 0;
        font-size: 2.5rem;
        font-weight: 700;
    }
    
    .header p {
        margin: 10px 0 0;
        font-size: 1.2rem;
        opacity: 0.9;
    }
    
    .card {
        background-color: var(--card-bg);
        border-radius: var(--border-radius);
        padding: 20px;
        margin: 10px 0;
        box-shadow: var(--shadow);
        border: 1px solid rgba(0, 0, 0, 0.04);
    }
    
    .button-primary {
        border-radius: 30px !important;
        background: linear-gradient(135deg, var(--primary-color), var(--secondary-color)) !important;
        color: white !important;
        font-size: 18px !important;
        padding: 10px 20px !important;
        border: none;
        box-shadow: 0 4px 8px rgba(251, 127, 13, 0.25);
        transition: transform 0.3s ease;
        text-align: center;
        font-weight: 600;
    }
    
    .button-primary:hover {
        transform: translateY(-2px);
        box-shadow: 0 6px 12px rgba(251, 127, 13, 0.3);
    }
    
    .section-title {
        display: flex;
        align-items: center;
        font-size: 20px;
        font-weight: 700;
        color: #333333;
        margin-bottom: 15px;
        padding-bottom: 8px;
        border-bottom: 2px solid #FF7F00;
    }
    
    .section-title i {
        margin-right: 10px;
        color: #FF7F00;
    }
    
    .guide-container {
        background-color: #FFF0E2;
        border-radius: var(--border-radius);
        padding: 1.5rem;
        margin-bottom: 1.5rem;
        border: 1px solid rgba(255, 127, 0, 0.1);
    }
    
    .guide-title {
        font-size: 1.3rem;
        font-weight: 700;
        color: var(--primary-color);
        margin-bottom: 1rem;
        display: flex;
        align-items: center;
    }
    
    .guide-title i {
        margin-right: 0.8rem;
        font-size: 1.3rem;
    }
    
    .guide-item {
        display: flex;
        margin-bottom: 0.8rem;
        align-items: flex-start;
    }
    
    .guide-number {
        background-color: var(--primary-color);
        color: white;
        width: 24px;
        height: 24px;
        border-radius: 50%;
        display: flex;
        align-items: center;
        justify-content: center;
        font-weight: bold;
        margin-right: 10px;
        flex-shrink: 0;
        font-size: 14px;
    }
    
    .guide-text {
        flex: 1;
        line-height: 1.6;
    }
    
    .feature-tag {
        display: inline-block;
        background-color: rgba(255, 127, 0, 0.1);
        color: #FF7F00;
        padding: 3px 10px;
        border-radius: 12px;
        font-size: 14px;
        font-weight: 600;
        margin-right: 8px;
        margin-bottom: 8px;
    }
    
    .input-label {
        font-weight: 600;
        margin-bottom: 8px;
        color: #444;
    }
    
    /* 그라디오 요소 스타일 커스터마이징 */
    .gr-input, .gr-text-input, .gr-sample-inputs {
        border-radius: var(--border-radius) !important;
        border: 1px solid #e5e7eb !important;
        padding: 12px !important;
        transition: all 0.3s ease !important;
    }
    
    .gr-input:focus, .gr-text-input:focus {
        border-color: var(--primary-color) !important;
        outline: none !important;
        box-shadow: 0 0 0 2px rgba(251, 127, 13, 0.2) !important;
    }
    
    /* 스크롤바 스타일링 */
    ::-webkit-scrollbar {
        width: 8px;
        height: 8px;
    }
    
    ::-webkit-scrollbar-track {
        background: rgba(0, 0, 0, 0.05);
        border-radius: 10px;
    }
    
    ::-webkit-scrollbar-thumb {
        background: var(--primary-color);
        border-radius: 10px;
    }
    """

    with gr.Blocks(css=css, theme=gr.themes.Soft(
        primary_hue=gr.themes.Color(
            c50="#FFF7ED", 
            c100="#FFEDD5",
            c200="#FED7AA",
            c300="#FDBA74",
            c400="#FB923C",
            c500="#F97316", 
            c600="#EA580C",
            c700="#C2410C",
            c800="#9A3412",
            c900="#7C2D12", 
            c950="#431407",
        ),
        secondary_hue="zinc",
        neutral_hue="zinc",
        font=("Pretendard", "sans-serif")
    )) as demo:
        
        gr.HTML(fontawesome)
        with gr.Row():
            with gr.Column(scale=1):
                # 왼쪽 입력 영역
                with gr.Column(elem_classes="card"):
                    gr.HTML('<div class="section-title"><i class="fas fa-edit"></i> 원본 텍스트</div>')
                    input_text = gr.Textbox(
                        label="",
                        placeholder="여기에 개선하고 싶은 텍스트를 입력하세요...",
                        lines=15
                    )
                
                with gr.Column(elem_classes="card"):
                    gr.HTML('<div class="section-title"><i class="fas fa-sliders"></i> 변환 설정</div>')
                    
                    improvement_type = gr.Radio(
                        choices=["맞춤법 검사기", "글 다듬기", "명언 인용하기", "맞춤형 변환"],
                        label="변환 유형 선택",
                        value="맞춤법 검사기"
                    )
                    
                    # 맞춤형 변환 옵션들
                    custom_purpose = gr.Textbox(
                        label="변환 목적",
                        placeholder="텍스트를 어떤 방향으로 변환하고 싶으신지 설명해주세요...",
                        lines=2,
                        visible=False
                    )

                    persona = gr.Textbox(
                        label="페르소나 설정",
                        placeholder="어떤 역할이나 성격을 가진 글쓰기 스타일을 원하시나요? (예: 유머러스한 개그맨, 전문적인 분석가)",
                        lines=2,
                        visible=False
                    )

                    with gr.Row(visible=False) as generation_settings:
                        temperature = gr.Slider(
                            minimum=0.0,
                            maximum=1.0,
                            value=0.7,
                            step=0.1,
                            label="창의성 수준",
                            info="높을수록 더 창의적인 결과가 생성됩니다"
                        )
                        top_p = gr.Slider(
                            minimum=0.0,
                            maximum=1.0,
                            value=0.9,
                            step=0.1,
                            label="다양성 수준",
                            info="높을수록 더 다양한 표현을 사용합니다"
                        )
                
                submit_btn = gr.Button(
                    "✨ 텍스트 변환하기",
                    elem_classes="button-primary"
                )
            
            with gr.Column(scale=1):
                # 오른쪽 출력 영역
                with gr.Column(elem_classes="card"):
                    gr.HTML('<div class="section-title"><i class="fas fa-wand-magic-sparkles"></i> 변환된 텍스트</div>')
                    output_text = gr.Textbox(
                        label="",
                        lines=22,
                        show_copy_button=True
                    )

        def update_custom_inputs_visibility(choice):
            is_custom = choice == "맞춤형 변환"
            return {
                custom_purpose: gr.update(visible=is_custom),
                persona: gr.update(visible=is_custom),
                generation_settings: gr.update(visible=is_custom)
            }

        improvement_type.change(
            fn=update_custom_inputs_visibility,
            inputs=[improvement_type],
            outputs=[custom_purpose, persona, generation_settings]
        )

        submit_btn.click(
            fn=process_text,
            inputs=[input_text, improvement_type, custom_purpose, persona, temperature, top_p],
            outputs=output_text,
            api_name="improve_text"
        )

    return demo

if __name__ == "__main__":
    logger.info("애플리케이션 시작")
    demo = create_interface()
    demo.queue()
    demo.launch()