import pandas as pd
from translate import Translator


def convert_and_translate_headers(input_file: str, output_file: str, sheet_name: str = None):
    try:
        # Читаем файл Excel
        data = pd.read_excel(input_file, sheet_name=sheet_name)

        # Инициализируем переводчик
        translator = Translator(from_lang='nl', to_lang='en')

        # Переводим названия колонок
        translated_columns = {col: translator.translate(col) for col in data.columns}
        data.rename(columns=translated_columns, inplace=True)

        # Сохраняем преобразованные данные в CSV
        data.to_csv(output_file, index=False)
        print(f"Файл успешно конвертирован и сохранен: {output_file}")
    except Exception as e:
        print(f"Произошла ошибка: {e}")


input_xlsx = "test.xlsx"  # Путь к входному .xlsx файлу
output_csv = "translated_output.csv"  # Путь к выходному .csv файлу
sheet = "Sheet1"  # Укажите имя листа, если нужно

convert_and_translate_headers(input_xlsx, output_csv, sheet)