generate audio minor refactor 2

app.py

@@ -172,22 +172,22 @@ def generate_audio(prompt: str, duration: int) -> gr.Component:
     Returns:
         gr.Component: The generated audio as a Gradio Audio component.
     """
-    
+
     DURATION_LIMIT = 30
-    
+    duration = duration if duration < DURATION_LIMIT else DURATION_LIMIT
+
     client = Tool.from_space(
         space_id="luke9705/MusicGen_custom",
         token=os.environ.get('HF_TOKEN'),
         name="Sound_Generator",
         description="Generate music or sound effects from a text prompt using MusicGen."
     )
-    
-    duration = duration if duration < DURATION_LIMIT else DURATION_LIMIT
 
     sound = client(prompt, duration)
 
     return gr.Audio(value=sound)
 
+
 @tool
 def generate_audio_from_sample(prompt: str, duration: int, sample_path: str = None) -> gr.Component:
     """
@@ -196,21 +196,22 @@ def generate_audio_from_sample(prompt: str, duration: int, sample_path: str = No
         prompt: The text prompt to generate the audio from.
         duration: Duration of the generated audio in seconds. Max 30 seconds.
         sample_path: audio sample path to guide generation.
-
     Returns:
         gr.Component: The generated audio as a Gradio Audio component.
     """
+
+    DURATION_LIMIT = 30
+    duration = duration if duration < DURATION_LIMIT else DURATION_LIMIT
+    
     client = Tool.from_space(
         space_id="luke9705/MusicGen_custom",
         token=os.environ.get('HF_TOKEN'),
         name="Sound_Generator",
         description="Generate music or sound effects from a text prompt using MusicGen."
     )
-    if duration > 30:
-        sound = client(prompt, 30, sample_path)
-    else:
-        sound = client(prompt, duration, sample_path)
     
+    sound = client(prompt, duration, sample_path)
+
     return gr.Audio(value=sound)
 
 @tool