o a4i&@sdZddlZddlZddlZddlZddlmZmZmZm Z ddl m Z e ddZ e dZe Gdd d ZGd d d Zed krHeZdSdS) a Smart Translator - Gemini Powered Technical Translation ======================================================= Replaces legacy Regex+DeepL approach with Context-Aware AI Translation. Uses 'dictionary.db' to enforce technical terminology while maintaining natural grammar. N)ListDictTupleOptional) dataclassDICTIONARY_PATHz/mnt/pdfs/dictionary_v2.dbOPENAI_API_KEYc@s:eZdZUeed<eed<eeefed<dZeed<dS)TranslationResult original_texttranslated_text used_terms? confidenceN)__name__ __module__ __qualname__str__annotations__rrfloatrr2/var/www/html/PEPCVSON/scripts/smart_translator.pyr s r c @seZdZdefdedefddZddZdd Zd"d ed ed edefddZ dedefddZ dede eeffddZ dede eeffddZ d#dededede eefdef ddZdedefd d!ZdS)$SmartTranslatorNapi_key dict_pathcCst|ptd|_||_d|_i|_|||jr1ddlm }|j |jd| d|_ dSd|_ t ddS)NGEMINI_API_KEYr)rzgemini-2.0-flashu1⚠️ Gemini API Key missing for SmartTranslator)osgetenvrrconn _term_cache _connect_db_load_dictionarygoogle.generativeai generativeai configureGenerativeModelmodelprint)selfrrgenairrr__init__s  zSmartTranslator.__init__c Cshztj|jrt|j|_tj|j_WdSWdSt y3}zt d|WYd}~dSd}~ww)Nu⚠️ DB Connection Error: ) rpathexistsrsqlite3connectrRow row_factory Exceptionr&)r'errrr.szSmartTranslator._connect_dbc Cs|jsdSzF|j}|dd}|D]*}|d}|d}||jvr2||j|<||jvr<||j|<|d7}qtd|dWdStye}ztd |WYd}~dSd}~ww) z4Load high-quality terms into memory for quick lookupNab SELECT canonical_tr, canonical_en, category FROM technical_terms WHERE (category IN ('parts-catalog', 'katalog-detailed', 'equipment', 'parts', 'auto-root', 'root', 'user-added') OR category IS NULL) AND length(canonical_en) > 2 AND length(canonical_tr) > 2 r canonical_tr canonical_enu✓ SmartTranslator loaded z terms.u⚠️ Dictionary Load Error: ) rcursorexecutefetchalllowerstriprr&r0)r'ccountrowtrenr1rrrr 6s     z SmartTranslator._load_dictionary user-addedr>r=categoryreturnc Cs|jsdSzQ|}|}|j}|d||f|}|r2|d||||dfn |d|||f|j||j|<||j|<td|d|WdSt yp}ztd |WYd }~dSd }~ww) z3Add a new term to the dictionary database and cacheFzISELECT id FROM technical_terms WHERE canonical_en = ? OR canonical_tr = ?z UPDATE technical_terms SET canonical_tr = ?, canonical_en = ?, category = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ? rz INSERT INTO technical_terms (canonical_en, canonical_tr, category, confidence_score) VALUES (?, ?, ?, 1.0) u✓ Term added/updated: z <-> Tu⚠️ Term Add Error: N) rr9r8r5r6fetchonecommitrr&r0)r'r>r=r@r:existingr1rrradd_termLs0       zSmartTranslator.add_termtermc Cs|jsdSzD|}|j}|d||f|jg}|j|}|r2|||||D] }||jvr?|j|=q4t d|WdSt yc}zt d|WYd}~dSd}~ww)z'Delete a term by English or Turkish keyFzFDELETE FROM technical_terms WHERE canonical_en = ? OR canonical_tr = ?u✓ Term deleted: Tu⚠️ Term Delete Error: N) rr9r8r5r6rCrgetappendr&r0)r'rFr:keys_to_deletevalkr1rrr delete_termos,        zSmartTranslator.delete_termtextc Csi}|}td|}g}d}t|}td|dD]}t||dD]} d|| | |} || q'q|D]} | |jvrK|j| || <q=|S)Nz\b\w+\br4 ) r8r9refindalllenrangejoinrHr) r'rM found_terms text_lowertokensn_gramsmax_nlengthnigramrrr_find_relevant_termss     z$SmartTranslator._find_relevant_termscCs ||S)N)r^)r'rMrrrget_relevant_termss z"SmartTranslator.get_relevant_termsENTR source_lang target_langterms_overridec Cs|js t||idS|dur|n||}d|d|dtj|ddd|d }z|j|}|j}t|||d WSty\} zt d | t||idWYd} ~ Sd} ~ ww) Ngu You are an expert technical translator specializing in CONSTRUCTION MACHINERY and HEAVY EQUIPMENT (Excavators, Loaders, Backhoes, Dozers, etc.). DOMAIN CONTEXT: - The text belongs to the "World of Heavy Machinery" (İş Makinası Dünyası). - Avoid generic, medical, or standard IT translations. - For example: "Travel" means "Seyir/Yürüyüş", not "Seyahat". "Shoe" means "Pabuç", not "Ayakkabı". TASK: Translate the input text to za with high technical accuracy. CRITICAL RULES: 1. If text is already in z, just correct terminology. 2. Maintain sentence structure and style. 3. STRICTLY USE these dictionary terms in your translation: F) ensure_asciizd 4. Output ONLY the translated text. Do not add explanations. Input Text: " " r zTranslation Error: ) r%r r^jsondumpsgenerate_contentrMr9r0r&) r'rMrbrcrdrelevant_termspromptresponser r1rrr translates*     zSmartTranslator.translatec Cs~|js|Sd|d}z|j|}|j}d|vr!|dd}|WSty>}ztd||WYd}~Sd}~ww)u`Kullanıcı sorgusundaki yazım hatalarını düzeltir (Gemini Powered) - Sadece Düzeltme yaparu Act as a spell checker and grammar corrector for Heavy Equipment/Machinery terminology. Task: 1. Correct spelling mistakes (typos) in the input text. 2. Fix broken sentence structures. 3. Normalize technical terms (e.g. "yüüyüş" -> "yürüyüş"). 4. Do NOT translate. Keep the original language (Turkish or English). 5. Output ONLY the corrected text. Input: "rf"zCorrection Error: N)r%rirMr9replacer0r&)r'rMrkrlcorrected_textr1rrrcorrect_terminologys   z#SmartTranslator.correct_terminology)r?)r`raN)rrrrrr)rr boolrErLrr^r_r rmrrrrrrrs#!(%r__main__)__doc__rrgr,rPtypingrrrr dataclassesrrrrr rrtrrrrs    [