Create agent_02_screenwriter.py

agent_02_screenwriter.py

@@ -0,0 +1,42 @@
+from crewai import Agent, Task
+
+# MODULE: AGENT 02 - THE SCREENWRITER
+# Responsibility: Turn the brief into a concrete script.
+# Address: Root Folder / agent_02_screenwriter.py
+
+class ScreenwriterUnit:
+    def __init__(self):
+        # This path is where the Librarian puts writing guides
+        self.knowledge_path = "/data/knowledge/writer"
+
+    def create_agent(self):
+        return Agent(
+            role='Lead Screenwriter',
+            goal='Write a high-retention script with clear visual cues.',
+            backstory="""You are a Hollywood screenwriter specializing in short-form content.
+            You focus on the first 3 seconds (The Hook).
+            You write in a dual-column format (Audio vs Visual).
+            You strictly follow the Creative Brief provided by the Strategist.""",
+            verbose=True,
+            allow_delegation=False,
+            llm='ollama/llama3:70b'
+        )
+
+    def create_task(self, creative_brief):
+        return Task(
+            description=f"""
+            Using the provided Creative Brief, write a 60-second Video Script.
+            
+            Format Requirements:
+            - SECTION 1: The Hook (0-3 seconds) - Must be visually shocking.
+            - SECTION 2: The Retension (3-15 seconds) - Open a loop.
+            - SECTION 3: The Payoff (15-50 seconds) - Deliver value.
+            - SECTION 4: The CTA (50-60 seconds).
+            
+            Output strictly in this format:
+            [TIME] | [VISUAL DESCRIPTION] | [AUDIO/DIALOGUE]
+            """,
+            agent=self.create_agent(),
+            context=[creative_brief], # This specifically links it to the previous task's output
+            expected_output="A dual-column video script with timestamps."
+        )