ismail/haikal
1
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch 'main' of http://10.10.1.120:3000/tenhal_admin/haikal

Browse Source
This commit is contained in:
gitea 2025-02-18 13:32:56 +00:00
commit d937f9f684
2 changed files with 37 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
inventory/management/commands/__pycache__/translate.cpython-311.pyc
View File

Binary file not shown.

69
inventory/management/commands/translate.py
View File

@ -1,43 +1,48 @@
from openai import OpenAI
from django.core.management.base import BaseCommand
from inventory.models import CarSerie, CarModel, CarMake, CarTrim, CarOption, CarSpecification
from inventory.models import CarModel
from django.conf import settings
class Command(BaseCommand):
help = 'Translates to Arabic and saves them in arabic_name field.'
help = 'Translates car model names to Arabic and saves them in the arabic_name field.'
def handle(self, *args, **kwargs):
client = OpenAI(api_key=settings.OPENAI_API_KEY)
en_value = CarModel.objects.all()
car_models = CarModel.objects.all()
total = en_value.count()
total = car_models.count()
print(f'Translating {total} names...')
for index, en_value in enumerate(en_value, start=1):
if not en_value.arabic_name:
try:
completion = client.chat.completions.create(
model="gpt-4o",
messages=[
{
"role": "system",
"content": (
"You are an assistant that translates English to Arabic."
"You are an assistant specialized in cars and automotive terms."
"If the model name is a number just write it as is"
"You can get the arabic names for makes, models, series, trims, options, and specifications."
)
},
{
"role": "user",
"content": en_value.name
}
],
temperature=0.2,
)
translation = completion.choices[0].message.content.strip()
en_value.arabic_name = translation
en_value.save()
print(f"[{index}/{total}] .. Done")
except Exception as e:
print(f"Error translating '{en_value.name}': {e}")
for index, car_model in enumerate(car_models, start=1):
if not car_model.arabic_name or car_model.arabic_name == '-':
if isinstance(car_model.name, int):
car_model.arabic_name = car_model.name
car_model.save()
print(f"[{index}/{total}] .. Skipped GPT (Numeric)")
else:
try:
completion = client.chat.completions.create(
model="gpt-4o",
messages=[
{
"role": "system",
"content": (
"You are an assistant that translates English car names to Arabic."
"If the name is purely numeric, keep it as is."
"For mixed names like 'D9', translate them as 'دي 9'."
)
},
{
"role": "user",
"content": car_model.name
}
],
temperature=0.2,
)
translation = completion.choices[0].message.content.strip()
car_model.arabic_name = translation
car_model.save()
print(f"[{index}/{total}] .. Done")
except Exception as e:
print(f"Error translating '{car_model.name}': {e}")