Update app.py

app.py

@@ -7,70 +7,23 @@ def process_audio(file, model, prompt, start, length):
     # Add your audio processing code here
     return processed_audio_path
 
-
-    with gr.Blocks() as demo:
-        gr.Markdown(
-            """
-            # Drums Generation in Different Models
-            Upload your audio file to process with AudioLDM2 or StableAudio
-            """
-        )
-
-with gr.Row():
-    with gr.Column():
-        audio_file = gr.Audio(type="numpy", label="Choose an audio file")
-        model_choice = gr.Radio(choices=['AudioLDM2', 'StableAudio'], label='Choose a model for processing')
-        prompt_text = gr.Textbox(label='Enter a prompt for the audio processing')
-        start_point = gr.Slider(0.0, 60.0, 0.0, step=1.0, label='Choose a starting point for the audio (seconds)')
-        output_length = gr.Slider(1.0, 60.0, 10.0, step=1.0, label='Choose the length of the output audio (seconds)')
-        gr.Markdown('**Note:** Longer audio takes more time to generate.')
-
-        process_button = gr.Button("Process Audio")
-        output_audio = gr.Audio(label="Processed Audio")
-        download_button = gr.File(label="Download Processed Audio")
-
-        def on_process(audio_file, model_choice, prompt_text, start_point, output_length):
-            processed_audio = process_audio(audio_file, model_choice, prompt_text, start_point, output_length)
-            return processed_audio, processed_audio
-
-        process_button.click(on_process, inputs=[audio_file, model_choice, prompt_text, start_point, output_length], outputs=[output_audio, download_button])
-
-gr.Markdown("---")
-gr.Markdown("## Examples")
-
-exmples_prompts = ["wiwi", "pipi", "wifi"]
-example_titles = ["Original Audio", "Separated Audio", "AudioLDM2", "StableAudio"]
-
-for prompt in exmples_prompts:
-    with gr.Group():
-        gr.Markdown(f"**Prompt: {prompt}**")
-        with gr.Row():
-            for title in example_titles:
-                with gr.Column():
-                    gr.Markdown(f"**{title}**")
-                    gr.Audio("goodres.wav", label=f"{title} Audio")
-
-gr.Markdown("---")
-gr.Markdown("## Useful Links")
-links = [
-    {"url": "https://example.com", "type": "Regular Website", "text": "This is a regular website"},
-    {"url": "https://github.com/example", "type": "GitHub", "text": "This is a GitHub repository"},
-    {"url": "https://colab.research.google.com", "type": "Google Colab", "text": "This is a Google Colab link"}
-]
-logo_urls = {
-    "GitHub": "https://github.githubassets.com/images/modules/logos_page/GitHub-Mark.png",
-    "Google Colab": "https://colab.research.google.com/img/colab_favicon_256px.png"
-}
-for link in links:
-    if link["type"] in logo_urls:
-        gr.HTML(f'<div style="display: flex; align-items: center;"><img src="{logo_urls[link["type"]]}" alt="{link["type"]} logo" width="24" style="margin-right: 10px;"><a href="{link["url"]}" target="_blank">{link["text"]}</a> - <i>{link["type"]}</i></div>')
-    else:
-        gr.Markdown(f"[{link['text']}]({link['url']}) - *{link['type']}*")
-
-gr.Markdown("""
-<div style="text-align: center; padding: 20px; background: rgba(0, 0, 0, 0.7); color: white; border-top: 1px solid #ddd; margin-top: 30px;">
-    Made with ❤️ using Gradio
-</div>
-""")
-
+demo = gr.Interface(
+    fn=process_audio,
+    inputs=[
+        gr.Audio(type="file", label="Upload Audio File"),
+        gr.Radio(choices=['AudioLDM2', 'StableAudio'], label='Choose a Model for Processing'),
+        gr.Textbox(label="Prompt", placeholder="Enter your text prompt here"),
+        gr.Slider(0.0, 60.0, value=0.0, step=1.0, label="Starting Point (seconds)"),
+        gr.Slider(1.0, 60.0, value=10.0, step=1.0, label="Output Length (seconds)", info="Longer audio takes more time to generate")
+    ],
+    outputs=gr.Audio(type="file", label="Processed Audio"),
+    title="Drums Generation in Different Models",
+    description="Upload your audio file and process it with AudioLDM2 or StableAudio based on your prompt and settings.",
+    examples=[
+        ["example.wav", "AudioLDM2", "Generate a rock beat", 0.0, 10.0],
+        ["example.wav", "StableAudio", "Create a serene soundscape", 5.0, 15.0],
+        ["example.wav", "AudioLDM2", "Simulate a forest ambiance", 10.0, 20.0],
+        ["example.wav", "StableAudio", "Recreate a gentle rainfall", 0.0, 25.0]
+    ]
+)
 demo.launch()