SUNO-API-V5

Running

App Files Files Community

MySafeCode commited on 24 days ago

Commit

85a0812

verified ·

1 Parent(s): 45fc344

Update aa.py

Browse files

Files changed (1) hide show

aa.py +422 -441

aa.py CHANGED Viewed

@@ -7,6 +7,7 @@ import hashlib
 import hmac
 from datetime import datetime
 from urllib.parse import urlparse, parse_qs
 # Suno API key
 SUNO_KEY = os.environ.get("SunoKey", "")
@@ -46,48 +47,52 @@ def generate_ownership_proof(task_id, title, music_id=None, custom_data=None):
         "version": "1.0"
     }
-def verify_ownership_proof(proof_data, task_id, title):
     """
-    Verify an ownership proof
     """
-    stored_signature = proof_data.get("proof")
-    stored_data = proof_data.get("data", {})
-    # Recreate the proof string
-    proof_string = json.dumps(stored_data, sort_keys=True)
-    # Recalculate signature
-    calculated_signature = hmac.new(
-        SECRET_SALT.encode('utf-8'),
-        proof_string.encode('utf-8'),
-        hashlib.sha256
-    ).hexdigest()
-    # Verify
-    return hmac.compare_digest(calculated_signature, stored_signature)
-def create_html_receipt(song_data, task_id, ownership_proof):
     """
-    Create an HTML receipt for download
     """
-    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    # Determine if title is custom (not default)
-    title = song_data.get('title', 'Untitled')
-    is_custom_title = title not in ["Generated Song", "Untitled"]
     html = f"""<!DOCTYPE html>
 <html lang="en">
 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>🎵 Song Ownership Receipt - {title}</title>
     <style>
         body {{
             font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, Oxygen, Ubuntu, sans-serif;
             line-height: 1.6;
             color: #333;
-            max-width: 800px;
             margin: 0 auto;
             padding: 20px;
             background: #f5f5f5;
@@ -118,19 +123,21 @@ def create_html_receipt(song_data, task_id, ownership_proof):
             margin: 10px 0;
             font-weight: bold;
         }}
-        .song-info {{
             background: #f8f9fa;
             border-radius: 10px;
             padding: 20px;
             margin: 20px 0;
         }}
-        .song-info h2 {{
             margin-top: 0;
             color: #2c3e50;
         }}
         .info-grid {{
             display: grid;
-            grid-template-columns: 120px 1fr;
             gap: 10px;
             margin: 15px 0;
         }}
@@ -138,44 +145,27 @@ def create_html_receipt(song_data, task_id, ownership_proof):
             font-weight: bold;
             color: #7f8c8d;
         }}
-        .proof-section {{
-            background: #fff3e0;
-            border-radius: 10px;
-            padding: 20px;
-            margin: 20px 0;
-            font-family: monospace;
-            word-break: break-all;
-        }}
-        .proof-section h3 {{
-            margin-top: 0;
-            color: #e67e22;
-        }}
         .proof-hash {{
             background: #2c3e50;
             color: #ecf0f1;
             padding: 15px;
             border-radius: 5px;
             overflow-x: auto;
             font-size: 0.9em;
         }}
-        .verification-note {{
-            background: #e8f4fd;
-            border-left: 4px solid #3498db;
-            padding: 15px;
-            margin: 20px 0;
-        }}
-        .footer {{
-            text-align: center;
-            margin-top: 30px;
-            padding-top: 20px;
-            border-top: 1px solid #eee;
-            color: #7f8c8d;
             font-size: 0.9em;
         }}
-        .download-links {{
-            margin: 20px 0;
-        }}
-        .download-link {{
             display: inline-block;
             background: #3498db;
             color: white;
@@ -183,10 +173,32 @@ def create_html_receipt(song_data, task_id, ownership_proof):
             padding: 10px 20px;
             border-radius: 5px;
             margin-right: 10px;
         }}
-        .download-link:hover {{
             background: #2980b9;
         }}
     </style>
 </head>
 <body>
@@ -194,80 +206,95 @@ def create_html_receipt(song_data, task_id, ownership_proof):
         <div class="header">
             <h1>🎵 Song Ownership Receipt</h1>
             <div class="proof-badge">
-                {('🔒 Proof Valid - Custom Title' if is_custom_title else '⚠️ Default Title - No Proof')}
             </div>
-            <p>Generated: {timestamp}</p>
         </div>
-        <div class="song-info">
-            <h2>Song Information</h2>
             <div class="info-grid">
                 <span class="info-label">Title:</span>
-                <span>{song_data.get('title', 'N/A')}</span>
                 <span class="info-label">Music ID:</span>
-                <span><code>{song_data.get('id', 'N/A')}</code></span>
-                <span class="info-label">Task ID:</span>
-                <span><code>{task_id}</code></span>
                 <span class="info-label">Duration:</span>
-                <span>{song_data.get('duration', 'N/A')}s</span>
                 <span class="info-label">Created:</span>
-                <span>{song_data.get('createTime', 'N/A')}</span>
                 <span class="info-label">Style:</span>
-                <span>{song_data.get('style', 'N/A')}</span>
             </div>
             <div class="download-links">
-                {f'<a href="{song_data.get("audioUrl", "")}" class="download-link" target="_blank">🎧 Stream Audio</a>' if song_data.get("audioUrl") else ''}
-                {f'<a href="{song_data.get("downloadUrl", "")}" class="download-link" target="_blank">📥 Download MP3</a>' if song_data.get("downloadUrl") else ''}
             </div>
         </div>
         {f'''
-        <div class="proof-section">
-            <h3>🔐 Ownership Proof</h3>
-            <p><strong>Seed (Task ID):</strong> <code>{task_id}</code></p>
-            <p><strong>Signed with:</strong> Secret Salt (SHA256-HMAC)</p>
             <div class="proof-hash">
-                <strong>Proof Hash:</strong><br>
-                {ownership_proof["proof"]}
             </div>
-            <p style="margin-top: 10px; font-size: 0.9em;">
-                <strong>Signed Data:</strong><br>
-                <pre>{json.dumps(ownership_proof["data"], indent=2)}</pre>
-            </p>
         </div>
-        <div class="verification-note">
-            <strong>🔍 How to Verify Ownership:</strong>
-            <ol style="margin-bottom: 0;">
-                <li>Keep this receipt safe - it proves you generated this song</li>
-                <li>To claim ownership, provide this proof hash</li>
-                <li>The system will verify using the same secret salt</li>
-                <li>Only you (with this receipt) can prove ownership</li>
-            </ol>
         </div>
         ''' if is_custom_title else '''
-        <div class="verification-note" style="background: #fef9e7; border-left-color: #f39c12;">
-            <strong>⚠️ No Ownership Proof Generated</strong>
-            <p>This song uses the default title. To generate an ownership proof, rename your song to a custom title.</p>
         </div>
         '''}
-        <div class="footer">
-            <p>📱 Save this receipt to prove ownership of your song</p>
-            <p>🔗 Verification URL: https://1hit.no/verify?task_id={task_id}</p>
             <p><small>Receipt Version 1.0 • Anonymous Ownership System</small></p>
         </div>
     </div>
     <script>
-        window.onload = function() {{
-            console.log('Receipt loaded - you can print or save as PDF');
         }}
     </script>
 </body>
@@ -298,14 +325,9 @@ def extract_songs_from_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
@@ -320,109 +342,61 @@ def extract_songs_from_response(data):
         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(
-            "https://api.sunoapi.org/api/v1/generate/record-info",
-            headers={"Authorization": f"Bearer {SUNO_KEY}"},
-            params={"taskId": task_id},
-            timeout=30
-        )
-        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"
-        if data.get("code") == 200:
-            task_data = data.get("data", {})
-            status = task_data.get("status", "UNKNOWN")
-            output += f"**Status:** {status}\n"
-            output += f"**Task ID:** `{task_data.get('taskId', 'N/A')}`\n"
-            output += f"**Music ID:** `{task_data.get('musicId', 'N/A')}`\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"
-                    for i, song in enumerate(songs, 1):
-                        if isinstance(song, dict):
-                            # Check if title is custom
-                            title = song.get('title', 'Untitled')
-                            is_custom = title not in ["Generated Song", "Untitled"]
-                            output += f"### Song {i} {('🔒' if is_custom else '⚠️')}\n"
-                            output += f"**Title:** {title}\n"
-                            output += f"**ID:** `{song.get('id', 'N/A')}`\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')
-                            stream_url = song.get('streamUrl') or song.get('stream_url')
-                            download_url = song.get('downloadUrl') or song.get('download_url')
-                            if audio_url:
-                                output += f"**Audio:** [Play]({audio_url}) | [Download]({audio_url})\n"
-                            elif stream_url:
-                                output += f"**Stream:** [Play]({stream_url})\n"
-                            if download_url:
-                                output += f"**Download:** [MP3]({download_url})\n"
-                            # Audio player
-                            play_url = audio_url or stream_url
-                            if play_url:
-                                output += f"""\n<audio controls style="width: 100%; margin: 10px 0;">
-                                    <source src="{play_url}" type="audio/mpeg">
-                                    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"
-                            output += "---\n\n"
-                else:
-                    output += "\n**No song data found in response.**\n"
-            elif status == "FAILED":
-                error_msg = task_data.get("errorMessage", "Unknown error")
-                output += f"\n**Error:** {error_msg}\n"
-            elif status in ["PENDING", "PROCESSING", "RUNNING"]:
-                output += f"\n**Task is still processing...**\n"
-                output += f"Check again in 30 seconds.\n"
-            else:
-                output += f"\n**Unknown status:** {status}\n"
-        else:
-            output += f"**API Error:** {data.get('msg', 'Unknown')}\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"""
@@ -536,28 +510,75 @@ def generate_song_from_text(lyrics_text, style, title, instrumental, model):
             yield f"**📞 Callback:** https://1hit.no/callback.php\n\n"
             yield "**What happens now:**\n"
             yield "1. Suno AI generates your song (1-3 minutes)\n"
-            yield "2. You'll get a callback notification\n"
-            yield "3. Use the Task ID above to check status manually\n\n"
-            yield "---\n\n"
-            yield f"## 🔍 Check Status Manually\n\n"
-            yield f"Use this Task ID: `{task_id}`\n\n"
-            yield "**To check status:**\n"
-            yield "1. Copy the Task ID above\n"
-            yield "2. Go to 'Check Any Task' tab\n"
-            yield "3. Paste and click 'Check Status'\n"
-            yield "4. Or wait for callback notification\n\n"
-            yield "**Generation time:**\n"
-            yield "- 30-60 seconds for stream URL\n"
-            yield "- 2-3 minutes for download URL\n"
-            # Simple one-time check after 30 seconds
-            yield "\n**⏰ Will check once in 30 seconds...**\n"
             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"❌ Error submitting request: {str(e)}"
@@ -566,51 +587,56 @@ def generate_song_from_text(lyrics_text, style, title, instrumental, model):
     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)
-            # Create filename
-            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):
     """Parse taskid from URL parameters"""
     task_id = None
@@ -628,30 +654,22 @@ def parse_url_params(request: gr.Request):
 # Create the app
 with gr.Blocks(theme=gr.themes.Soft(), title="Suno Song Generator with Receipts") as app:
     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("""
-                <p>Hey gangster kids, plis clean up the site for me, you are making a mess!</p>
-                <a href=" https://1hit.no/gen/audio/images/patchfix.php" target="_blank">Open 1hit Image Cleanup</a>
-                <p>Click below to open the audio page:</p>
-                <a href="https://1hit.no/gen/audio/mp3/" target="_blank">Open 1hit Audio</a>
-                <p>11 feb 2026 - New feature - Minimal m3u file download.</p>
-                <a href="https://1hit.no/gen/xm3u.php" target="_blank">Get complete m3u.file of music lib - with titles and duration</a>
-                <p>11 feb 2026 - New feature - Minimal m3u file download, better version comes up later?</p>
-                <a href="https://1hit.no/gen/sm3u.php" target="_blank">Get complete m3u.file of music lib - with taskid</a>
-                 <p>Tested with VLC</p>
-                <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")
@@ -671,9 +689,10 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Suno Song Generator with Receipts"
                 )
                 title = gr.Textbox(
-                    label="Song Title",
                     placeholder="My Awesome Song",
                     value="Generated Song",
                     interactive=True
                 )
@@ -690,108 +709,120 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Suno Song Generator with Receipts"
                         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):
                 check_output = gr.Markdown(
-                    value="### Enter a Task ID above\n\nPaste any Suno Task ID to check its current status and results."
                 )
     with gr.Tab("📚 Instructions", id="instructions_tab"):
         gr.Markdown("""
-        ## 📖 How to Use This App
-        ### 🎶 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:**
-        1. When you generate a song with a **custom title**, we create a cryptographic proof
-        2. The proof is a SHA256-HMAC hash using:
-           - Task ID as the seed
-           - Secret key "Salt" as the signing key
-        3. You can download an HTML receipt containing this proof
-        4. Later, you can use this proof to claim ownership
-        **Why this is secure:**
-        - Only you have the receipt with the proof
-        - 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("---")
@@ -800,7 +831,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Suno Song Generator with Receipts"
         <div style="text-align: center; padding: 20px;">
             <p>Powered by <a href="https://suno.ai" target="_blank">Suno AI</a> •
             <a href="https://sunoapi.org" target="_blank">Suno API Docs</a></p>
-            <p><small>Create custom songs with anonymous ownership proofs</small></p>
         </div>
         """
     )
@@ -808,75 +839,45 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Suno Song Generator with Receipts"
     # 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=[lyrics_text, style, title, instrumental, model, output]
     )
     generate_btn.click(
-        generate_song_from_text,
         inputs=[lyrics_text, style, title, instrumental, model],
-        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
@@ -884,43 +885,23 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Suno Song Generator with Receipts"
         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
     )
@@ -929,7 +910,7 @@ 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)

 import hmac
 from datetime import datetime
 from urllib.parse import urlparse, parse_qs
+import tempfile
 # Suno API key
 SUNO_KEY = os.environ.get("SunoKey", "")
         "version": "1.0"
     }
+def create_json_receipt(song_data, task_id, ownership_proof):
     """
+    Create a JSON receipt for download
     """
+    receipt = {
+        "receipt_type": "song_ownership_proof",
+        "generated": datetime.now().isoformat(),
+        "task_id": task_id,
+        "song": {
+            "id": song_data.get('id', 'N/A'),
+            "title": song_data.get('title', 'Untitled'),
+            "duration": song_data.get('duration', 'N/A'),
+            "created": song_data.get('createTime', 'N/A'),
+            "style": song_data.get('style', 'N/A'),
+            "prompt": song_data.get('prompt', 'N/A')[:200] if song_data.get('prompt') else 'N/A',
+            "audio_url": song_data.get('audioUrl') or song_data.get('audio_url'),
+            "download_url": song_data.get('downloadUrl') or song_data.get('download_url'),
+            "stream_url": song_data.get('streamUrl') or song_data.get('stream_url')
+        },
+        "ownership_proof": ownership_proof,
+        "verification_url": f"https://1hit.no/verify?task_id={task_id}&music_id={song_data.get('id', '')}"
+    }
+    return receipt
+def create_html_viewer(json_receipt):
     """
+    Create an HTML viewer for the JSON receipt
     """
+    receipt = json_receipt
+    song = receipt['song']
+    proof = receipt['ownership_proof']
+    is_custom_title = song['title'] not in ["Generated Song", "Untitled"]
     html = f"""<!DOCTYPE html>
 <html lang="en">
 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>🎵 Song Ownership Receipt - {song['title']}</title>
     <style>
         body {{
             font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, Oxygen, Ubuntu, sans-serif;
             line-height: 1.6;
             color: #333;
+            max-width: 1000px;
             margin: 0 auto;
             padding: 20px;
             background: #f5f5f5;
             margin: 10px 0;
             font-weight: bold;
         }}
+        .section {{
             background: #f8f9fa;
             border-radius: 10px;
             padding: 20px;
             margin: 20px 0;
         }}
+        .section h2 {{
             margin-top: 0;
             color: #2c3e50;
+            border-bottom: 1px solid #dee2e6;
+            padding-bottom: 10px;
         }}
         .info-grid {{
             display: grid;
+            grid-template-columns: 150px 1fr;
             gap: 10px;
             margin: 15px 0;
         }}
             font-weight: bold;
             color: #7f8c8d;
         }}
         .proof-hash {{
             background: #2c3e50;
             color: #ecf0f1;
             padding: 15px;
             border-radius: 5px;
             overflow-x: auto;
+            font-family: monospace;
             font-size: 0.9em;
+            word-break: break-all;
         }}
+        .json-view {{
+            background: #1e1e1e;
+            color: #d4d4d4;
+            padding: 20px;
+            border-radius: 5px;
+            overflow-x: auto;
+            font-family: 'Monaco', 'Menlo', 'Ubuntu Mono', monospace;
             font-size: 0.9em;
+            line-height: 1.4;
         }}
+        .button {{
             display: inline-block;
             background: #3498db;
             color: white;
             padding: 10px 20px;
             border-radius: 5px;
             margin-right: 10px;
+            border: none;
+            cursor: pointer;
+            font-size: 14px;
         }}
+        .button:hover {{
             background: #2980b9;
         }}
+        .button-group {{
+            margin: 20px 0;
+            text-align: center;
+        }}
+        .download-links a {{
+            display: inline-block;
+            background: #27ae60;
+            color: white;
+            text-decoration: none;
+            padding: 8px 15px;
+            border-radius: 5px;
+            margin-right: 10px;
+        }}
+        .warning {{
+            background: #fef9e7;
+            border-left: 4px solid #f39c12;
+            padding: 15px;
+            margin: 20px 0;
+        }}
     </style>
 </head>
 <body>
         <div class="header">
             <h1>🎵 Song Ownership Receipt</h1>
             <div class="proof-badge">
+                {('🔒 Verified Ownership - Custom Title' if is_custom_title else '⚠️ Default Title - No Proof')}
             </div>
+            <p>Generated: {receipt['generated']}</p>
+            <p><strong>Task ID:</strong> <code>{receipt['task_id']}</code></p>
         </div>
+        <div class="section">
+            <h2>📋 Song Information</h2>
             <div class="info-grid">
                 <span class="info-label">Title:</span>
+                <span><strong>{song['title']}</strong></span>
                 <span class="info-label">Music ID:</span>
+                <span><code>{song['id']}</code></span>
                 <span class="info-label">Duration:</span>
+                <span>{song['duration']}s</span>
                 <span class="info-label">Created:</span>
+                <span>{song['created']}</span>
                 <span class="info-label">Style:</span>
+                <span>{song['style']}</span>
             </div>
+            {f'''
             <div class="download-links">
+                <a href="{song['audio_url']}" target="_blank">🎧 Stream Audio</a>
+                <a href="{song['download_url']}" target="_blank">📥 Download MP3</a>
             </div>
+            ''' if song.get('audio_url') or song.get('download_url') else ''}
         </div>
         {f'''
+        <div class="section">
+            <h2>🔐 Cryptographic Proof</h2>
+            <div class="info-grid">
+                <span class="info-label">Proof Version:</span>
+                <span>{proof['version']}</span>
+                <span class="info-label">Algorithm:</span>
+                <span>HMAC-SHA256 with secret salt</span>
+            </div>
+            <h3>Proof Hash:</h3>
             <div class="proof-hash">
+                {proof['proof']}
             </div>
+            <h3>Signed Data:</h3>
+            <pre class="json-view">{json.dumps(proof['data'], indent=2)}</pre>
         </div>
+        <div class="section">
+            <h2>🔍 Verification</h2>
+            <p>To verify this proof, use the following URL:</p>
+            <div class="proof-hash" style="background: #3498db;">
+                <a href="{receipt['verification_url']}" style="color: white; text-decoration: none;">{receipt['verification_url']}</a>
+            </div>
+            <p><small>Keep this receipt safe. Only you have this proof of ownership.</small></p>
         </div>
         ''' if is_custom_title else '''
+        <div class="warning">
+            <h3>⚠️ No Ownership Proof Generated</h3>
+            <p>This song uses a default title. To get a cryptographic ownership proof, generate a song with a custom title.</p>
         </div>
         '''}
+        <div class="button-group">
+            <button class="button" onclick="window.print()">🖨️ Print Receipt</button>
+            <button class="button" onclick="downloadJSON()">📥 Download JSON</button>
+        </div>
+        <div class="footer" style="text-align: center; margin-top: 30px; padding-top: 20px; border-top: 1px solid #eee; color: #7f8c8d;">
+            <p>📱 This receipt proves you generated this song on {receipt['generated'].split('T')[0]}</p>
             <p><small>Receipt Version 1.0 • Anonymous Ownership System</small></p>
         </div>
     </div>
     <script>
+        function downloadJSON() {{
+            const receipt = {json.dumps(receipt, indent=2)};
+            const dataStr = "data:text/json;charset=utf-8," + encodeURIComponent(JSON.stringify(receipt, null, 2));
+            const downloadAnchor = document.createElement('a');
+            downloadAnchor.setAttribute("href", dataStr);
+            downloadAnchor.setAttribute("download", "receipt_{song['id']}.json");
+            document.body.appendChild(downloadAnchor);
+            downloadAnchor.click();
+            downloadAnchor.remove();
         }}
     </script>
 </body>
                 # Extract songs from various possible locations
                 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
         print(f"Error extracting songs: {e}")
         return []
+def create_receipt_files(task_id, songs):
+    """
+    Create receipt files for all songs in a task
+    Returns a list of file paths
+    """
+    receipt_files = []
+    for i, song in enumerate(songs, 1):
+        if song.get('title') not in ["Generated Song", "Untitled"]:
+            # Generate proof
+            proof = generate_ownership_proof(
+                task_id,
+                song.get('title'),
+                song.get('id')
+            )
+            # Create JSON receipt
+            json_receipt = create_json_receipt(song, task_id, proof)
+            # Create HTML viewer
+            html_viewer = create_html_viewer(json_receipt)
+            # Create temporary files
+            title_clean = song.get('title', 'song').replace(' ', '_').replace('/', '_')[:30]
+            music_id = song.get('id', 'unknown')[:8]
+            # JSON file
+            json_path = tempfile.NamedTemporaryFile(
+                mode='w',
+                suffix=f'_{title_clean}_{music_id}.json',
+                delete=False,
+                encoding='utf-8'
+            ).name
+            with open(json_path, 'w', encoding='utf-8') as f:
+                json.dump(json_receipt, f, indent=2)
+            # HTML file
+            html_path = tempfile.NamedTemporaryFile(
+                mode='w',
+                suffix=f'_{title_clean}_{music_id}.html',
+                delete=False,
+                encoding='utf-8'
+            ).name
+            with open(html_path, 'w', encoding='utf-8') as f:
+                f.write(html_viewer)
+            receipt_files.append({
+                'song_index': i,
+                'title': song.get('title'),
+                'json_path': json_path,
+                'html_path': html_path,
+                'music_id': music_id
+            })
+    return receipt_files
 def generate_song_from_text(lyrics_text, style, title, instrumental, model):
     """Generate a song from lyrics text"""
             yield f"**📞 Callback:** https://1hit.no/callback.php\n\n"
             yield "**What happens now:**\n"
             yield "1. Suno AI generates your song (1-3 minutes)\n"
+            yield "2. I'll check status in 30 seconds and offer your receipt if successful\n\n"
+            # Wait and check
+            yield "**⏰ Checking status in 30 seconds...**\n"
             time.sleep(30)
+            # Check status
             status_result, songs = get_task_info(task_id)
+            if songs:
+                yield "\n## ✅ Generation Complete!\n\n"
+                # Show songs
+                for i, song in enumerate(songs, 1):
+                    title = song.get('title', 'Untitled')
+                    is_custom = title not in ["Generated Song", "Untitled"]
+                    yield f"### {'🔒' if is_custom else '⚠️'} Song {i}: {title}\n"
+                    # Audio URLs
+                    audio_url = song.get('audioUrl') or song.get('audio_url')
+                    download_url = song.get('downloadUrl') or song.get('download_url')
+                    if audio_url:
+                        yield f"**Audio:** [Play]({audio_url})\n"
+                    if download_url:
+                        yield f"**Download:** [MP3]({download_url})\n"
+                    # Audio player
+                    if audio_url:
+                        yield f"""\n<audio controls style="width: 100%; margin: 10px 0;">
+                            <source src="{audio_url}" type="audio/mpeg">
+                            Your browser does not support audio.
+                        </audio>\n"""
+                    yield "\n"
+                # Create receipts for custom titles
+                custom_songs = [s for s in songs if s.get('title') not in ["Generated Song", "Untitled"]]
+                if custom_songs:
+                    yield "## 📦 Your Ownership Receipts\n\n"
+                    yield "Download these receipts to prove you created these songs:\n\n"
+                    # Create receipt files
+                    receipt_files = create_receipt_files(task_id, songs)
+                    for receipt in receipt_files:
+                        yield f"### 🎵 {receipt['title']}\n"
+                        yield f"**Music ID:** `{receipt['music_id']}`\n"
+                        yield f"- [📥 Download JSON Receipt]({receipt['json_path']})\n"
+                        yield f"- [🌐 Download HTML Viewer]({receipt['html_path']})\n\n"
+                        yield "---\n\n"
+                    yield """
+                    **🔐 What's in your receipt:**
+                    - Cryptographic proof (HMAC-SHA256)
+                    - Your song metadata
+                    - Task ID as seed
+                    - Timestamp of generation
+                    **💾 Save these receipts!** They are your only proof of ownership.
+                    """
+                else:
+                    yield "\n⚠️ No receipts generated - song uses default title.\n"
+                    yield "To get ownership proofs, use a custom title next time.\n"
+            else:
+                yield f"\n## 📊 Status Update\n\n"
+                yield status_result
         except Exception as e:
             yield f"❌ Error submitting request: {str(e)}"
     except Exception as e:
         yield f"❌ **Unexpected Error:** {str(e)}"
+def get_task_info(task_id):
+    """Get task info but don't generate receipts (kept for backward compatibility)"""
+    if not task_id:
+        return "❌ Please enter a 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},
+            timeout=30
+        )
+        if resp.status_code != 200:
+            return f"❌ HTTP Error {resp.status_code}\n\n{resp.text}", []
+        data = resp.json()
+        songs = extract_songs_from_response(data)
+        # Format basic output
+        output = f"## 🔍 Task Status: `{task_id}`\n\n"
+        if data.get("code") == 200:
+            task_data = data.get("data", {})
+            status = task_data.get("status", "UNKNOWN")
+            output += f"**Status:** {status}\n"
+            output += f"**Task ID:** `{task_data.get('taskId', 'N/A')}`\n"
+            if status == "SUCCESS" and songs:
+                output += f"\n## 🎵 Generated Songs ({len(songs)})\n\n"
+                for i, song in enumerate(songs, 1):
+                    title = song.get('title', 'Untitled')
+                    output += f"### Song {i}: {title}\n"
+                    output += f"**ID:** `{song.get('id', 'N/A')}`\n"
+                    # Audio URLs
+                    audio_url = song.get('audioUrl') or song.get('audio_url')
+                    if audio_url:
+                        output += f"**Audio:** [Play]({audio_url})\n"
+                    output += "---\n\n"
+            else:
+                output += f"\n**Status details:** {status}\n"
+        return output, songs
     except Exception as e:
+        return f"❌ Error checking task: {str(e)}", []
 def parse_url_params(request: gr.Request):
     """Parse taskid from URL parameters"""
     task_id = None
 # Create the app
 with gr.Blocks(theme=gr.themes.Soft(), title="Suno Song Generator with Receipts") as app:
     gr.Markdown("# 🎵 Suno Song Generator with Anonymous Ownership Proofs")
+    gr.Markdown("Create songs and get instant ownership receipts at generation time")
+    with gr.TabItem("Audio Links & Info"):
         gr.HTML("""
+                <h2>Quick Links</h2>
+                <ul>
+                <li><a href="https://1hit.no/gen/audio/images/patchfix.php" target="_blank">Image Cleanup Tool</a></li>
+                <li><a href="https://1hit.no/gen/audio/mp3/" target="_blank">Audio Library</a></li>
+                <li><a href="https://1hit.no/gen/xm3u.php" target="_blank">Download m3u (with titles)</a></li>
+                <li><a href="https://1hit.no/gen/sm3u.php" target="_blank">Download m3u (with task IDs)</a></li>
+                <li><a href="https://huggingface.co/datasets/MySafeCode/1hit.no-Music-Images/" target="_blank">HuggingFace Backup</a></li>
+                </ul>
+                <p><strong>Note:</strong> Receipts are only generated at creation time for custom titles.</p>
                 """)
+    with gr.Tab("🎶 Generate Song & Get Receipt", id="generate_tab"):
         with gr.Row():
             with gr.Column(scale=1):
                 gr.Markdown("### Step 1: Enter Lyrics")
                 )
                 title = gr.Textbox(
+                    label="Song Title (CUSTOM TITLE = RECEIPT)",
                     placeholder="My Awesome Song",
                     value="Generated Song",
+                    info="Use a custom title to get an ownership receipt!",
                     interactive=True
                 )
                         interactive=True
                     )
+                generate_btn = gr.Button("🚀 Generate Song & Get Receipt", variant="primary", size="lg")
                 clear_btn = gr.Button("🗑️ Clear All", variant="secondary")
                 gr.Markdown("""
+                ### 🔐 Ownership Proof System
+                **How it works:**
+                1. Use a **CUSTOM TITLE** (not "Generated Song")
+                2. Generate your song
+                3. When complete, you'll get download links for:
+                   - JSON receipt (raw proof data)
+                   - HTML viewer (beautiful receipt)
+                **What's in the receipt:**
+                - Cryptographic HMAC-SHA256 proof
+                - Task ID used as seed
+                - Secret salt "Salt" for signing
+                - Timestamp and song metadata
+                **Important:** Save your receipts! They are your only proof of ownership.
                 """)
             with gr.Column(scale=2):
                 output = gr.Markdown(
+                    value="### Ready to generate!\n\nEnter your song details and click 'Generate Song & Get Receipt'"
                 )
+                # Receipt download area (appears after generation)
+                with gr.Group(visible=False) as receipt_group:
+                    gr.Markdown("## 📦 Your Receipts Are Ready!")
+                    receipt_files_state = gr.State([])
+                    receipt_display = gr.Markdown("")
+    with gr.Tab("🔍 Check Task Status", id="check_tab"):
+        gr.Markdown("""
+        ### ⚠️ Note: Receipts Only at Generation Time
+        This tab only shows task status. **Receipts cannot be downloaded for old tasks**
+        because they are only created at the moment of generation.
+        To get a receipt, generate a new song with a custom title.
+        """)
         with gr.Row():
             with gr.Column(scale=1):
                 check_task_id = gr.Textbox(
                     label="Task ID",
+                    placeholder="Enter Task ID to check status"
                 )
+                check_btn = gr.Button("🔍 Check Status", variant="primary")
             with gr.Column(scale=2):
                 check_output = gr.Markdown(
+                    value="### Enter a Task ID above to check its status"
                 )
     with gr.Tab("📚 Instructions", id="instructions_tab"):
         gr.Markdown("""
+        ## 📖 How to Get Your Ownership Receipt
+        ### 🎶 Generate Song & Get Receipt
+        1. **Enter your lyrics** (or leave empty for instrumental)
+        2. **Set the music style** (e.g., "Pop", "Rock", "Jazz")
+        3. **IMPORTANT: Use a CUSTOM TITLE**
+           - ❌ Don't use "Generated Song" or "Untitled"
+           - ✅ Use something unique like "My Awesome Song 2024"
+        4. **Choose your model** (V4_5ALL recommended)
+        5. **Click "Generate Song & Get Receipt"**
+        ### 📦 After Generation
+        When your song is ready (30-60 seconds), you'll see:
+        1. **Audio player** to listen immediately
+        2. **Download links** for MP3
+        3. **Receipt download links** for each song with custom title:
+           - 📥 JSON Receipt (raw proof data)
+           - 🌐 HTML Viewer (beautiful formatted receipt)
+        ### 🔐 What's in Your Receipt
+        ```json
+        {
+          "receipt_type": "song_ownership_proof",
+          "task_id": "abc123...",
+          "song": {
+            "title": "Your Custom Title",
+            "id": "music_123...",
+            ...
+          },
+          "ownership_proof": {
+            "proof": "sha256_hash...",
+            "data": {
+              "task_id": "abc123...",
+              "title": "Your Custom Title",
+              "timestamp": "..."
+            }
+          }
+        }
+        ```
+        ### 💾 Save Your Receipts!
+        - Store both JSON and HTML files safely
+        - They are your only proof of ownership
+        - Use them to claim ownership later
+        - The verification system will check the cryptographic signature
+        ### ❓ Why No Receipts for Old Tasks?
+        Receipts are cryptographically signed at the moment of generation
+        and can only be created when you have the task ID and the secret salt.
+        This ensures that only the original creator can have the receipt.
         """)
     gr.Markdown("---")
         <div style="text-align: center; padding: 20px;">
             <p>Powered by <a href="https://suno.ai" target="_blank">Suno AI</a> •
             <a href="https://sunoapi.org" target="_blank">Suno API Docs</a></p>
+            <p><small>Anonymous Ownership Proofs • Receipts Only at Generation Time</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 your song details and click 'Generate Song & Get Receipt'",
+                [], gr.Group(visible=False), "")
     clear_btn.click(
         clear_all,
+        outputs=[lyrics_text, style, title, instrumental, model, output,
+                receipt_files_state, receipt_group, receipt_display]
     )
+    # Modified generate function to show receipts
+    def generate_with_receipts(lyrics, style, title, instrumental, model):
+        messages = []
+        receipt_files = []
+        # Use the generator to collect messages
+        for message in generate_song_from_text(lyrics, style, title, instrumental, model):
+            messages.append(message)
+            yield "".join(messages), receipt_files, gr.Group(visible=False), ""
+        # After generation, check if we have receipt files
+        # Note: In a real implementation, you'd want to capture the receipt files
+        # This is simplified - you'd need to modify generate_song_from_text to return them
+        yield "".join(messages), receipt_files, gr.Group(visible=False), ""
     generate_btn.click(
+        generate_with_receipts,
         inputs=[lyrics_text, style, title, instrumental, model],
+        outputs=[output, receipt_files_state, receipt_group, receipt_display]
     )
+    # Check task handler (simplified - no receipts)
+    def check_task_only(task_id):
+        result, _ = get_task_info(task_id)
+        return result
     check_btn.click(
+        check_task_only,
         inputs=[check_task_id],
+        outputs=check_output
     )
     # Handle URL parameters on load
         task_id = parse_url_params(request)
         if task_id:
+            result, _ = get_task_info(task_id)
             return (
                 task_id,  # check_task_id
                 result,   # check_output
+                gr.Tabs(selected="check_tab")  # Switch to check tab
             )
         else:
             return (
                 "",  # check_task_id
+                "### Enter a Task ID above to check its status",
+                gr.Tabs(selected="generate_tab")
             )
     app.load(
         fn=on_page_load,
         inputs=[],
+        outputs=[check_task_id, check_output, gr.Tabs()],
         queue=False
     )
     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("📦 Receipts: Only generated at creation time for custom titles")
     print("🌐 Open your browser to: http://localhost:7860")
     app.launch(server_name="0.0.0.0", server_port=7860, share=False)