Update utils.py

utils.py

@@ -24,9 +24,25 @@ def truncate_text(text, max_tokens=2048):
         return tokenizer.decode(tokens[:max_tokens])
     return text
 
-def generate_script(system_prompt: str, input_text: str, tone: str):
+def generate_script(system_prompt: str, input_text: str, tone: str, target_length: str):
     input_text = truncate_text(input_text)
-    prompt = f"{system_prompt}\nTONE: {tone}\nINPUT TEXT: {input_text}"
+    word_limit = 300 if target_length == "Short (1-2 min)" else 750  # Assuming 150 words per minute
+    
+    prompt = f"""
+    {system_prompt}
+    TONE: {tone}
+    TARGET LENGTH: {target_length} (approximately {word_limit} words)
+    INPUT TEXT: {input_text}
+
+    Generate a complete, well-structured podcast script that:
+    1. Starts with a proper introduction
+    2. Covers the main points from the input text
+    3. Has a natural flow of conversation between Maria and Sarah
+    4. Concludes with a summary and sign-off
+    5. Fits within the {word_limit} word limit for the target length of {target_length}
+
+    Ensure the script is not abruptly cut off and forms a complete conversation.
+    """
     
     response = groq_client.chat.completions.create(
         messages=[