import gradio as gr

def process_input(input_text, function):
    # Это просто мок-функция для демонстрации, в реальности здесь будет ваша логика обработки
    return f"Результат для функции '{function}': обработано"

def main():
    with gr.Blocks() as demo:
        gr.Markdown("### AI Research Assistant")
        with gr.Row():
            text_input = gr.Textbox(label="Введите текст или загрузите документ")
            function_select = gr.Dropdown(choices=[
                "Суммаризатор", "Поиск новых статей", "Учитель", "Критик", "Тестировщик",
                "Визуализатор связей", "Советник", "Соавтор", "Переводчик", "Аннотатор",
                "Факт-чекер", "Аналитик данных", "Стилистический редактор", "Рецензент",
                "Презентатор", "Грант-райтер", "Научный сторителлер", "Библиограф"
            ], label="Выберите функцию")
        with gr.Row():
            submit_button = gr.Button("Обработать")
            output_text = gr.Textbox(label="Результат")
        
        submit_button.click(
            fn=process_input,
            inputs=[text_input, function_select],
            outputs=output_text
        )

    demo.launch()

if __name__ == "__main__":
    main()