TranslateMarkdown / translation_utils.py
Hisab Cloud
Upload 5 files
08b7f89 verified
raw
history blame contribute delete
No virus
4.18 kB
import requests
# from flask_cors import CORS
from groq import Groq
# CORS()
# GROQ_API_URL = "https://api.groq.com/translate" # استبدل هذا بعنوان API الخاص بـ Groq
GROQ_API_KEY = "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" # ضع هنا مفتاح API الخاص بك
client = Groq(api_key=GROQ_API_KEY)
def translate_texts_groq(texts, target_language="ar"):
"""
Translates a list of texts to the target language using Groq.
:param texts: List of texts to translate
:param target_language: Target language code
:return: List of translated texts
"""
# headers = {
# "Authorization": f"Bearer {GROQ_API_KEY}",
# "Content-Type": "application/json"
# }
translated_texts = []
# for text in texts:
# payload = {
# "source_language": "en", # تأكد من ضبط لغة المصدر
# "target_language": target_language,
# "text": text
# }
# response = requests.post(GROQ_API_URL, json=payload, headers=headers)
# if response.status_code == 200:
# translated_text = response.json().get("translated_text")
# translated_texts.append(translated_text)
# else:
# # في حالة حدوث خطأ، يمكنك التعامل معه هنا
# translated_texts.append(text) # احتفظ بالنص الأصلي في حال فشل الترجمة
chat_completion = client.chat.completions.create(
messages=[
{
"role": "user",
"content": f"Translate the text fllowing into Arabic language:{texts}",
}
],
model="llama3-8b-8192",
)
translated_texts.append(chat_completion.choices[0].message.content)
return translated_texts
def translate_content(content, target_language):
"""
Translates the content of a markdown file, preserving non-translatable parts.
:param content: Markdown content
:param target_language: Target language code
:return: Translated content
"""
translatable_texts = extract_translatable_text(content)
translated_texts = translate_texts_groq(translatable_texts, target_language)
# Reconstruct the content with translated texts
for original, translated in zip(translatable_texts, translated_texts):
content = content.replace(original, translated)
return content
# from transformers import MarianMTModel, MarianTokenizer
# # Initialize the model and tokenizer
# model_name = 'Helsinki-NLP/opus-mt-en-<target_language>' # Replace <target_language> with the target language code
# model = MarianMTModel.from_pretrained(model_name)
# tokenizer = MarianTokenizer.from_pretrained(model_name)
# def translate_texts(texts, target_language):
# """
# Translates a list of texts to the target language.
# :param texts: List of texts to translate
# :param target_language: Target language code
# :return: List of translated texts
# """
# translated_texts = []
# for text in texts:
# inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True)
# translated = model.generate(**inputs)
# translated_text = tokenizer.decode(translated[0], skip_special_tokens=True)
# translated_texts.append(translated_text)
# return translated_texts
# def translate_content(content, target_language):
# """
# Translates the content of a markdown file, preserving non-translatable parts.
# :param content: Markdown content
# :param target_language: Target language code
# :return: Translated content
# """
# translatable_texts = extract_translatable_text(content)
# translated_texts = translate_texts(translatable_texts, target_language)
# # Reconstruct the content with translated texts
# for original, translated in zip(translatable_texts, translated_texts):
# content = content.replace(original, translated)
# return content