feat: add ai translations as a translation provider

edxval/migrations/0004_add_edx_ai_translations_provider.py

@@ -0,0 +1,18 @@
+# Generated by Django 4.2.16 on 2024-11-27 17:14
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('edxval', '0003_delete_transcriptcredentials'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='videotranscript',
+            name='provider',
+            field=models.CharField(choices=[('Custom', 'Custom'), ('3PlayMedia', '3PlayMedia'), ('Cielo24', 'Cielo24'), ('edx_ai_translations', 'edx_ai_translations')], default='Custom', max_length=30),
+        ),
+    ]

edxval/models.py

@@ -378,13 +378,20 @@ class for providing tuple choices.
     CUSTOM = 'Custom'
     THREE_PLAY_MEDIA = '3PlayMedia'
     CIELO24 = 'Cielo24'
+    EDX_AI_TRANSLATIONS = 'edx_ai_translations'
 
     CHOICES = (
         (CUSTOM, CUSTOM),
         (THREE_PLAY_MEDIA, THREE_PLAY_MEDIA),
         (CIELO24, CIELO24),
     )
 
+    # Choices specififcally for only the VideoTranscript model
+    TRANSCRIPT_MODEL_CHOICES = (
+        *CHOICES,
+        (EDX_AI_TRANSLATIONS, EDX_AI_TRANSLATIONS),
+    )
+
 
 class CustomizableFileField(models.FileField):
     """
@@ -425,7 +432,7 @@ class VideoTranscript(TimeStampedModel):
     language_code = models.CharField(max_length=50, db_index=True)
     provider = models.CharField(
         max_length=30,
-        choices=TranscriptProviderType.CHOICES,
+        choices=TranscriptProviderType.TRANSCRIPT_MODEL_CHOICES,
         default=TranscriptProviderType.CUSTOM,
     )
     file_format = models.CharField(max_length=20, db_index=True, choices=TranscriptFormat.CHOICES)

edxval/tests/test_views.py

@@ -880,7 +880,7 @@ def test_update_existing_transcript(self):
                 'file_format': TranscriptFormat.SRT
             },
             'message': '"xyz" provider is not supported. Supported transcription providers are "{}"'.format(
-                sorted(dict(TranscriptProviderType.CHOICES).keys())
+                sorted(dict(TranscriptProviderType.TRANSCRIPT_MODEL_CHOICES).keys())
             )
         },
     )

edxval/views.py

@@ -151,7 +151,7 @@ def post(self, request):
             ).format(format=file_format, supported_formats=supported_formats)
             return Response(status=status.HTTP_400_BAD_REQUEST, data={'message': message})
 
-        supported_providers = sorted(dict(TranscriptProviderType.CHOICES).keys())
+        supported_providers = sorted(dict(TranscriptProviderType.TRANSCRIPT_MODEL_CHOICES).keys())
         if provider not in supported_providers:
             message = (
                 '"{provider}" provider is not supported. Supported transcription providers are "{supported_providers}"'