app.py

import gradio as gr
from transformers import pipeline

# 한국어 채팅 모델 불러오기 (예: KoAlpaca, LLaMA, Mistral 등)
chat = pipeline("text-generation", model="beomi/KoAlpaca-Polyglot-5.8B")

# 분석 함수 정의
def analyze_korean_text(text):
    prompt = f"다음 문장을 분석하고 무례하거나 과격한 표현이 있으면 지적하고, 더 친절하게 바꿔줘:\n\n{text}"
    response = chat(prompt, max_new_tokens=200)[0]['generated_text']
    return response

# Gradio 인터페이스
iface = gr.Interface(
    fn=analyze_korean_text,
    inputs=gr.Textbox(label="분석할 문장 입력"),
    outputs=gr.Textbox(label="분석 결과"),
    title="문장 분석기 (친절하게 바꿔주는 AI)",
    description="문장을 입력하면 무례한 표현을 지적하고 더 예의 있게 바꿔주는 도우미입니다."
)

# 실행
iface.launch()