# %%
from openai import OpenAI
import gradio as gr



# %%
client=OpenAI()

# Lista reguł prostego języka
rules = [
    "Używaj krótkich zdań",
    "Unikaj trudnych słów",
    "Stosuj proste konstrukcje gramatyczne",
    "Unikaj żargonu i skrótów",
    "Stosuj aktywną stronę"
]

def generate_response(prompt, selected_rules=None):
    if selected_rules is None:
        selected_rules = rules
        
    # Przygotuj zasady jako tekst
    rules_text = "\n".join([f"- {rule}" for rule in selected_rules])
    
    # Stwórz pełen prompt z regułami
    full_prompt = f"""Uprość poniższy tekst zgodnie z zasadami prostego języka:
{rules_text}

Tekst:
{prompt}"""
    
    response = client.chat.completions.create(
        model="gpt-4.1",
        messages=[
            {"role": "system", "content": "Jesteś pomocnym asystentem upraszczającym teksty."},
            {"role": "user", "content": full_prompt}
        ],
        temperature=0.1,
    )
    return response.choices[0].message.content

with gr.Blocks() as demo:
    gr.Markdown("# Uproszczenie Tekstu")
    gr.Markdown("Wprowadź tekst, który chcesz uprościć. Model AI uprości go, aby był bardziej zrozumiały.")
    
    with gr.Accordion("Reguły prostego języka", open=False):
        rules_checkboxes = gr.CheckboxGroup(
            choices=rules,
            value=rules,  # domyślnie wszystkie zaznaczone
            label="Zastosuj reguły"
        )
    
    input_text = gr.TextArea(label="Tekst do uproszczenia")
    output_text = gr.TextArea(label="Uproszczony Tekst")
    
    submit_button = gr.Button("Uprość tekst")
    submit_button.click(
        generate_response,
        inputs=[input_text, rules_checkboxes],
        outputs=output_text
    )

demo.launch(inbrowser=True)