Update src/agents/agent_transcript/func.py

src/agents/agent_transcript/func.py

@@ -145,6 +145,9 @@ def script_writer_single(state: State):
     # Add word count guidance to the prompt
     if current_index == 0:
         word_prompt = f"Hãy viết script đầu tiên với khoảng {words_per_script} từ."
+    elif current_index == script_count - 1:
+        word_prompt = f"ok, viết cho tôi phần tiếp theo và cũng là phần cuối cùng, bám sát cấu trúc, khoảng {words_per_script} từ cho script này. Ở cuối phần này, hãy kêu gọi người xem like, subscribe và comment để ủng hộ video để có động lực ra video mới. Đừng quên đối tượng khán giả là người Mỹ,giới tính nữ, trên 20 tuổi, bắt đầu, trình bày thành dạng câu văn liền mạch, dùng để làm văn nói cho video YouTube, không dùng icon"
+
     else:
         word_prompt = f"ok, viết cho tôi phần tiếp theo, bám sát cấu trúc, khoảng {words_per_script} từ cho script này, các công thức tạo cảm xúc và đừng quên đối tượng khán giả là người Mỹ,giới tính nữ, trên 20 tuổi, bắt đầu, trình bày thành dạng câu văn liền mạch, dùng để làm văn nói cho video YouTube, không dùng icon"