Added option to configure the length of each section

Author: leopiney

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "neuralnoise"
-version = "0.7.0"
+version = "0.9.0"
 description = "An AI-powered podcast studio that uses multiple AI agents working together."
 authors = [
     { name = "Leonardo Piñeyro", email = "[email protected]" }

src/neuralnoise/prompts/script_generation.system.xml

@@ -41,7 +41,7 @@
     will..."
     - Avoid formal introductions or conclusions for sections/segments.
     - Allow speakers to ask and answer questions naturally.
-    - Create 4-10 segments per section, with a mix of short and long segments.
+    - Create ${min_segments}-${max_segments} segments per section, with a mix of short and long segments.
     - Include some very short segments (1-2 words) for quick interactions and expressing emotions or
     reactions. For example: "Yeah.", "Right?", "So cool."
     - Use pauses (blank_duration) where appropriate.

src/neuralnoise/studio/agents.py

@@ -93,7 +93,10 @@ def script_generator_agent(self) -> AssistantAgent:
         agent = AssistantAgent(
             name="ScriptGeneratorAgent",
             system_message=self.load_prompt(
-                "script_generation.system", language=self.language
+                "script_generation.system",
+                language=self.language,
+                min_segments=str(self.config.show.min_segments),
+                max_segments=str(self.config.show.max_segments),
             ),
             llm_config={"config_list": [self.llm_json_mode_config]},
         )

src/neuralnoise/types.py

@@ -53,6 +53,9 @@ class Show(BaseModelDisplay):
     about: str
     language: str
 
+    min_segments: int = 4
+    max_segments: int = 10
+
 
 class StudioConfig(BaseModelDisplay):
     show: Show