import gradio as gr
from openai import OpenAI

client = OpenAI()

def simplify_text(text):
    if not text:
        return "", "0 tokenów", "0 tokenów"
    
    response = client.chat.completions.create(
        model="gpt-4o-mini",
        messages=[
            {"role": "system", "content": "Jesteś ekspertem w upraszczaniu tekstów. Twoim zadaniem jest uproszczenie podanego tekstu, zachowując jego główne znaczenie, ale używając prostszego języka. Nie formatuj tekstu za pomocą znaczników Markdown."},
            {"role": "user", "content": f"Uprość ten tekst:\n{text}"}
        ],
        temperature=0.7
    )
    
    simplified_text = response.choices[0].message.content
    input_tokens = response.usage.prompt_tokens
    output_tokens = response.usage.completion_tokens
    
    return (
        simplified_text,
        f"{input_tokens} tokenów",
        f"{output_tokens} tokenów"
    )

with gr.Blocks() as demo:
    gr.Markdown("# Prostownik - narzędzie do upraszczania tekstów")
    
    with gr.Row():
        with gr.Column():
            input_area = gr.TextArea(
                label="Tekst wejściowy",
                placeholder="Wklej tutaj tekst do uproszczenia...",
                lines=10
            )
            input_tokens = gr.Textbox(label="Liczba tokenów wejściowych")
            
        with gr.Column():
            output_area = gr.TextArea(
                label="Tekst uproszczony",
                lines=10,
                show_copy_button=True
            )
            output_tokens = gr.Textbox(label="Liczba tokenów wyjściowych")
    
    input_area.change(
        fn=simplify_text,
        inputs=input_area,
        outputs=[output_area, input_tokens, output_tokens]
    )

    demo.launch(share=True)