Apply suggestions from code review

Co-authored-by: Rithvik Nishad <[email protected]>
ohcnetwork · Feb 11, 2024 · 4c07449 · 4c07449
1 parent ecb6272
commit 4c07449
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/ayushma/models/enums.py b/ayushma/models/enums.py
@@ -20,8 +20,8 @@ class STTEngine(IntegerChoices):
 
 
 class TTSEngine(IntegerChoices):
-    OPENAI = 1
-    GOOGLE = 2
+    OPENAI = (1, "openai")
+    GOOGLE = (2, "google")
 
 
 class FeedBackRating(IntegerChoices):

diff --git a/ayushma/utils/language_helpers.py b/ayushma/utils/language_helpers.py
@@ -47,7 +47,7 @@ def text_to_speech(text, language_code, service):
 
         text = sanitize_text(text)
 
-        if service == "google":
+        if service == TTSEngine.GOOGLE:
             client = texttospeech.TextToSpeechClient()
 
             synthesis_input = texttospeech.SynthesisInput(text=text)
@@ -66,7 +66,7 @@ def text_to_speech(text, language_code, service):
             )
 
             return response.audio_content
-        elif service == "openai":
+        elif service == TTSEngine.OPENAI:
             client = OpenAI(api_key=settings.OPENAI_API_KEY)
             response = client.audio.speech.create(
                 model="tts-1-hd",