import gradio as gr
from utils.blago_core import BLAGO

# Инициализация BLAGO
blago = BLAGO()

def submit_data(bio, connections, assets, skills, needs):
    user_data = {
        "bio": bio,
        "connections": connections.split(","),
        "assets": assets.split(","),
        "skills": skills.split(","),
        "needs": needs.split(",")
    }
    blago.add_user_data(user_data)
    return "Данные успешно добавлены!"

def get_community_summary():
    summary = blago.get_summary()
    return f"Участников: {summary['total_users']}\nОбщие активы: {summary['total_assets']}\nОбщие потребности: {summary['total_needs']}"

# Интерфейс Gradio
with gr.Blocks(title="BLAGO Demo") as demo:
    gr.Markdown("# BLAGO Neural Network Demo")
    gr.Markdown("Введите свои данные для участия в сообществе BLAGO.")
    
    with gr.Row():
        with gr.Column():
            bio = gr.Textbox(label="Кто я (биография)")
            connections = gr.Textbox(label="Кого знаю (личные связи, через запятую)")
            assets = gr.Textbox(label="Что имею (активы, через запятую)")
            skills = gr.Textbox(label="Что умею (навыки, через запятую)")
            needs = gr.Textbox(label="Что хочу (потребности, через запятую)")
            submit_btn = gr.Button("Отправить данные")
        with gr.Column():
            output = gr.Textbox(label="Результат")
            summary_btn = gr.Button("Показать сводку сообщества")
            summary_output = gr.Textbox(label="Сводка сообщества")
    
    submit_btn.click(
        fn=submit_data,
        inputs=[bio, connections, assets, skills, needs],
        outputs=output
    )
    summary_btn.click(
        fn=get_community_summary,
        inputs=None,
        outputs=summary_output
    )

demo.launch()