Upload app.py

app.py

@@ -1,77 +1,160 @@
+import gradio as gr
+import requests
 import os
 import time
-import requests
-import gradio as gr
-SUNO_KEY = os.environ.get("SunoKey")
-if not SUNO_KEY:
-    raise ValueError("Suno API key not found! Please set the secret 'SunoKey'.")
 
-API_BASE = "https://api.sunoapi.org/api/v1"
-
-def generate_lyrics(prompt):
-    """Generate lyrics from Suno API with full error feedback."""
-    try:
-        # Submit the lyrics generation task
-        submit_resp = requests.post(
-            f"{API_BASE}/lyrics",
-            headers={
-                "Authorization": f"Bearer {SUNO_KEY}",
-                "Content-Type": "application/json"
-            },
-            json={
-                "prompt": prompt,
-                "callBackUrl": ""  # empty if we want to poll instead
-            }
-        )
-        submit_resp.raise_for_status()
-        submit_data = submit_resp.json()
-        if submit_data.get("code") != 200 or "taskId" not in submit_data.get("data", {}):
-            return f"Failed to submit task: {submit_data}"
-        
-        task_id = submit_data["data"]["taskId"]
-        print(f"Submitted task. Task ID: {task_id}")
-
-        # Polling for status
-        max_attempts = 30
-        for attempt in range(max_attempts):
-            time.sleep(2)  # 2 seconds between polls
-            status_resp = requests.get(
-                f"{API_BASE}/lyrics/details?taskId={task_id}",
-                headers={"Authorization": f"Bearer {SUNO_KEY}"}
-            )
-            if status_resp.status_code == 404:
-                # Task not ready yet
-                continue
-            try:
-                status_resp.raise_for_status()
-            except requests.exceptions.HTTPError as e:
-                # Full feedback on HTTP error
-                return f"HTTP Error during polling: {e}\nResponse body: {status_resp.text}"
+# Load Suno API key from environment variable
+SUNO_KEY = os.environ.get("SUNO_KEY")  # Changed to match typical naming
 
-            status_data = status_resp.json()
-            if status_data.get("code") != 200:
-                return f"API returned error during polling: {status_data}"
+# Check if API key exists
+if not SUNO_KEY:
+    print("⚠️ Warning: SUNO_KEY environment variable not set!")
 
-            lyrics_items = status_data.get("data", {}).get("data", [])
-            if lyrics_items:
-                # Return all variants
-                return "\n\n".join([f"Title: {item['title']}\nLyrics:\n{item['text']}" for item in lyrics_items])
+# Store tasks in a dictionary for multiple users (better than global)
+user_tasks = {}
 
-        return "Task timed out: lyrics not ready after polling."
+def generate_lyrics(prompt, username="default"):
+    """Submit lyrics generation task"""
+    if not SUNO_KEY:
+        return "Error: API key not configured. Please set SUNO_KEY environment variable."
+    
+    url = "https://api.sunoapi.org/api/v1/lyrics"
+    headers = {
+        "Authorization": f"Bearer {SUNO_KEY}",
+        "Content-Type": "application/json"
+    }
+    payload = {
+        "prompt": prompt,
+        "callBackUrl": ""
+    }
 
+    try:
+        resp = requests.post(url, headers=headers, json=payload, timeout=30)
+        resp.raise_for_status()
+        data = resp.json()
+        
+        if data.get("code") == 200 and "data" in data:
+            task_id = data["data"]["taskId"]
+            user_tasks[username] = task_id
+            return f"✅ Submitted!\nTask ID: {task_id}\nPrompt: {prompt[:50]}..."
+        else:
+            return f"❌ API Error: {data.get('message', 'Unknown error')}"
+            
     except requests.exceptions.RequestException as e:
-        return f"Request Exception: {e}"
+        return f"❌ Connection Error: {str(e)}"
     except Exception as e:
-        return f"Unexpected Exception: {e}"
+        return f"❌ Unexpected Error: {str(e)}"
+
+def poll_task(username="default"):
+    """Poll the current user's task"""
+    if not SUNO_KEY:
+        return "Error: API key not configured."
+    
+    task_id = user_tasks.get(username)
+    if not task_id:
+        return "No task submitted yet! Generate lyrics first."
+    
+    url = f"https://api.sunoapi.org/api/v1/lyrics/details?taskId={task_id}"
+    headers = {"Authorization": f"Bearer {SUNO_KEY}"}
+    
+    try:
+        resp = requests.get(url, headers=headers, timeout=30)
+        resp.raise_for_status()
+        data = resp.json()
+        
+        if data.get("code") == 200 and "data" in data:
+            task_data = data["data"]
+            status = task_data.get("status", "unknown")
+            
+            if status == "completed" and "data" in task_data:
+                lyrics_data = task_data["data"]
+                output_lines = []
+                
+                for i, item in enumerate(lyrics_data, 1):
+                    output_lines.append(f"🎵 Variant {i}: {item.get('title', 'Untitled')}")
+                    output_lines.append(f"{item.get('text', 'No lyrics')}")
+                    output_lines.append("-" * 40)
+                
+                return "\n".join(output_lines)
+            else:
+                return f"⏳ Status: {status}\nTask ID: {task_id}\n\nCheck back in a few seconds..."
+        else:
+            return f"❌ API Error: {data.get('message', 'Unknown error')}"
+            
+    except requests.exceptions.RequestException as e:
+        return f"❌ Connection Error: {str(e)}"
+
+def auto_poll_task(username="default"):
+    """Auto-poll with loading animation"""
+    if username not in user_tasks:
+        return "No task submitted yet!"
+    
+    result = poll_task(username)
+    
+    # If still processing, show animation
+    if "⏳ Status: processing" in result:
+        return f"🔄 Processing...\n{result}"
+    
+    return result
 
-# Gradio interface
-iface = gr.Interface(
-    fn=generate_lyrics,
-    inputs=gr.Textbox(label="Enter Lyrics Prompt", placeholder="Write your song idea here..."),
-    outputs=gr.Textbox(label="Generated Lyrics"),
-    title="Suno AI Lyrics Generator",
-    description="Generate lyrics using Suno API with full error feedback."
-)
+# Enhanced Gradio interface
+with gr.Blocks(theme=gr.themes.Soft(), title="Suno Lyrics Generator") as app:
+    gr.Markdown("# 🎵 Suno AI Lyrics Generator")
+    gr.Markdown("Generate song lyrics using Suno's AI API")
+    
+    with gr.Row():
+        with gr.Column(scale=2):
+            prompt_input = gr.Textbox(
+                label="Lyrics Prompt",
+                placeholder="Enter your lyrics idea (e.g., 'A love song about Paris in spring')",
+                lines=3
+            )
+            
+            with gr.Row():
+                submit_btn = gr.Button("✨ Generate Lyrics", variant="primary")
+                poll_btn = gr.Button("🔄 Check Status", variant="secondary")
+                auto_poll_btn = gr.Button("🔁 Auto Refresh", variant="secondary")
+                
+            gr.Markdown("### How to use:")
+            gr.Markdown("""
+            1. Enter your lyrics idea
+            2. Click **Generate Lyrics** to submit
+            3. Wait 10-30 seconds
+            4. Click **Check Status** to see results
+            """)
+            
+        with gr.Column(scale=3):
+            output_area = gr.Textbox(
+                label="Output",
+                lines=20,
+                placeholder="Your generated lyrics will appear here..."
+            )
+    
+    # Event handlers
+    submit_btn.click(
+        generate_lyrics,
+        inputs=prompt_input,
+        outputs=output_area
+    )
+    
+    poll_btn.click(
+        poll_task,
+        inputs=None,
+        outputs=output_area
+    )
+    
+    auto_poll_btn.click(
+        auto_poll_task,
+        inputs=None,
+        outputs=output_area,
+        every=5  # Auto-refresh every 5 seconds while visible
+    )
 
+# Launch with better configuration
 if __name__ == "__main__":
-    iface.launch()
+    app.launch(
+        share=False,  # Set to True for temporary public link
+        server_name="0.0.0.0",
+        server_port=7860,
+        show_error=True
+    )