app.py

import gradio as gr
from huggingface_hub import InferenceClient
import os

# 🎨 아기자기하고 트렌디한 테마 설정
theme = gr.themes.Soft(primary_hue="indigo", secondary_hue="blue")

# ⭐ 에러 없이 가장 똑똑하고 빠른 Qwen 2.5 Instruct 모델 사용
hf_token = os.environ.get("HF_TOKEN")
client = InferenceClient("Qwen/Qwen2.5-7B-Instruct", token=hf_token)

def transform_english(basic_text):
    if not basic_text:
        return "⚠️ 영작할 문장을 입력해주세요.", "", ""
    if not hf_token:
        return "❌ HF_TOKEN 설정 오류", "토큰을 확인해주세요.", "토큰을 확인해주세요."

    # AI에게 3가지 인격을 동시에 부여하는 강력한 프롬프트
    prompt = f"""
    You are an expert English linguist. Rewrite the following basic English text into three totally different styles.
    
    Basic Text: "{basic_text}"
    
    Task:
    1. [Gen Z / TikTok]: Rewrite it using trendy internet slang, Gen-Z vocabulary, emojis, and a highly dramatic, casual tone.
    2. [Harvard / Business]: Rewrite it in a highly sophisticated, formal, and elegant academic/business tone (like Harvard Business Review).
    3. [Korean CSAT (수능)]: Rewrite it using unnecessarily complex grammatical structures typical of the Korean CSAT reading section. Forcefully include inversions (e.g., Not only did...), participial phrases, and heavy abstract vocabulary.
    
    Format your response EXACTLY like this with no other text:
    [TikTok]
    (your text here)
    [Harvard]
    (your text here)
    [CSAT]
    (your text here)
    """
    
    try:
        response = client.chat_completion(
            messages=[{"role": "user", "content": prompt}],
            max_tokens=800,
            temperature=0.8 # 살짝 창의력을 높여서 더 극적인 변화를 유도
        )
        result = response.choices[0].message.content
        
        # AI의 응답을 3개의 구역으로 예쁘게 쪼개기
        try:
            tiktok = result.split("[TikTok]")[1].split("[Harvard]")[0].strip()
            harvard = result.split("[Harvard]")[1].split("[CSAT]")[0].strip()
            csat = result.split("[CSAT]")[1].strip()
        except Exception:
            # AI가 가끔 양식을 안 지킬 때를 대비한 예외 처리
            return result, "결과 파싱 에러", "결과 파싱 에러"
            
        return tiktok, harvard, csat
        
    except Exception as e:
        return f"❌ 오류 발생: {str(e)}", "", ""

# UI 인터페이스 구성
with gr.Blocks(theme=theme) as demo:
    gr.Markdown("# 🎭 초딩 영작을 하버드생으로! 다중인격 교정기")
    gr.Markdown("### 콩글리시나 아주 단순한 문장을 입력하면, AI가 3가지 인격으로 완벽하게 포장해드립니다.")
    
    with gr.Row():
        with gr.Column(scale=1):
            gr.Markdown("### ✍️ 1. 나의 기초 영작 입력")
            user_input = gr.Textbox(
                label="아주 단순한 영어로 적어보세요", 
                placeholder="예: I go to PC room and play game. It was fun.",
                lines=5
            )
            
            gr.Examples(
                examples=[
                    "I go to PC room and play game. It was fun.",
                    "I ate a very big hamburger. My stomach is full.",
                    "I sleep during math class because the teacher is boring."
                ],
                inputs=user_input,
                label="🌟 꿀잼 콩글리시 예시 (클릭!)"
            )
            
            run_btn = gr.Button("✨ 다중인격 변환 시작! ✨", variant="primary")
            
        with gr.Column(scale=2):
            gr.Markdown("### 🪄 2. AI의 3단 변신 결과")
            with gr.Row():
                out_tiktok = gr.Textbox(label="📱 원어민 10대 슬랭 (틱톡/인스타 감성)", lines=6, interactive=False)
                out_harvard = gr.Textbox(label="👔 비즈니스 / 하버드 (고급 학술 톤)", lines=6, interactive=False)
                out_csat = gr.Textbox(label="😈 수능 29번 어법 (복잡한 도치/분사구문)", lines=6, interactive=False)

    run_btn.click(
        transform_english, 
        inputs=[user_input], 
        outputs=[out_tiktok, out_harvard, out_csat]
    )

if __name__ == "__main__":
    demo.launch()