Create utils.py

utils.py

@@ -0,0 +1,48 @@
+import os
+from groq import Groq
+from gtts import gTTS
+from pydantic import BaseModel, ValidationError
+from typing import List, Literal
+import tiktoken
+
+groq_client = Groq()
+tokenizer = tiktoken.get_encoding("cl100k_base")
+
+class DialogueItem(BaseModel):
+    speaker: Literal["Host", "Guest"]
+    text: str
+
+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}"
+    
+    response = groq_client.chat.completions.create(
+        messages=[
+            {"role": "system", "content": prompt},
+        ],
+        model="llama2-70b-4096",
+        max_tokens=2048,
+        temperature=0.7
+    )
+    
+    try:
+        dialogue = Dialogue.model_validate_json(response.choices[0].message.content)
+    except ValidationError as e:
+        raise ValueError(f"Failed to parse dialogue JSON: {e}")
+    
+    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