Update app.py

app.py

@@ -9,8 +9,82 @@ SUNO_KEY = os.environ.get("SunoKey", "")
 if not SUNO_KEY:
     print("⚠️ SunoKey not set!")
 
-def separate_vocals(task_id, audio_id, separation_type):
-    """Separate vocals and instruments from Suno tracks - semi working, you need some knowledge and api backroom access."""
+def get_audio_info(task_id):
+    """Get audio information from Suno task ID"""
+    if not SUNO_KEY:
+        return "❌ Error: SunoKey not configured in environment variables", []
+    
+    if not task_id.strip():
+        return "❌ Error: Please enter Task ID", []
+    
+    try:
+        resp = requests.get(
+            "https://api.sunoapi.org/api/v1/generate/record-info",
+            headers={"Authorization": f"Bearer {SUNO_KEY}"},
+            params={"taskId": task_id.strip()},
+            timeout=30
+        )
+        
+        if resp.status_code != 200:
+            return f"❌ Request failed: HTTP {resp.status_code}", []
+        
+        data = resp.json()
+        if data.get("code") != 200:
+            return f"❌ API error: {data.get('msg', 'Unknown')}", []
+        
+        # Check if task is complete
+        status = data["data"].get("status", "PENDING")
+        if status != "SUCCESS":
+            return f"⏳ Task status: {status}. Please wait for generation to complete.", []
+        
+        # Parse sunoData to get audio options
+        suno_data = data["data"]["response"].get("sunoData", [])
+        if not suno_data:
+            return "❌ No audio data found in response", []
+        
+        # Create audio options for dropdown
+        audio_options = []
+        for i, audio in enumerate(suno_data):
+            audio_id = audio.get("id", "")
+            prompt = audio.get("prompt", "No prompt")
+            title = audio.get("title", "Untitled")
+            duration = audio.get("duration", 0)
+            
+            # Truncate long prompts for display
+            display_prompt = prompt[:50] + "..." if len(prompt) > 50 else prompt
+            
+            # Create display text and store full data
+            display_text = f"Track {i+1}: {title} ({duration:.1f}s) - {display_prompt}"
+            audio_options.append((display_text, audio_id, audio))
+        
+        if not audio_options:
+            return "❌ No valid audio tracks found", []
+        
+        # Format output message
+        output = f"✅ **Task Found!**\n"
+        output += f"**Task ID:** `{task_id}`\n"
+        output += f"**Status:** {status}\n"
+        output += f"**Found {len(audio_options)} audio track(s):**\n\n"
+        
+        for i, (display_text, audio_id, audio) in enumerate(audio_options):
+            output += f"**Track {i+1}:**\n"
+            output += f"- **ID:** `{audio_id}`\n"
+            output += f"- **Title:** {audio.get('title', 'Untitled')}\n"
+            output += f"- **Prompt:** {audio.get('prompt', 'No prompt')[:100]}...\n"
+            output += f"- **Duration:** {audio.get('duration', 0):.1f}s\n"
+            if audio.get('audioUrl'):
+                output += f"- **Audio:** [Listen]({audio.get('audioUrl')})\n"
+            output += "\n"
+        
+        output += "👇 **Select a track from the dropdown below to separate stems**"
+        
+        return output, audio_options
+        
+    except Exception as e:
+        return f"❌ Error: {str(e)}", []
+
+def separate_vocals(task_id, audio_id, separation_type, audio_data=None):
+    """Separate vocals and instruments from Suno tracks"""
     if not SUNO_KEY:
         yield "❌ Error: SunoKey not configured in environment variables"
         return
@@ -146,17 +220,28 @@ with gr.Blocks(title="Suno Stem Separator", theme="soft") as app:
     
     with gr.Row():
         with gr.Column(scale=1):
-            task_id = gr.Textbox(
-                label="Original Task ID",
+            # Step 1: Enter Task ID
+            task_id_input = gr.Textbox(
+                label="Task ID",
                 placeholder="Example: 5c79****be8e",
-                info="The task ID from your Suno music generation"
+                info="Enter your Suno generation task ID",
+                elem_id="task_id_input"
             )
-            audio_id = gr.Textbox(
-                label="Audio ID",
-                placeholder="Example: e231****-****-****-****-****8cadc7dc",
-                info="The specific audio ID to separate"
+            
+            get_audio_btn = gr.Button("🔍 Find Audio Tracks", variant="secondary")
+            
+            # Will be updated after getting audio info
+            audio_dropdown = gr.Dropdown(
+                label="Select Audio Track to Separate",
+                choices=[],
+                info="Choose which audio track to process",
+                visible=False,
+                interactive=True
             )
             
+            # Hidden store for audio data
+            audio_data_store = gr.State(value=None)
+            
             separation_type = gr.Radio(
                 label="Separation Type",
                 choices=[
@@ -164,37 +249,38 @@ with gr.Blocks(title="Suno Stem Separator", theme="soft") as app:
                     ("split_stem (12 stems - 5 credits)", "split_stem")
                 ],
                 value="separate_vocal",
-                info="Choose separation mode"
+                info="Choose separation mode",
+                visible=False
             )
             
-            btn = gr.Button("🎵 Separate Stems", variant="primary", scale=1)
+            separate_btn = gr.Button("🎵 Separate Stems", variant="primary", visible=False)
             
-            gr.Markdown("""
-            **How it works:**
-            1. Enter Task ID and Audio ID from your Suno track
-            2. Choose separation type
-            3. Click Separate Stems
-            4. Wait 30-120 seconds
-            5. Get download links for each stem
+            gr.Markdown("""**Workflow:**
+            1. Enter Task ID → Click "Find Audio Tracks"
+            2. Select which audio track to separate
+            3. Choose separation type
+            4. Click "Separate Stems"
+            5. Wait 30-120 seconds
+            6. Get download links for each stem
             
             **Separation types:**
             - 🎤 **separate_vocal**: Vocals + Instrumental (2 stems, 1 credit)
             - 🎛️ **split_stem**: 12 detailed stems (5 credits)
             
-            **Stem Types in split_stem:**
-            - Vocals, Backing Vocals, Drums, Bass
-            - Guitar, Keyboard, Strings, Brass
-            - Woodwinds, Percussion, Synth, FX/Other
-            
             ⚠️ **Important:** 
             - Each request consumes credits
             - Links expire in 14 days
             """)
         
         with gr.Column(scale=2):
-            output = gr.Markdown(
+            status_output = gr.Markdown(
+                label="Status",
+                value="Enter a Task ID and click 'Find Audio Tracks'..."
+            )
+            
+            results_output = gr.Markdown(
                 label="Separation Results",
-                value="Your separated stems will appear here..."
+                visible=False
             )
     
     gr.Markdown("---")
@@ -205,13 +291,71 @@ with gr.Blocks(title="Suno Stem Separator", theme="soft") as app:
             <a href="https://sunoapi.org" target="_blank">Suno API Docs</a> • 
             <a href="https://docs.sunoapi.org/separate-vocals" target="_blank">Stem Separation Guide</a></p>
             <p><small>This app uses the Suno API to separate tracks into individual stems.</small></p>
-            <p><small>💡 <strong>Tip:</strong> Find your Task ID and Audio ID in your Suno generation history or API responses.</small></p>
         </div>
         """,
         elem_id="footer"
     )
     
-    btn.click(separate_vocals, [task_id, audio_id, separation_type], output)
+    # Step 1: Get audio info when button is clicked
+    def on_get_audio(task_id):
+        if not task_id:
+            return (
+                "❌ Please enter a Task ID",
+                gr.Dropdown(choices=[], visible=False),
+                gr.Radio(visible=False),
+                gr.Button(visible=False),
+                gr.Markdown(visible=False),
+                None
+            )
+        
+        output, audio_options = get_audio_info(task_id)
+        
+        if not audio_options:
+            # No audio found, show error
+            return (
+                output,
+                gr.Dropdown(choices=[], visible=False),
+                gr.Radio(visible=False),
+                gr.Button(visible=False),
+                gr.Markdown(visible=False),
+                None
+            )
+        
+        # Create choices for dropdown
+        choices = [(display_text, audio_id) for display_text, audio_id, _ in audio_options]
+        
+        # Store full audio data for later use
+        audio_data_map = {audio_id: audio for _, audio_id, audio in audio_options}
+        
+        return (
+            output,
+            gr.Dropdown(choices=choices, value=choices[0][1] if choices else None, visible=True),
+            gr.Radio(visible=True),
+            gr.Button(visible=True),
+            gr.Markdown(visible=False),
+            audio_data_map
+        )
+    
+    # Step 2: Separate stems when button is clicked
+    def on_separate(task_id, audio_id, separation_type, audio_data_store):
+        # Use the stored task_id from the first step
+        yield from separate_vocals(task_id, audio_id, separation_type)
+    
+    # Connect events
+    get_audio_btn.click(
+        fn=on_get_audio,
+        inputs=[task_id_input],
+        outputs=[status_output, audio_dropdown, separation_type, separate_btn, results_output, audio_data_store]
+    )
+    
+    separate_btn.click(
+        fn=on_separate,
+        inputs=[task_id_input, audio_dropdown, separation_type, audio_data_store],
+        outputs=[results_output]
+    ).then(
+        lambda: gr.Markdown(visible=True),
+        outputs=[results_output]
+    )
 
 if __name__ == "__main__":
     print("🚀 Starting Suno Stem Separator")