Update utils.py

utils.py

@@ -1,12 +1,9 @@
-import os
 from groq import Groq
-from gtts import gTTS
 from pydantic import BaseModel, ValidationError
 from typing import List, Literal
-import tiktoken
+import os
 
-groq_client = Groq()
-tokenizer = tiktoken.get_encoding("cl100k_base")
+groq_client = Groq(api_key=os.environ["GROQ_API_KEY"])
 
 class DialogueItem(BaseModel):
     speaker: Literal["Host", "Guest"]
@@ -15,12 +12,6 @@ class DialogueItem(BaseModel):
 class Dialogue(BaseModel):
     dialogue: List[DialogueItem]
 
-def truncate_text(text, max_tokens=2048):
-    tokens = tokenizer.encode(text)
-    if len(tokens) > max_tokens:
-        return tokenizer.decode(tokens[:max_tokens])
-    return text
-
 def generate_script(system_prompt: str, input_text: str, tone: str):
     input_text = truncate_text(input_text)
     prompt = f"{system_prompt}\nTONE: {tone}\nINPUT TEXT: {input_text}"
@@ -29,7 +20,7 @@ def generate_script(system_prompt: str, input_text: str, tone: str):
         messages=[
             {"role": "system", "content": prompt},
         ],
-        model="llama2-70b-4096",
+        model="llama-3.1-70b-versatile",  # Updated to the correct model name
         max_tokens=2048,
         temperature=0.7
     )
@@ -41,8 +32,4 @@ def generate_script(system_prompt: str, input_text: str, tone: str):
     
     return dialogue
 
-def generate_audio(text: str, speaker: str) -> str:
-    tts = gTTS(text, lang='en', tld='com' if speaker == "Host" else 'co.uk')
-    filename = f"{speaker.lower()}_audio.mp3"
-    tts.save(filename)
-    return filename
+# Make sure the truncate_text function is defined here or imported if it's in another file