Update app.py

app.py

@@ -67,13 +67,16 @@ def generate_sfx(text, duration=30):
     
     # Calculate total samples for the specified duration
     total_samples = duration * 22050  # Samples for 30 seconds
+    
     # Generate audio samples
     generated_samples = model.predict(input_data)
     
-    # Check the length of generated samples and ensure it meets the required duration
-    if len(generated_samples) < total_samples:
-        raise ValueError(f"Generated audio is shorter than {duration} seconds.")
-
+    # Ensure generated samples meet the required duration
+    if generated_samples.shape[1] < total_samples:
+        # Pad with zeros if not enough audio is generated
+        padding = np.zeros((1, total_samples - generated_samples.shape[1]))
+        generated_samples = np.concatenate([generated_samples, padding], axis=1)
+    
     # Convert the prediction to audio data
     audio_data = convert_to_audio(generated_samples)
 
@@ -83,12 +86,12 @@ def generate_sfx(text, duration=30):
 
     return output_filename
 
-# Define the Gradio interface
+# Define the Gradio interface with updated slider settings
 interface = gr.Interface(
     fn=generate_sfx,
     inputs=[
         gr.Textbox(label="Enter a Word", placeholder="Write a Word To Convert it into SFX Sound"),
-        gr.Slider(label="Duration (seconds)", minimum=30, maximum=120, value=30)  # Set duration options
+        gr.Slider(label="Duration (seconds)", minimum=2, maximum=20, value=5)  # Set duration options
     ],
     outputs=gr.Audio(label="Generated SFX", type="filepath"),
     live=False,