Update app.py

app.py

@@ -54,7 +54,7 @@ def reset_model():
         print(f"Failed to reinitialize model: {e}")
         return False
 
-def generate_speech(inp_audio, inp_text, infer_timestep, p_w, t_w):
+def generate_speech(inp_audio, inp_text, infer_timestep, p_w, t_w, speed_factor):
     if not inp_audio or not inp_text:
         gr.Warning("Please provide both reference audio and text to generate.")
         return None
@@ -82,6 +82,11 @@ def generate_speech(inp_audio, inp_text, infer_timestep, p_w, t_w):
             with torch.no_grad():  # Use no_grad for inference
                 resource_context = infer_pipe.preprocess(file_content)
                 wav_bytes = infer_pipe.forward(resource_context, inp_text, time_step=infer_timestep, p_w=p_w, t_w=t_w)
+            
+            # Apply speed adjustment if needed
+            if speed_factor != 1.0:
+                wav_bytes = adjust_speed(wav_bytes, speed_factor)
+            
             # Clean up memory after successful generation
             cleanup_memory()
             return wav_bytes
@@ -101,6 +106,43 @@ def generate_speech(inp_audio, inp_text, infer_timestep, p_w, t_w):
         cleanup_memory()
         return None
 
+def adjust_speed(wav_bytes, speed_factor):
+    """Adjust the speed of the audio without changing pitch"""
+    try:
+        # Create temp file
+        temp_input = "temp_input.wav"
+        temp_output = "temp_output.wav"
+        
+        with open(temp_input, "wb") as f:
+            f.write(wav_bytes)
+        
+        # Load audio
+        audio = AudioSegment.from_file(temp_input)
+        
+        # Apply speed change
+        if speed_factor != 1.0:
+            # Manually adjust frame rate to change speed without pitch alteration
+            new_frame_rate = int(audio.frame_rate * speed_factor)
+            audio = audio._spawn(audio.raw_data, overrides={
+                "frame_rate": new_frame_rate
+            }).set_frame_rate(audio.frame_rate)
+        
+        # Export result
+        audio.export(temp_output, format="wav")
+        
+        # Read and return
+        with open(temp_output, "rb") as f:
+            result = f.read()
+        
+        # Clean up temp files
+        os.remove(temp_input)
+        os.remove(temp_output)
+        
+        return result
+    except Exception as e:
+        print(f"Speed adjustment failed: {e}")
+        return wav_bytes  # Return original if adjustment fails
+
 def cleanup_memory():
     """Clean up system memory."""
     gc.collect()
@@ -199,6 +241,14 @@ with gr.Blocks(title="MegaTTS3 Voice Cloning") as demo:
                     maximum=10.0,
                     step=0.1
                 )
+                speed_factor = gr.Slider(
+                    label="Speed Adjustment",
+                    value=1.0,
+                    minimum=0.5,
+                    maximum=2.0,
+                    step=0.1,
+                    info="1.0 = normal speed, <1.0 = slower, >1.0 = faster"
+                )
             
             generate_btn = gr.Button("Generate Speech", variant="primary")
         
@@ -207,7 +257,7 @@ with gr.Blocks(title="MegaTTS3 Voice Cloning") as demo:
     
     generate_btn.click(
         fn=generate_speech,
-        inputs=[reference_audio, text_input, infer_timestep, p_w, t_w],
+        inputs=[reference_audio, text_input, infer_timestep, p_w, t_w, speed_factor],
         outputs=[output_audio]
     )