Update app.py

app.py

@@ -4,9 +4,13 @@ import requests
 import time
 from datetime import datetime
 
-# Polling function - defined FIRST
+# Polling function
 def poll_status(task_id, single_poll=False):
-    """Poll Suno API for task status using /api/v1/wav/record-info"""
+    """Poll Suno API for task status including task ID"""
+    if not task_id:
+        st.error("❌ Task ID cannot be empty")
+        return False
+    
     headers = {
         "Authorization": f"Bearer {st.session_state.suno_api_key}",
         "Content-Type": "application/json"
@@ -14,24 +18,32 @@ def poll_status(task_id, single_poll=False):
     
     try:
         with st.spinner(f"Checking status for task: {task_id}..."):
-            # Poll using the correct endpoint
-            response = requests.get(
-                f"https://api.sunoapi.org/api/v1/wav/record-info",
-                headers=headers,
-                timeout=30
-            )
+            # Two possible endpoints - let's try both formats
+            endpoints =[
+                f"https://api.sunoapi.org/api/v1/wav/record-info/{task_id}",
+                f"https://api.sunoapi.org/api/v1/wav/record-info?taskId={task_id}"
+            ]
             
+            response = None
+            for endpoint in endpoints:
+                try:
+                    response = requests.get(
+                        endpoint,
+                        headers=headers,
+                        timeout=30
+                    )
+                    if response.status_code == 200:
+                        break
+                except:
+                    continue
+            
+            if not response:
+                st.error("❌ Failed to connect to polling endpoint")
+                return False
+                
             if response.status_code == 200:
                 result = response.json()
                 
-                # Add to poll results
-                poll_entry = {
-                    "timestamp": datetime.now().strftime("%H:%M:%S"),
-                    "task_id": task_id,
-                    "response": result
-                }
-                st.session_state.poll_results.append(poll_entry)
-                
                 # Display result
                 if result.get("code") == 200:
                     data = result.get("data", {})
@@ -131,18 +143,18 @@ if 'auto_poll' not in st.session_state:
 if 'suno_api_key' not in st.session_state:
     st.session_state.suno_api_key = SUNO_API_KEY
 
+# Title and info
 st.title("🎵 Suno API Audio Generator")
 
-# Info section
 with st.expander("📋 How it works", expanded=True):
     st.markdown("""
     ### Workflow:
-    1. **Step 1**: Submit task to Suno API → Get `taskId`
+    1. **Step 1**: Submit task to Suno API → Get `taskId` from response
     2. **Step 2**: Suno processes audio (async)
     3. **Step 3**: Suno sends callback to: `https://1hit.no/wav/cb.php`
     4. **Step 4**: You can view callbacks at: `https://1hit.no/wav/view.php`
-    5. **Step 5**: OR use Poll button to check status via `/api/v1/wav/record-info`
-    
+    5. **Step 5**: OR use Poll button to check status via `/api/v1/record-info` endpoint
+
     ### Polling Endpoint:
     ```javascript
     fetch('https://api.sunoapi.org/api/v1/wav/record-info', {
@@ -152,8 +164,13 @@ with st.expander("📋 How it works", expanded=True):
       }
     })
     ```
+    
+    ### Error Handling:
+    - **400**: Task ID cannot be empty
+    - Make sure to include task ID in polling request
     """)
 
+# Main layout
 col1, col2 = st.columns([2, 1])
 
 with col1:
@@ -179,15 +196,21 @@ with col1:
                 "Content-Type": "application/json"
             }
             
+            # Test sending to correct endpoint
+            endpoint_url = "https://api.sunoapi.org/api/v1/wav/generate"
+            st.info(f"Making request to: {endpoint_url}")
+            
             payload = {
                 "taskId": task_id,
                 "audioId": audio_id,
                 "callBackUrl": CALLBACK_URL
             }
             
+            st.json(payload)
+            
             try:
                 response = requests.post(
-                    "https://api.sunoapi.org/api/v1/wav/generate",
+                    endpoint_url,
                     headers=headers,
                     json=payload,
                     timeout=30
@@ -217,6 +240,10 @@ with col1:
                         """)
                     else:
                         st.error("❌ Unexpected response format")
+                        
+                        # Debug info
+                        st.markdown("**Debug Response:**")
+                        st.code(str(result))
                 else:
                     st.error(f"❌ API Error: {response.status_code}")
                     try:
@@ -270,23 +297,6 @@ if auto_poll_btn:
         st.session_state.auto_poll_task_id = poll_task_id
         st.rerun()
 
-# Auto-polling logic
-if st.session_state.get('auto_poll') and st.session_state.get('auto_poll_task_id'):
-    task_id = st.session_state.auto_poll_task_id
-    
-    # Create a placeholder for auto-poll status
-    poll_placeholder = st.empty()
-    
-    with poll_placeholder.container():
-        st.info(f"🔁 Auto-polling task: `{task_id[:12]}...` (every 10 seconds)")
-        
-        # Poll once
-        poll_status(task_id, single_poll=True)
-        
-        # Schedule next poll
-        time.sleep(10)
-        st.rerun()
-
 # Display poll results if any
 if st.session_state.poll_results:
     st.markdown("---")