Update app.py

app.py

@@ -51,9 +51,7 @@ print(f"Reading {demo_script_path} to apply environment-specific modifications..
 try:
     modified_content = demo_script_path.read_text()
 
-    # --- Patch Definitions ---
-
-    # Define the original model loading block to be replaced.
+    # Define the original model loading block using a list of lines for robustness.
     original_model_lines = [
         '        self.model = VibeVoiceForConditionalGenerationInference.from_pretrained(',
         '            self.model_path,',
@@ -63,19 +61,10 @@ try:
         '        )'
     ]
     original_model_block = "\n".join(original_model_lines)
-
-    # Define the generation method signature to add the GPU decorator to.
-    original_method_lines = [
-        '    def generate_podcast_streaming(self, ',
-        '                                 num_speakers: int,',
-        '                                 script: str,',
-        '                                 speaker_1: str = None,',
-        '                                 speaker_2: str = None,',
-        '                                 speaker_3: str = None,',
-        '                                 speaker_4: str = None,',
-        '                                 cfg_scale: float = 1.3) -> Iterator[tuple]:'
-    ]
-    original_method_signature = "\n".join(original_method_lines)
+    
+    # More robustly define the generation method signature to patch.
+    # We only need the first line to find our target.
+    original_method_signature = "    def generate_podcast_streaming(self,"
 
     if USE_ZEROGPU:
         print("Optimizing for ZeroGPU execution...")
@@ -94,8 +83,8 @@ try:
         ]
         replacement_model_block_gpu = "\n".join(replacement_model_lines_gpu)
         
-        # Add the @spaces.GPU decorator to the generation method instead of the class.
-        replacement_method_signature_gpu = "@spaces.GPU(duration=120)\n" + original_method_signature
+        # Add the @spaces.GPU decorator *with correct indentation* before the method.
+        replacement_method_signature_gpu = "    @spaces.GPU(duration=120)\n" + original_method_signature
 
         # --- Apply Patches for GPU ---
 
@@ -104,14 +93,16 @@ try:
             modified_content = modified_content.replace(original_method_signature, replacement_method_signature_gpu)
             print("Successfully applied GPU decorator to the generation method.")
         else:
-            print("\033[91mWarning: Could not find the generation method signature to apply the GPU decorator.\033[0m")
+            print("\033[91mError: Could not find the generation method signature to apply the GPU decorator.\033[0m")
+            sys.exit(1)
 
         # Patch 2: Modify the model loading
         if original_model_block in modified_content:
             modified_content = modified_content.replace(original_model_block, replacement_model_block_gpu)
             print("Successfully patched the model loading block for ZeroGPU.")
         else:
-            print("\033[91mWarning: The original model loading block was not found. Patching may have failed.\033[0m")
+            print("\033[91mError: The original model loading block was not found. Patching may have failed.\033[0m")
+            sys.exit(1)
 
     else: # Pure CPU execution
         print("Modifying for pure CPU execution...")
@@ -131,8 +122,8 @@ try:
             modified_content = modified_content.replace(original_model_block, replacement_model_block_cpu)
             print("Script modified for CPU successfully.")
         else:
-            print("\033[91mWarning: The original model loading block was not found. Patching may have failed.\033[0m")
-
+            print("\033[91mError: The original model loading block was not found. Patching may have failed.\03-3[0m")
+            sys.exit(1)
 
     # Write the dynamically modified content back to the demo file
     demo_script_path.write_text(modified_content)
@@ -142,7 +133,7 @@ except Exception as e:
     sys.exit(1)
 
 # --- 4. Launch the Gradio Demo ---
-model_id = "microsoft/V_VibeVoice-1.5B"
+model_id = "microsoft/VibeVoice-1.5B"
 
 # Construct the command to run the modified demo script
 command = [