Update app.py

app.py

@@ -222,19 +222,33 @@ class VibeVoiceDemo:
 
 
 
+    @staticmethod
+    def _infer_num_speakers_from_script(script: str) -> int:
+        """
+        Infer number of speakers by counting distinct 'Speaker X:' tags in the script.
+        Robust to 0- or 1-indexed labels and repeated turns.
+        Falls back to 1 if none found.
+        """
+        import re
+        ids = re.findall(r'(?mi)^\s*Speaker\s+(\d+)\s*:', script)
+        return len({int(x) for x in ids}) if ids else 1
+
     def load_example_scripts(self):
         examples_dir = os.path.join(os.path.dirname(__file__), "text_examples")
         self.example_scripts = []
         if not os.path.exists(examples_dir):
             return
-        txt_files = sorted([f for f in os.listdir(examples_dir)
-                            if f.lower().endswith('.txt')])
+
+        txt_files = sorted(
+            [f for f in os.listdir(examples_dir) if f.lower().endswith('.txt')]
+        )
         for txt_file in txt_files:
             try:
                 with open(os.path.join(examples_dir, txt_file), 'r', encoding='utf-8') as f:
                     script_content = f.read().strip()
                 if script_content:
-                    self.example_scripts.append([1, script_content])
+                    num_speakers = self._infer_num_speakers_from_script(script_content)
+                    self.example_scripts.append([num_speakers, script_content])
             except Exception as e:
                 print(f"Error loading {txt_file}: {e}")