SUNO-API-V5

Running

App Files Files Community

MySafeCode commited on 24 days ago

Commit

45fc344

verified ·

1 Parent(s): 51e4d18

Update aa.py

Browse files

Files changed (1) hide show

aa.py +212 -182

aa.py CHANGED Viewed

@@ -187,15 +187,6 @@ def create_html_receipt(song_data, task_id, ownership_proof):
         .download-link:hover {{
             background: #2980b9;
         }}
-        .qr-placeholder {{
-            text-align: center;
-            margin: 20px 0;
-        }}
-        .qr-placeholder img {{
-            max-width: 150px;
-            border: 1px solid #ddd;
-            border-radius: 5px;
-        }}
     </style>
 </head>
 <body>
@@ -275,7 +266,6 @@ def create_html_receipt(song_data, task_id, ownership_proof):
     </div>
     <script>
-        // Add print functionality
         window.onload = function() {{
             console.log('Receipt loaded - you can print or save as PDF');
         }}
@@ -285,20 +275,55 @@ def create_html_receipt(song_data, task_id, ownership_proof):
     return html
-def download_receipt(song_data, task_id, ownership_proof):
     """
-    Create a downloadable receipt file
     """
-    html_content = create_html_receipt(song_data, task_id, ownership_proof)
-    title = song_data.get('title', 'song').replace(' ', '_').replace('/', '_')
-    filename = f"receipt_{title}_{task_id[:8]}.html"
-    return html_content, filename
 def get_task_info(task_id):
     """Manually check any Suno task status"""
     if not task_id:
-        return "❌ Please enter a Task ID"
     try:
         resp = requests.get(
@@ -309,10 +334,13 @@ def get_task_info(task_id):
         )
         if resp.status_code != 200:
-            return f"❌ HTTP Error {resp.status_code}\n\n{resp.text}"
         data = resp.json()
         # Format the response for display
         output = f"## 🔍 Task Status: `{task_id}`\n\n"
@@ -326,25 +354,6 @@ def get_task_info(task_id):
             output += f"**Created:** {task_data.get('createTime', 'N/A')}\n"
             if status == "SUCCESS":
-                response_data = task_data.get("response", {})
-                # Try to parse response (could be string or dict)
-                if isinstance(response_data, str):
-                    try:
-                        response_data = json.loads(response_data)
-                    except:
-                        output += f"\n**Raw Response:**\n```\n{response_data}\n```\n"
-                        response_data = {}
-                # Check for song data
-                songs = []
-                if isinstance(response_data, dict):
-                    songs = response_data.get("sunoData", [])
-                    if not songs:
-                        songs = response_data.get("data", [])
-                elif isinstance(response_data, list):
-                    songs = response_data
                 if songs:
                     output += f"\n## 🎵 Generated Songs ({len(songs)})\n\n"
@@ -361,8 +370,8 @@ def get_task_info(task_id):
                             # Generate ownership proof if title is custom
                             if is_custom:
                                 proof = generate_ownership_proof(task_id, title, song.get('id'))
-                                output += f"**🔐 Proof Generated:** [Download Receipt](#)\n"
-                                output += f"**Proof Hash:** `{proof['proof'][:16]}...`\n"
                             # Audio URLs
                             audio_url = song.get('audioUrl') or song.get('audio_url')
@@ -385,18 +394,6 @@ def get_task_info(task_id):
                                     Your browser does not support audio.
                                 </audio>\n"""
-                            # Add receipt download button if custom title
-                            if is_custom:
-                                # Create a unique ID for this song's receipt
-                                receipt_id = f"receipt_{task_id}_{i}"
-                                output += f"""\n<div style="margin: 10px 0;">
-                                    <button onclick="downloadReceipt('{task_id}', {i})"
-                                            style="background: #27ae60; color: white; border: none;
-                                                   padding: 10px 20px; border-radius: 5px; cursor: pointer;">
-                                        📥 Download Ownership Receipt
-                                    </button>
-                                </div>\n"""
                             output += f"**Prompt:** {song.get('prompt', 'N/A')[:100]}...\n"
                             output += f"**Duration:** {song.get('duration', 'N/A')}s\n"
                             output += f"**Created:** {song.get('createTime', 'N/A')}\n\n"
@@ -420,12 +417,12 @@ def get_task_info(task_id):
         # Show raw JSON for debugging
         output += "\n## 📋 Raw Response\n"
-        output += f"```json\n{json.dumps(data, indent=2)}\n```"
-        return output
     except Exception as e:
-        return f"❌ Error checking task: {str(e)}"
 def generate_song_from_text(lyrics_text, style, title, instrumental, model):
     """Generate a song from lyrics text"""
@@ -527,8 +524,6 @@ def generate_song_from_text(lyrics_text, style, title, instrumental, model):
                 task_id = data["taskId"]
             elif "data" in data and "taskId" in data["data"]:
                 task_id = data["data"]["taskId"]
-            elif data.get("data") and "taskId" in data.get("data", {}):
-                task_id = data["data"]["taskId"]
             if not task_id:
                 yield f"❌ Could not extract Task ID from response"
@@ -560,7 +555,7 @@ def generate_song_from_text(lyrics_text, style, title, instrumental, model):
             time.sleep(30)
             # Single status check
-            status_result = get_task_info(task_id)
             yield "\n## 📊 Status Check (30s)\n\n"
             yield status_result
@@ -571,35 +566,30 @@ def generate_song_from_text(lyrics_text, style, title, instrumental, model):
     except Exception as e:
         yield f"❌ **Unexpected Error:** {str(e)}"
-# Function to download receipt for a specific song
-def download_song_receipt(task_id, song_index, songs_data):
     """
     Generate and return a downloadable receipt for a specific song
     """
     try:
-        # Parse songs data if it's a string
-        if isinstance(songs_data, str):
-            songs_data = json.loads(songs_data)
-        # Get the specific song
-        if isinstance(songs_data, dict):
-            songs = songs_data.get("sunoData", [])
-            if not songs:
-                songs = songs_data.get("data", [])
-        elif isinstance(songs_data, list):
-            songs = songs_data
         else:
-            songs = []
         if 0 <= song_index - 1 < len(songs):
             song = songs[song_index - 1]
-            # Generate ownership proof
-            proof = generate_ownership_proof(
-                task_id,
-                song.get('title'),
-                song.get('id')
-            )
             # Create HTML receipt
             html_content = create_html_receipt(song, task_id, proof)
@@ -608,12 +598,17 @@ def download_song_receipt(task_id, song_index, songs_data):
             title = song.get('title', 'song').replace(' ', '_').replace('/', '_')
             filename = f"receipt_{title}_{task_id[:8]}.html"
-            return html_content, filename
         else:
-            return None, None
     except Exception as e:
         print(f"Error generating receipt: {e}")
-        return None, None
 # Function to handle URL parameters
 def parse_url_params(request: gr.Request):
@@ -624,7 +619,6 @@ def parse_url_params(request: gr.Request):
             query_params = parse_qs(urlparse(request.request.url).query)
             if 'taskid' in query_params:
                 task_id = query_params['taskid'][0]
-                # Remove any whitespace
                 task_id = task_id.strip()
         except Exception as e:
             print(f"Error parsing URL params: {e}")
@@ -636,10 +630,9 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Suno Song Generator with Receipts"
     gr.Markdown("# 🎵 Suno Song Generator with Anonymous Ownership Proofs")
     gr.Markdown("Create songs and get cryptographically signed ownership receipts")
-    # We'll use a hidden component to track initial load
-    initial_load_done = gr.State(value=False)
-    # Store songs data for receipt generation
-    current_songs_data = gr.State(value={})
     with gr.TabItem("Audio Link"):
         gr.HTML("""
@@ -656,29 +649,23 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Suno Song Generator with Receipts"
                 <a href=" https://www.videolan.org/vlc/" target="_blank">Download VLC media player</a>
                 <p>13 feb 2026 - Making a backup of dataset available, but made to many commits. :)</p>
                  <a href="https://huggingface.co/datasets/MySafeCode/1hit.no-Music-Images/" target="_blank">https://huggingface.co/datasets/MySafeCode/1hit.no-Music-Images/</a>
                 """)
-    with gr.Tab("🎶 Generate Song", id="generate_tab") as tab_generate:
         with gr.Row():
             with gr.Column(scale=1):
-                # Lyrics Input
                 gr.Markdown("### Step 1: Enter Lyrics")
                 lyrics_text = gr.Textbox(
                     label="Lyrics",
-                    placeholder="Paste your lyrics here...\n\nExample:\n(Verse 1)\nSun is shining, sky is blue\nBirds are singing, just for you...",
                     lines=10,
                     interactive=True
                 )
-                # Song Settings
                 gr.Markdown("### Step 2: Song Settings")
                 style = gr.Textbox(
                     label="Music Style",
-                    placeholder="Example: Pop, Rock, Jazz, Classical, Electronic, Hip Hop, Country",
                     value="Folk soul flamenco glam rock goa trance fusion",
                     interactive=True
                 )
@@ -703,71 +690,54 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Suno Song Generator with Receipts"
                         interactive=True
                     )
-                # Action Buttons
                 generate_btn = gr.Button("🚀 Generate Song", variant="primary")
                 clear_btn = gr.Button("🗑️ Clear All", variant="secondary")
-                # Instructions
                 gr.Markdown("""
-                **How to use:**
-                1. Paste lyrics (or leave empty for instrumental)
-                2. Set music style
-                3. Enter song title
-                4. Choose model
-                5. Click Generate!
                 **Ownership Proof:**
                 - Use a **custom title** to generate a cryptographic proof
                 - Default titles won't generate proofs
                 - Download HTML receipt to prove ownership
-                **Tips:**
-                - V4_5ALL: Best overall quality
-                - V5: Latest model
-                - Instrumental: No vocals, just music
                 """)
             with gr.Column(scale=2):
-                # Output Area
                 output = gr.Markdown(
                     value="### Ready to generate!\n\nEnter lyrics and settings, then click 'Generate Song'"
                 )
-    with gr.Tab("🔍 Check Any Task", id="check_tab") as tab_check:
         with gr.Row():
             with gr.Column(scale=1):
                 gr.Markdown("### Check Task Status")
-                gr.Markdown("Enter any Suno Task ID to check its status and download receipts")
                 check_task_id = gr.Textbox(
                     label="Task ID",
-                    placeholder="Enter Task ID from generation or separation",
-                    info="From Song Generator or Vocal Separator"
                 )
-                check_btn = gr.Button("🔍 Check Status", variant="primary")
-                check_clear_btn = gr.Button("🗑️ Clear", variant="secondary")
-                # Receipt download section (hidden initially)
-                receipt_download = gr.File(
-                    label="📥 Download Receipt",
-                    visible=False,
-                    interactive=False
                 )
-                # URL parameter info
                 gr.Markdown("""
                 **Quick access via URL:**
                 Add `?taskid=YOUR_TASK_ID` to the URL
-                Example:
-                `https://1hit.no/gen/view.php?task_id=fa3529d5cbaa93427ee4451976ed5c4b`
                 **Ownership Proof:**
                 - Custom titles get SHA256-HMAC proofs
                 - Proof uses Task ID as seed
                 - Secret salt "Salt" for signing
-                - Download HTML receipt to prove ownership
                 """)
             with gr.Column(scale=2):
@@ -781,19 +751,14 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Suno Song Generator with Receipts"
         ### 🎶 Generate Song Tab
         1. **Enter Lyrics** (or leave empty for instrumental)
-        2. **Set Music Style** (e.g., "Pop", "Rock", "Jazz")
-        3. **Enter Song Title** (use a custom title for ownership proof)
         4. **Choose Model** (V4_5ALL recommended)
         5. **Click "Generate Song"**
         ### 🔍 Check Any Task via URL
         Add `?taskid=YOUR_TASK_ID` to the URL
-        Example:
-        ```
-         https://1hit.no/gen/view.php?task_id=fa3529d5cbaa93427ee4451976ed5c4b
-        ```
         ### 🔐 Anonymous Ownership System
         **How it works:**
@@ -809,53 +774,24 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Suno Song Generator with Receipts"
         - The proof can be verified using the same secret salt
         - No need to store personal data
         - Task ID is unique to each generation
-        **Using receipts:**
-        - Save the HTML receipt locally
-        - It contains all song metadata and the proof
-        - Can be used to claim ownership later
-        - Verification system coming soon!
-        ### 📞 Callback Status
-        https://1hit.no/gen/view.php
-        **No audio links?**
-        - Wait 2-3 minutes
-        - Check status again
-        - Generation may have failed
         """)
     with gr.Tab("📚 Less Instructions", id="less_instructions_tab"):
         gr.Markdown("""
-        ## 📖 How to Use This App
-        ### 🎶 Generate Song Tab
-        1. **Enter Lyrics** (or leave empty for instrumental)
-        2. **Set Music Style** (e.g., "Pop", "Rock", "Jazz")
-        3. **Enter Song Title**
-        4. **Choose Model** (V4_5ALL recommended)
-        5. **Click "Generate Song"**
-        ### 🔍 Check Any Task via URL
-        Add `?taskid=YOUR_TASK_ID` to the URL
-        Example:
-        ```
-         https://1hit.no/gen/view.php?task_id=fa3529d5cbaa93427ee4451976ed5c4b
-        ```
         ### 🔐 Anonymous Ownership
         - Custom titles get cryptographic proofs
         - Download HTML receipt to prove ownership
         - Use Task ID as seed with secret salt "Salt"
-        ### 📞 Callback Status
-        https://1hit.no/gen/view.php
-        **No audio links?**
-        - Wait 2-3 minutes
-        - Check status again
-        - Generation may have failed
         """)
     gr.Markdown("---")
@@ -866,13 +802,13 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Suno Song Generator with Receipts"
             <a href="https://sunoapi.org" target="_blank">Suno API Docs</a></p>
             <p><small>Create custom songs with anonymous ownership proofs</small></p>
         </div>
-        """,
-        elem_id="footer"
     )
-    # Event handlers for Generate Song tab
     def clear_all():
-        return "", "Folk soul flamenco glam rock goa trance fusion", "Generated Song", False, "V4_5ALL", "### Ready to generate!\n\nEnter lyrics and settings, then click 'Generate Song'"
     clear_btn.click(
         clear_all,
@@ -885,21 +821,115 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Suno Song Generator with Receipts"
         outputs=output
     )
-    # Event handlers for Check Any Task tab
     def clear_check():
-        return "", "### Enter a Task ID above\n\nPaste any Suno Task ID to check its current status and results.", gr.File(visible=False)
     check_clear_btn.click(
         clear_check,
-        outputs=[check_task_id, check_output, receipt_download]
     )
-    def check_and_store(task_id):
-        """Check task and store songs data for receipt generation"""
-        result = get_task_info(task_id)
-        # Try to extract songs data for receipt generation
-        songs_data = {}
         try:
-            # Parse the result to extract songs data
-            # This is a bit hacky - in production you'd want to parse the actual API response

         .download-link:hover {{
             background: #2980b9;
         }}
     </style>
 </head>
 <body>
     </div>
     <script>
         window.onload = function() {{
             console.log('Receipt loaded - you can print or save as PDF');
         }}
     return html
+def extract_songs_from_response(data):
     """
+    Properly extract songs data from the API response
+    Returns a list of song dictionaries
     """
+    songs = []
+    try:
+        if data.get("code") == 200:
+            task_data = data.get("data", {})
+            if task_data.get("status") == "SUCCESS":
+                response_data = task_data.get("response", {})
+                # Handle different response formats
+                if isinstance(response_data, str):
+                    try:
+                        response_data = json.loads(response_data)
+                    except:
+                        response_data = {}
+                # Extract songs from various possible locations
+                if isinstance(response_data, dict):
+                    # Try different possible keys where songs might be
+                    songs = response_data.get("sunoData", [])
+                    if not songs:
+                        songs = response_data.get("data", [])
+                    if not songs:
+                        # Sometimes the response itself is a list
+                        if isinstance(response_data.get("response"), list):
+                            songs = response_data.get("response", [])
+                elif isinstance(response_data, list):
+                    songs = response_data
+                # Add task_id and other metadata to each song
+                for song in songs:
+                    if isinstance(song, dict):
+                        song['taskId'] = task_data.get('taskId')
+                        song['status'] = task_data.get('status')
+        return songs
+    except Exception as e:
+        print(f"Error extracting songs: {e}")
+        return []
 def get_task_info(task_id):
     """Manually check any Suno task status"""
     if not task_id:
+        return "❌ Please enter a Task ID", []
     try:
         resp = requests.get(
         )
         if resp.status_code != 200:
+            return f"❌ HTTP Error {resp.status_code}\n\n{resp.text}", []
         data = resp.json()
+        # Extract songs for receipt generation
+        songs = extract_songs_from_response(data)
         # Format the response for display
         output = f"## 🔍 Task Status: `{task_id}`\n\n"
             output += f"**Created:** {task_data.get('createTime', 'N/A')}\n"
             if status == "SUCCESS":
                 if songs:
                     output += f"\n## 🎵 Generated Songs ({len(songs)})\n\n"
                             # Generate ownership proof if title is custom
                             if is_custom:
                                 proof = generate_ownership_proof(task_id, title, song.get('id'))
+                                song['_proof'] = proof  # Store proof with song
+                                output += f"**🔐 Proof Generated:** `{proof['proof'][:16]}...`\n"
                             # Audio URLs
                             audio_url = song.get('audioUrl') or song.get('audio_url')
                                     Your browser does not support audio.
                                 </audio>\n"""
                             output += f"**Prompt:** {song.get('prompt', 'N/A')[:100]}...\n"
                             output += f"**Duration:** {song.get('duration', 'N/A')}s\n"
                             output += f"**Created:** {song.get('createTime', 'N/A')}\n\n"
         # Show raw JSON for debugging
         output += "\n## 📋 Raw Response\n"
+        output += f"```json\n{json.dumps(data, indent=2)[:1000]}...\n```"
+        return output, songs
     except Exception as e:
+        return f"❌ Error checking task: {str(e)}", []
 def generate_song_from_text(lyrics_text, style, title, instrumental, model):
     """Generate a song from lyrics text"""
                 task_id = data["taskId"]
             elif "data" in data and "taskId" in data["data"]:
                 task_id = data["data"]["taskId"]
             if not task_id:
                 yield f"❌ Could not extract Task ID from response"
             time.sleep(30)
             # Single status check
+            status_result, songs = get_task_info(task_id)
             yield "\n## 📊 Status Check (30s)\n\n"
             yield status_result
     except Exception as e:
         yield f"❌ **Unexpected Error:** {str(e)}"
+def download_song_receipt(task_id, song_index, songs_json):
     """
     Generate and return a downloadable receipt for a specific song
     """
     try:
+        # Parse songs data
+        if isinstance(songs_json, str):
+            songs = json.loads(songs_json)
         else:
+            songs = songs_json
+        # Get the specific song
         if 0 <= song_index - 1 < len(songs):
             song = songs[song_index - 1]
+            # Check if we already have a proof stored
+            proof = song.get('_proof')
+            if not proof:
+                # Generate ownership proof
+                proof = generate_ownership_proof(
+                    task_id,
+                    song.get('title'),
+                    song.get('id')
+                )
             # Create HTML receipt
             html_content = create_html_receipt(song, task_id, proof)
             title = song.get('title', 'song').replace(' ', '_').replace('/', '_')
             filename = f"receipt_{title}_{task_id[:8]}.html"
+            # Save temporarily and return path
+            temp_path = f"/tmp/{filename}"
+            with open(temp_path, 'w', encoding='utf-8') as f:
+                f.write(html_content)
+            return temp_path
         else:
+            return None
     except Exception as e:
         print(f"Error generating receipt: {e}")
+        return None
 # Function to handle URL parameters
 def parse_url_params(request: gr.Request):
             query_params = parse_qs(urlparse(request.request.url).query)
             if 'taskid' in query_params:
                 task_id = query_params['taskid'][0]
                 task_id = task_id.strip()
         except Exception as e:
             print(f"Error parsing URL params: {e}")
     gr.Markdown("# 🎵 Suno Song Generator with Anonymous Ownership Proofs")
     gr.Markdown("Create songs and get cryptographically signed ownership receipts")
+    # State to store songs data for receipt generation
+    current_songs = gr.State(value=[])
+    current_task_id = gr.State(value="")
     with gr.TabItem("Audio Link"):
         gr.HTML("""
                 <a href=" https://www.videolan.org/vlc/" target="_blank">Download VLC media player</a>
                 <p>13 feb 2026 - Making a backup of dataset available, but made to many commits. :)</p>
                  <a href="https://huggingface.co/datasets/MySafeCode/1hit.no-Music-Images/" target="_blank">https://huggingface.co/datasets/MySafeCode/1hit.no-Music-Images/</a>
                 """)
+    with gr.Tab("🎶 Generate Song", id="generate_tab"):
         with gr.Row():
             with gr.Column(scale=1):
                 gr.Markdown("### Step 1: Enter Lyrics")
                 lyrics_text = gr.Textbox(
                     label="Lyrics",
+                    placeholder="Paste your lyrics here...",
                     lines=10,
                     interactive=True
                 )
                 gr.Markdown("### Step 2: Song Settings")
                 style = gr.Textbox(
                     label="Music Style",
+                    placeholder="Example: Pop, Rock, Jazz",
                     value="Folk soul flamenco glam rock goa trance fusion",
                     interactive=True
                 )
                         interactive=True
                     )
                 generate_btn = gr.Button("🚀 Generate Song", variant="primary")
                 clear_btn = gr.Button("🗑️ Clear All", variant="secondary")
                 gr.Markdown("""
                 **Ownership Proof:**
                 - Use a **custom title** to generate a cryptographic proof
                 - Default titles won't generate proofs
                 - Download HTML receipt to prove ownership
                 """)
             with gr.Column(scale=2):
                 output = gr.Markdown(
                     value="### Ready to generate!\n\nEnter lyrics and settings, then click 'Generate Song'"
                 )
+    with gr.Tab("🔍 Check Any Task", id="check_tab"):
         with gr.Row():
             with gr.Column(scale=1):
                 gr.Markdown("### Check Task Status")
                 check_task_id = gr.Textbox(
                     label="Task ID",
+                    placeholder="Enter Task ID from generation"
                 )
+                with gr.Row():
+                    check_btn = gr.Button("🔍 Check Status", variant="primary")
+                    check_clear_btn = gr.Button("🗑️ Clear", variant="secondary")
+                # Receipt download section
+                gr.Markdown("### 📥 Download Receipts")
+                receipt_dropdown = gr.Dropdown(
+                    label="Select Song for Receipt",
+                    choices=[],
+                    interactive=True,
+                    visible=False
                 )
+                download_btn = gr.Button("📥 Download Selected Receipt", variant="secondary", visible=False)
+                receipt_file = gr.File(label="Downloaded Receipt", visible=False)
                 gr.Markdown("""
                 **Quick access via URL:**
                 Add `?taskid=YOUR_TASK_ID` to the URL
                 **Ownership Proof:**
                 - Custom titles get SHA256-HMAC proofs
                 - Proof uses Task ID as seed
                 - Secret salt "Salt" for signing
                 """)
             with gr.Column(scale=2):
         ### 🎶 Generate Song Tab
         1. **Enter Lyrics** (or leave empty for instrumental)
+        2. **Set Music Style**
+        3. **Enter Song Title** (use custom title for ownership proof)
         4. **Choose Model** (V4_5ALL recommended)
         5. **Click "Generate Song"**
         ### 🔍 Check Any Task via URL
         Add `?taskid=YOUR_TASK_ID` to the URL
         ### 🔐 Anonymous Ownership System
         **How it works:**
         - The proof can be verified using the same secret salt
         - No need to store personal data
         - Task ID is unique to each generation
         """)
     with gr.Tab("📚 Less Instructions", id="less_instructions_tab"):
         gr.Markdown("""
+        ## 📖 Quick Guide
+        ### 🎶 Generate Song
+        1. Enter lyrics
+        2. Set style & title
+        3. Click Generate
+        ### 🔍 Check Task
+        Add `?taskid=YOUR_TASK_ID` to URL
         ### 🔐 Anonymous Ownership
         - Custom titles get cryptographic proofs
         - Download HTML receipt to prove ownership
         - Use Task ID as seed with secret salt "Salt"
         """)
     gr.Markdown("---")
             <a href="https://sunoapi.org" target="_blank">Suno API Docs</a></p>
             <p><small>Create custom songs with anonymous ownership proofs</small></p>
         </div>
+        """
     )
+    # Event handlers
     def clear_all():
+        return ("", "Folk soul flamenco glam rock goa trance fusion", "Generated Song",
+                False, "V4_5ALL", "### Ready to generate!\n\nEnter lyrics and settings, then click 'Generate Song'")
     clear_btn.click(
         clear_all,
         outputs=output
     )
     def clear_check():
+        return ("", "### Enter a Task ID above\n\nPaste any Suno Task ID to check its current status and results.",
+                [], "", gr.Dropdown(choices=[], visible=False), gr.Button(visible=False), gr.File(visible=False))
     check_clear_btn.click(
         clear_check,
+        outputs=[check_task_id, check_output, current_songs, current_task_id,
+                receipt_dropdown, download_btn, receipt_file]
+    )
+    def check_and_update(task_id):
+        """Check task and update UI with receipt options"""
+        if not task_id:
+            return ("Please enter a Task ID", [], "",
+                   gr.Dropdown(choices=[], visible=False), gr.Button(visible=False), gr.File(visible=False))
+        result, songs = get_task_info(task_id)
+        # Update dropdown with songs that have custom titles
+        receipt_choices = []
+        for i, song in enumerate(songs, 1):
+            if song.get('title') not in ["Generated Song", "Untitled"]:
+                receipt_choices.append(f"{i}. {song.get('title', 'Untitled')}")
+        show_receipt_section = len(receipt_choices) > 0
+        return (result, songs, task_id,
+                gr.Dropdown(choices=receipt_choices, visible=show_receipt_section),
+                gr.Button(visible=show_receipt_section),
+                gr.File(visible=False))
+    check_btn.click(
+        check_and_update,
+        inputs=[check_task_id],
+        outputs=[check_output, current_songs, current_task_id,
+                receipt_dropdown, download_btn, receipt_file]
     )
+    def download_selected_receipt(task_id, selection, songs):
+        """Download receipt for selected song"""
+        if not selection or not songs:
+            return None
         try:
+            # Extract song index from selection
+            song_index = int(selection.split('.')[0])
+            receipt_path = download_song_receipt(task_id, song_index, songs)
+            return receipt_path
+        except Exception as e:
+            print(f"Error in download: {e}")
+            return None
+    download_btn.click(
+        download_selected_receipt,
+        inputs=[current_task_id, receipt_dropdown, current_songs],
+        outputs=[receipt_file]
+    )
+    # Handle URL parameters on load
+    def on_page_load(request: gr.Request):
+        task_id = parse_url_params(request)
+        if task_id:
+            result, songs = get_task_info(task_id)
+            # Update dropdown with songs that have custom titles
+            receipt_choices = []
+            for i, song in enumerate(songs, 1):
+                if song.get('title') not in ["Generated Song", "Untitled"]:
+                    receipt_choices.append(f"{i}. {song.get('title', 'Untitled')}")
+            show_receipt_section = len(receipt_choices) > 0
+            return (
+                task_id,  # check_task_id
+                result,   # check_output
+                songs,    # current_songs
+                task_id,  # current_task_id
+                gr.Tabs(selected="check_tab"),  # Switch to check tab
+                gr.Dropdown(choices=receipt_choices, visible=show_receipt_section),
+                gr.Button(visible=show_receipt_section),
+                gr.File(visible=False)
+            )
+        else:
+            return (
+                "",  # check_task_id
+                "### Enter a Task ID above\n\nPaste any Suno Task ID to check its current status and results.",
+                [],  # current_songs
+                "",  # current_task_id
+                gr.Tabs(selected="generate_tab"),
+                gr.Dropdown(choices=[], visible=False),
+                gr.Button(visible=False),
+                gr.File(visible=False)
+            )
+    app.load(
+        fn=on_page_load,
+        inputs=[],
+        outputs=[check_task_id, check_output, current_songs, current_task_id,
+                gr.Tabs(), receipt_dropdown, download_btn, receipt_file],
+        queue=False
+    )
+# Launch the app
+if __name__ == "__main__":
+    print("🚀 Starting Suno Song Generator with Anonymous Ownership Proofs")
+    print(f"🔑 SunoKey: {'✅ Set' if SUNO_KEY else '❌ Not set'}")
+    print(f"🔐 Secret Salt: {'✅ Configured' if SECRET_SALT else '❌ Not set'}")
+    print("🌐 Open your browser to: http://localhost:7860")
+    print("🔗 Use URL parameter: http://localhost:7860?taskid=YOUR_TASK_ID")
+    app.launch(server_name="0.0.0.0", server_port=7860, share=False)