from langdetect import detect
from google_trans_new import google_translator
import pandas as pd
import multiprocessing as mp
import numpy as np
from dotenv import load_dotenv
from tqdm import tqdm
import logging

from database import Data_table
from database import insert_translated
from clean_data import clean_content
from setup_logging import setup_logging

def translate(text, target_lang='en'):
    try:
        input_lang = detect(text)
    except Exception as e:
        print(f'error when detecting language: {e}')
        return ''
    if input_lang == target_lang: return text
    translator = google_translator()
    try:
        text_trans = translator.translate(text, lang_src=input_lang, lang_tgt=target_lang)
    except Exception as e:
        print(f'error when translating text: {e}')
        return ''
    return text_trans

def get_untranslated():
    with Data_table() as dt:
        query = (
            f'SELECT * FROM {dt._name} '
            'WHERE content_en IS NULL '
            'ORDER BY random() '
            'LIMIT 1 '
        )
        df = pd.read_sql_query(query, dt.con)
    if len(df)==0: return None
    val_dict = {
        'id': int(df['id'][0]),
        'content': str(df['content'][0])
    }
    return val_dict

def continously_translate():
    logging.info('began translating')
    while True:
        try:
            val_dict = get_untranslated()
        except Exception as e:
            logging.warning(f'error when getting untranslated content from database: {e}')
            continue
        if val_dict is None: break
        # clean
        try:
            content_clean = clean_content(val_dict['content'])
        except Exception as e:
            logging.warning(f'error when cleaning content: {e}')
            continue
        # translate
        try:
            content_en = translate(content_clean)
        except Exception as e:
            logging.warning(f'error when translating content: {e}')
            continue
        # upload to database
        val_dict['content_en'] = content_en
        try:
            insert_translated(val_dict)
        except Exception as e:
            logging.warning(f'error when sending translated content to database: {e}')
            continue
        logging.info('translated content')
    logging.info('no more untranslated content located')

if __name__ == '__main__':
    load_dotenv()
    setup_logging()
    continously_translate()
    # later try to set all empty strings back to null and retranslate failed...