SUNO-API-V5

Paused

App Files Files Community

MySafeCode commited on Feb 18

Commit

4ca1338

verified ·

1 Parent(s): 85a0812

Update aa.py

Browse files

Files changed (1) hide show

aa.py +191 -646

aa.py CHANGED Viewed

@@ -12,29 +12,25 @@ import tempfile
 # Suno API key
 SUNO_KEY = os.environ.get("SunoKey", "")
 # Secret key for ownership proofs
-SECRET_SALT = "Salt"  # You might want to make this configurable
 if not SUNO_KEY:
     print("⚠️ SunoKey not set!")
-def generate_ownership_proof(task_id, title, music_id=None, custom_data=None):
     """
-    Generate a SHA256 hash proof of ownership using the task_id as seed
-    and "Salt" as the secret key.
     """
-    # Create a unique string to hash
     proof_data = {
         "task_id": task_id,
         "title": title,
         "music_id": music_id,
         "timestamp": datetime.utcnow().isoformat(),
-        "custom": custom_data or {}
     }
-    # Convert to string for hashing
     proof_string = json.dumps(proof_data, sort_keys=True)
-    # Create HMAC-SHA256 using the secret salt
     signature = hmac.new(
         SECRET_SALT.encode('utf-8'),
         proof_string.encode('utf-8'),
@@ -47,52 +43,44 @@ def generate_ownership_proof(task_id, title, music_id=None, custom_data=None):
         "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;
@@ -114,8 +102,8 @@ def create_html_viewer(json_receipt):
             margin: 0;
             font-size: 2.5em;
         }}
-        .proof-badge {{
-            background: {('#27ae60' if is_custom_title else '#95a5a6')};
             color: white;
             padding: 8px 15px;
             border-radius: 20px;
@@ -129,15 +117,9 @@ def create_html_viewer(json_receipt):
             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;
         }}
@@ -155,16 +137,6 @@ def create_html_viewer(json_receipt):
             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;
@@ -172,7 +144,7 @@ def create_html_viewer(json_receipt):
             text-decoration: none;
             padding: 10px 20px;
             border-radius: 5px;
-            margin-right: 10px;
             border: none;
             cursor: pointer;
             font-size: 14px;
@@ -180,22 +152,9 @@ def create_html_viewer(json_receipt):
         .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;
         }}
@@ -204,50 +163,40 @@ def create_html_viewer(json_receipt):
 <body>
     <div class="receipt">
         <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>
@@ -256,32 +205,25 @@ def create_html_viewer(json_receipt):
             </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>
@@ -291,7 +233,7 @@ def create_html_viewer(json_receipt):
             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();
@@ -302,122 +244,21 @@ def create_html_viewer(json_receipt):
     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):
-                    songs = response_data.get("sunoData", [])
-                    if not songs:
-                        songs = response_data.get("data", [])
-                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 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"""
     if not SUNO_KEY:
-        yield "❌ Error: SunoKey not configured in environment variables"
-        return
-    if not lyrics_text.strip() and not instrumental:
-        yield "❌ Error: Please provide lyrics or select instrumental"
-        return
-    if not style.strip():
-        yield "❌ Error: Please provide a music style"
         return
-    if not title.strip():
-        yield "❌ Error: Please provide a song title"
-        return
     try:
-        # Prepare request data
         request_data = {
             "customMode": True,
             "instrumental": instrumental,
@@ -428,489 +269,193 @@ def generate_song_from_text(lyrics_text, style, title, instrumental, model):
         }
         if not instrumental:
-            # Apply character limits
-            if model == "V4" and len(lyrics_text) > 3000:
-                lyrics_text = lyrics_text[:3000]
-                yield f"⚠️ Lyrics truncated to 3000 characters for V4 model\n\n"
-            elif model in ["V4_5", "V4_5PLUS", "V4_5ALL", "V5"] and len(lyrics_text) > 5000:
-                lyrics_text = lyrics_text[:5000]
-                yield f"⚠️ Lyrics truncated to 5000 characters for {model} model\n\n"
             request_data["prompt"] = lyrics_text
         else:
             request_data["prompt"] = ""
-        # Apply style length limits
-        if model == "V4" and len(style) > 200:
-            style = style[:200]
-            yield f"⚠️ Style truncated to 200 characters for V4 model\n\n"
-        elif model in ["V4_5", "V4_5PLUS", "V4_5ALL", "V5"] and len(style) > 1000:
-            style = style[:1000]
-            yield f"⚠️ Style truncated to 1000 characters for {model} model\n\n"
-        # Apply title length limits
-        if model in ["V4", "V4_5ALL"] and len(title) > 80:
-            title = title[:80]
-            yield f"⚠️ Title truncated to 80 characters for {model} model\n\n"
-        elif model in ["V4_5", "V4_5PLUS", "V5"] and len(title) > 100:
-            title = title[:100]
-            yield f"⚠️ Title truncated to 100 characters for {model} model\n\n"
-        request_data["style"] = style
-        request_data["title"] = title
-        yield f"## 🚀 Submitting Song Request\n\n"
-        yield f"**Title:** {title}\n"
-        yield f"**Style:** {style}\n"
-        yield f"**Model:** {model}\n"
-        yield f"**Instrumental:** {'Yes' if instrumental else 'No'}\n"
-        if not instrumental:
-            yield f"**Lyrics length:** {len(lyrics_text)} characters\n\n"
-        yield f"**Callback URL:** https://1hit.no/callback.php\n\n"
-        # Submit generation request
-        try:
-            resp = requests.post(
-                "https://api.sunoapi.org/api/v1/generate",
-                json=request_data,
-                headers={
-                    "Authorization": f"Bearer {SUNO_KEY}",
-                    "Content-Type": "application/json"
-                },
-                timeout=30
-            )
-            if resp.status_code != 200:
-                yield f"❌ Submission failed: HTTP {resp.status_code}"
-                yield f"\n**Response:**\n```\n{resp.text}\n```"
-                return
-            data = resp.json()
-            print(f"Submission response: {json.dumps(data, indent=2)}")
-            if data.get("code") != 200:
-                yield f"❌ API error: {data.get('msg', 'Unknown')}"
-                return
-            # Extract task ID from response
-            task_id = None
-            if "taskId" in data:
-                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"
-                yield f"\n**Raw Response:**\n```json\n{json.dumps(data, indent=2)}\n```"
-                return
-            yield f"## ✅ Request Submitted Successfully!\n\n"
-            yield f"**🎯 Task ID:** `{task_id}`\n\n"
-            yield f"**⏳ Status:** Generation started\n"
-            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)}"
-            return
-    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
-    if request:
-        try:
-            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}")
-    return task_id
-# 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")
                 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
                 )
                 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
                 )
                 with gr.Row():
-                    instrumental = gr.Checkbox(
-                        label="Instrumental (No Vocals)",
-                        value=False,
-                        interactive=True
-                    )
                     model = gr.Dropdown(
                         label="Model",
-                        choices=["V5", "V4_5PLUS", "V4_5ALL", "V4_5", "V4"],
-                        value="V4_5ALL",
-                        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("---")
-    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
-    def on_page_load(request: gr.Request):
-        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
-    )
-# 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("📦 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)

 # Suno API key
 SUNO_KEY = os.environ.get("SunoKey", "")
 # Secret key for ownership proofs
+SECRET_SALT = "Salt"
 if not SUNO_KEY:
     print("⚠️ SunoKey not set!")
+def generate_ownership_proof(task_id, title, music_id=None, status="pending"):
     """
+    Generate a SHA256 hash proof using task_id as seed
     """
     proof_data = {
         "task_id": task_id,
         "title": title,
         "music_id": music_id,
         "timestamp": datetime.utcnow().isoformat(),
+        "status": status
     }
     proof_string = json.dumps(proof_data, sort_keys=True)
     signature = hmac.new(
         SECRET_SALT.encode('utf-8'),
         proof_string.encode('utf-8'),
         "version": "1.0"
     }
+def create_immediate_receipt(task_id, title):
     """
+    Create an immediate receipt with just task ID and title
     """
+    proof = generate_ownership_proof(task_id, title, status="pending")
     receipt = {
+        "receipt_type": "song_ownership_proof_pending",
         "generated": datetime.now().isoformat(),
         "task_id": task_id,
+        "title": title,
+        "status": "pending",
+        "message": "Your song is being generated. This receipt proves you requested this song with this title.",
+        "ownership_proof": proof,
+        "check_url": f"https://1hit.no/gen/view.php?taskid={task_id}"
     }
     return receipt
+def create_pending_html_receipt(task_id, title):
     """
+    Create an HTML receipt for pending generation
     """
+    receipt = create_immediate_receipt(task_id, title)
     proof = receipt['ownership_proof']
     html = f"""<!DOCTYPE html>
 <html lang="en">
 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>⏳ Pending Song 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;
             margin: 0;
             font-size: 2.5em;
         }}
+        .pending-badge {{
+            background: #f39c12;
             color: white;
             padding: 8px 15px;
             border-radius: 20px;
             padding: 20px;
             margin: 20px 0;
         }}
         .info-grid {{
             display: grid;
+            grid-template-columns: 120px 1fr;
             gap: 10px;
             margin: 15px 0;
         }}
             font-size: 0.9em;
             word-break: break-all;
         }}
         .button {{
             display: inline-block;
             background: #3498db;
             text-decoration: none;
             padding: 10px 20px;
             border-radius: 5px;
+            margin: 5px;
             border: none;
             cursor: pointer;
             font-size: 14px;
         .button:hover {{
             background: #2980b9;
         }}
         .warning {{
+            background: #fff3cd;
+            border-left: 4px solid #ffc107;
             padding: 15px;
             margin: 20px 0;
         }}
 <body>
     <div class="receipt">
         <div class="header">
+            <h1>⏳ Pending Song Receipt</h1>
+            <div class="pending-badge">🔜 Generation in Progress</div>
             <p>Generated: {receipt['generated']}</p>
+        </div>
+        <div class="warning">
+            <strong>🎵 Your song is being generated!</strong>
+            <p>This receipt proves you requested a song with this title at this time.</p>
+            <p>Check back later for the complete receipt with audio links.</p>
         </div>
         <div class="section">
+            <h2>📋 Request Information</h2>
             <div class="info-grid">
                 <span class="info-label">Title:</span>
+                <span><strong>{title}</strong></span>
+                <span class="info-label">Task ID:</span>
+                <span><code>{task_id}</code></span>
+                <span class="info-label">Status:</span>
+                <span>⏳ Pending</span>
             </div>
         </div>
         <div class="section">
+            <h2>🔐 Ownership Proof</h2>
+            <p><strong>This proof already establishes you created this request:</strong></p>
             <div class="info-grid">
                 <span class="info-label">Proof Version:</span>
                 <span>{proof['version']}</span>
                 <span class="info-label">Algorithm:</span>
+                <span>HMAC-SHA256</span>
             </div>
             <h3>Proof Hash:</h3>
             </div>
             <h3>Signed Data:</h3>
+            <pre class="proof-hash" style="background: #34495e;">{json.dumps(proof['data'], indent=2)}</pre>
         </div>
         <div class="section">
+            <h2>🔍 Check Status</h2>
+            <p>Use this link to check your song status:</p>
+            <div class="proof-hash" style="background: #27ae60;">
+                <a href="{receipt['check_url']}" style="color: white; text-decoration: none;" target="_blank">{receipt['check_url']}</a>
             </div>
         </div>
+        <div style="text-align: center; margin-top: 20px;">
             <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>📱 Save this receipt now! It proves you requested this song.</p>
+            <p><small>You can check back later for the complete receipt with audio links.</small></p>
         </div>
     </div>
             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", "pending_receipt_{task_id[:8]}.json");
             document.body.appendChild(downloadAnchor);
             downloadAnchor.click();
             downloadAnchor.remove();
     return html
 def generate_song_from_text(lyrics_text, style, title, instrumental, model):
+    """Generate a song and provide immediate receipt"""
     if not SUNO_KEY:
+        yield "❌ Error: SunoKey not configured"
         return
+    # Initial submission messages
+    yield f"## 🚀 Submitting Song Request\n\n"
+    yield f"**Title:** {title}\n"
+    yield f"**Style:** {style}\n"
+    yield f"**Model:** {model}\n"
+    yield f"**Instrumental:** {'Yes' if instrumental else 'No'}\n\n"
     try:
+        # Prepare request
         request_data = {
             "customMode": True,
             "instrumental": instrumental,
         }
         if not instrumental:
             request_data["prompt"] = lyrics_text
         else:
             request_data["prompt"] = ""
+        # Submit request
+        resp = requests.post(
+            "https://api.sunoapi.org/api/v1/generate",
+            json=request_data,
+            headers={"Authorization": f"Bearer {SUNO_KEY}"},
+            timeout=30
+        )
+        if resp.status_code != 200:
+            yield f"❌ Submission failed: HTTP {resp.status_code}"
+            return
+        data = resp.json()
+        # Extract task ID
+        task_id = None
+        if "taskId" in data:
+            task_id = data["taskId"]
+        elif "data" in data and "taskId" in data["data"]:
+            task_id = data["data"]["taskId"]
+        if not task_id:
+            yield "❌ Could not extract Task ID"
+            return
+        # Show success
+        yield f"## ✅ Request Submitted!\n\n"
+        yield f"**🎯 Task ID:** `{task_id}`\n\n"
+        # Generate immediate receipt if custom title
+        if title not in ["Generated Song", "Untitled", ""]:
+            # Create receipt files
+            temp_dir = tempfile.mkdtemp()
+            # JSON receipt
+            json_receipt = create_immediate_receipt(task_id, title)
+            json_path = os.path.join(temp_dir, f"receipt_pending_{task_id[:8]}.json")
+            with open(json_path, 'w') as f:
+                json.dump(json_receipt, f, indent=2)
+            # HTML receipt
+            html_content = create_pending_html_receipt(task_id, title)
+            html_path = os.path.join(temp_dir, f"receipt_pending_{task_id[:8]}.html")
+            with open(html_path, 'w') as f:
+                f.write(html_content)
+            # Show receipt download links IMMEDIATELY
+            yield f"""
+### 📥 YOUR INSTANT RECEIPT IS READY!
+Download this receipt NOW to prove you created this request:
+- [📥 Download JSON Receipt]({json_path})
+- [🌐 Download HTML Viewer]({html_path})
+**🔐 This receipt already contains:**
+- Cryptographic proof linking you to Task ID: `{task_id}`
+- Song title: "{title}"
+- Timestamp of request
+- HMAC-SHA256 signature
+**⏳ What happens next:**
+1. Your song is being generated (1-3 minutes)
+2. Keep this receipt safe
+3. Check status with the Task ID above
+4. Later you can get an updated receipt with audio links
+**💾 SAVE THIS RECEIPT NOW!** It's your proof of creation.
+"""
+        else:
+            yield "⚠️ No receipt generated - use custom title for ownership proof\n\n"
+        yield f"**📊 Check status:** https://1hit.no/gen/view.php?taskid={task_id}\n\n"
+        yield "---\n\n"
+        # Poll for completion (optional - still show progress)
+        yield "**⏰ Checking generation progress...**\n\n"
+        for attempt in range(1, 7):  # Check up to 6 times
+            time.sleep(30)
+            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:
+                    data = resp.json()
+                    if data.get("code") == 200:
+                        task_data = data.get("data", {})
+                        status = task_data.get("status")
+                        yield f"**Attempt {attempt}:** Status = {status}\n"
+                        # If we have songs, show them
+                        if status == "TEXT_SUCCESS" or status == "SUCCESS":
+                            yield "\n✅ **Generation complete! Check the task page for audio links.**\n"
+                            break
+            except:
+                yield f"**Attempt {attempt}:** Checking...\n"
+        yield "\n**Done! You can continue checking with the Task ID above.**"
     except Exception as e:
+        yield f"❌ Error: {str(e)}"
+# ... (keep other functions like get_task_info, parse_url_params, etc. from previous version)
+# Create the app (simplified UI focusing on immediate receipts)
+with gr.Blocks(theme=gr.themes.Soft(), title="Suno Song Generator") as app:
+    gr.Markdown("# 🎵 Suno Song Generator with Instant Ownership Proofs")
+    gr.Markdown("Get a cryptographic receipt IMMEDIATELY when you submit your song request")
+    with gr.Tab("🎶 Generate Song"):
         with gr.Row():
             with gr.Column(scale=1):
                 lyrics_text = gr.Textbox(
                     label="Lyrics",
                     placeholder="Paste your lyrics here...",
+                    lines=10
                 )
                 style = gr.Textbox(
                     label="Music Style",
+                    value="pop rock",
                     interactive=True
                 )
                 title = gr.Textbox(
                     label="Song Title (CUSTOM TITLE = RECEIPT)",
+                    value="My Awesome Song",
+                    info="✅ Use a custom title to get an instant ownership receipt!",
                     interactive=True
                 )
                 with gr.Row():
+                    instrumental = gr.Checkbox(label="Instrumental", value=False)
                     model = gr.Dropdown(
                         label="Model",
+                        choices=["V5", "V4_5ALL", "V4"],
+                        value="V5"
                     )
+                generate_btn = gr.Button("🚀 Generate & Get Instant Receipt", variant="primary", size="lg")
             with gr.Column(scale=2):
+                output = gr.Markdown("### Ready to generate!\n\nEnter your details and click generate.")
+    with gr.Tab("🔍 Check Task"):
         with gr.Row():
+            check_task_id = gr.Textbox(label="Task ID", placeholder="Enter Task ID")
+            check_btn = gr.Button("🔍 Check Status")
+        check_output = gr.Markdown("### Enter a Task ID to check status")
     # Event handlers
     generate_btn.click(
+        generate_song_from_text,
         inputs=[lyrics_text, style, title, instrumental, model],
+        outputs=output
     )
+    def check_task(task_id):
+        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:
+                data = resp.json()
+                return f"```json\n{json.dumps(data, indent=2)}\n```"
+            return f"HTTP Error: {resp.status_code}"
+        except Exception as e:
+            return f"Error: {str(e)}"
+    check_btn.click(check_task, inputs=[check_task_id], outputs=check_output)
 if __name__ == "__main__":
+    print("🚀 Starting with INSTANT RECEIPTS!")
     app.launch(server_name="0.0.0.0", server_port=7860, share=False)