Update app.py

app.py

@@ -1,274 +1,109 @@
 import streamlit as st
-import requests
 import json
-import os
 from datetime import datetime
 
 # Page configuration
 st.set_page_config(
-    page_title="🎵 Suno AI Audio Generator",
+    page_title="Suno API Generator",
     page_icon="🎵",
-    layout="wide",
-    initial_sidebar_state="expanded"
+    layout="centered"
 )
 
-# Custom CSS
-st.markdown("""
-<style>
-    .stApp {
-        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
-    }
-    .api-card {
-        background: rgba(255, 255, 255, 0.95);
-        backdrop-filter: blur(10px);
-        border: 1px solid rgba(255, 255, 255, 0.2);
-        box-shadow: 0 8px 32px 0 rgba(31, 38, 135, 0.37);
-    }
-    .secret-mask {
-        font-family: 'Monaco', 'Menlo', 'Ubuntu Mono', monospace;
-        background: #f3f4f6;
-        padding: 2px 6px;
-        border-radius: 4px;
-        font-size: 12px;
-        color: #374151;
-    }
-    .success-badge {
-        background: #d1fae5;
-        color: #065f46;
-        border-radius: 9999px;
-        padding: 4px 12px;
-        display: inline-flex;
-        align-items: center;
-        font-size: 12px;
-        margin: 4px;
-    }
-    .error-badge {
-        background: #fee2e2;
-        color: #7c2d1a;
-        border-radius: 9999px;
-        padding: 4px 12px;
-        display: inline-flex;
-        align-items: center;
-        font-size: 12px;
-        margin: 4px;
-    }
-</style>
-""", unsafe_allow_html=True)
+# Title
+st.title("🎵 Suno API Audio Generator")
 
-# Main App
-def main():
-    # Header
-    st.title("🎵 Suno AI Audio Generator")
-    st.markdown("Generate audio using Suno API with secure secret management")
-    
-    # Initialize session state
-    if 'secrets_loaded' not in st.session_state:
-        st.session_state.secrets_loaded = False
-        st.session_state.suno_api_key = None
-        st.session_state.callback_url = None
-        st.session_state.last_api_response = None
-        
-    # Sidebar for configuration
-    with st.sidebar:
-        st.header("⚙️ Configuration")
-        st.markdown("---")
-        
-        # Secret Loading Section
-        st.subheader("🔐 Load Secrets")
-        
-        # Load secrets from environment variables
-        suno_api_key = st.secrets.get("SUNO_API_KEY", "").strip() or os.environ.get("SUNO_API_KEY", "")
-        callback_url = st.secrets.get("CALLBACK_URL", "").strip() or os.environ.get("CALLBACK_URL", "")
-        
-        # Show loaded secrets status
-        if suno_api_key and callback_url:
-            st.success("Secrets loaded!")
-            
-            # Display masked secrets
-            with st.expander("🔐 Loaded Secrets", expanded=True):
-                st.markdown(f"""
-                **API Key Status:** ✅ Loaded
-                **Callback URL:** ✅ Loaded
-                
-                **Usage:** API calls will automatically use these secrets.
-                """)
-            
-            st.session_state.suno_api_key = suno_api_key
-            st.session_state.callback_url = callback_url
-            st.session_state.secrets_loaded = True
-        else:
-            st.warning("Secrets not configured")
-            st.markdown("""
-            **To configure:**
-            1. Set **SUNO_API_KEY** in `.streamlit/secrets.toml`
-            2. Set **CALLBACK_URL** in `.streamlit/secrets.toml`
-            """)
-        
-        st.markdown("---")
-        
-        # Manual input for testing
-        st.subheader("🔧 Manual Override (Testing)")
-        manual_api_key = st.text_input("Enter API Key:", type="password")
-        manual_callback_url = st.text_input("Enter Callback URL:", value=callback_url if callback_url else "")
-        
-        if st.button("🔧 Use Manual Credentials"):
-            if manual_api_key and manual_callback_url:
-                st.session_state.suno_api_key = manual_api_key
-                st.session_state.callback_url = manual_callback_url
-                st.session_state.secrets_loaded = True
-                st.success("Using manual credentials")
-                st.rerun()
-    
-    # Main content area
-    col1, col2 = st[3, 1]
-    
-    with col1:
-        st.markdown("### 🎛️ API Parameters")
-        
-        # Task ID input
-        task_id = st.text_input(
-            "Task ID",
-            value="5c79****be8e",
-            help="Enter your Suno API task ID"
-        )
-        
-        # Audio ID input
-        audio_id = st.text_input(
-            "Audio ID",
-            value="e231****-****-****-****-****8cadc7dc",
-            help="Enter your Suno API audio ID"
-        )
-        
-        # API Endpoint selection
-        endpoint = st.selectbox(
-            "API Endpoint",
-            options=["/api/v1/wav/generate", "/api/v1/audio/sync", "/api/v1/audio/status"],
-            help="Select the Suno API endpoint to use"
-        )
-        
-        # Advanced options
-        with st.expander("🔧 Advanced Options"):
-            timeout = st.number_input("Timeout (seconds)", min_value=5, max_value=60, value=30, step=5)
-            retry_count = st.slider("Retry Count", 0, 3, 1)
-            show_debug = st.checkbox("Show Debug Info")
-    
-    with col2:
-        st.markdown("### 📊 Quick Actions")
-        if st.button("🚀 Test API Connection"):
-            test_api_connection()
-        
-        if st.button("🧪 Test with Sample Data"):
-            test_with_sample_data()
-        
-        if st.button("📋 Copy Configuration"):
-            copy_configuration()
-    
-    # API Call Section
-    st.markdown("---")
-    st.markdown("### 🚀 Make API Call")
-    
-    # Make API call button
-    if st.button("🎵 Generate Audio", key="generate_audio", type="primary", use_container_width=True):
-        make_api_call(task_id, audio_id, st.secrets.get("SUNO_API_KEY", manual_api_key if manual_api_key else suno_api_key), callback_url if callback_url else manual_callback_url)
-    
-    # Results section
-    st.markdown("---")
-    if 'last_api_response' in st.session_state:
-        st.markdown("### 📋 Last API Response")
-        st.json(st.session_state.last_api_response)
-        
-        # Download JSON button
-        st.download_button(
-            label="📥 Download Response JSON",
-            data=json.dumps(st.session_state.last_api_response, indent=2),
-            file_name=f"suno_response_{datetime.now().strftime('%Y%m%d_%H%M%S')}.json",
-            mime="application/json"
-        )
+# Load secrets from Hugging Face Space
+SUNO_API_KEY = st.secrets.get("SUNO_API_KEY", "")
+CALLBACK_URL = st.secrets.get("CALLBACK_URL", "")
 
-# Test API Connection
-def test_api_connection():
-    with st.spinner("Testing API connection..."):
-        # Implement API connection test
-        try:
-            test_response = {"status": "connected"}
-            st.success("✅ API connection successful!")
-            st.json(test_response)
-        except Exception as e:
-            st.error(f"❌ API connection failed: {str(e)}")
+# Show secret status
+if SUNO_API_KEY and CALLBACK_URL:
+    st.success("✅ Secrets loaded from Hugging Face!")
+else:
+    st.error("❌ Secrets not configured in Hugging Face Space")
+    st.info("""
+    Add these secrets in your Hugging Face Space settings:
+    1. Go to **Settings** → **Repository secrets**
+    2. Add:
+       - `SUNO_API_KEY`: Your Suno API Bearer token
+       - `CALLBACK_URL`: Your callback URL
+    """)
 
-# Test with Sample Data
-def test_with_sample_data():
-    sample_data = {
-        "taskId": "5c79****be8e",
-        "audioId": "e231****-****-****-****-****8cadc7dc",
-        "callBackUrl": st.session_state.callback_url
-    }
-    
-    st.info("🧪 Testing with sample data...")
-    st.json(sample_data)
+# API Form
+st.markdown("---")
+st.markdown("### API Parameters")
 
-# Copy Configuration
-def copy_configuration():
-    config = {
-        "api_key": st.session_state.suno_api_key,
-        "callback_url": st.session_state.callback_url,
-        "task_id": st.session_state.get("task_id", ""),
-        "audio_id": st.session_state.get("audio_id", "")
-    }
-    
-    st.code(json.dumps(config, indent=2), language="json")
-    st.success("📋 Configuration copied to clipboard!")
+task_id = st.text_input("Task ID", value="5c79****be8e")
+audio_id = st.text_input("Audio ID", value="e231****-****-****-****-****8cadc7dc")
 
-# Make API Call Function
-def make_api_call(task_id, audio_id, api_key, callback_url):
-    with st.spinner("🎵 Generating audio..."):
-        # Prepare the API call
-        headers = {
-            "Authorization": f"Bearer {api_key}",
-            "Content-Type": "application/json"
-        }
-        
-        data = {
-            "taskId": task_id,
-            "audioId": audio_id,
-            "callBackUrl": callback_url
-        }
-        
-        # Show request details
-        with st.expander("📡 Request Details", expanded=False):
-            st.markdown(f"""
-            **Endpoint:** `https://api.sunoapi.org/api/v1/wav/generate`
-            **Headers:** ```python
-            {json.dumps(headers, indent=2)}
-            ```
-            **Body:** ```json
-            {json.dumps(data, indent=2)}
-            ```
-            """)
-        
-        # Make the API call
-        try:
-            #response = requests.post("https://api.sunoapi.org/api/v1/wav/generate", headers=headers, json=data)
-            #st.session_state.last_api_response = response.json()
+# Make API call button
+if st.button("🎵 Generate Audio", type="primary", use_container_width=True):
+    if not SUNO_API_KEY or not CALLBACK_URL:
+        st.error("Please configure secrets in Hugging Face Space settings")
+    elif not task_id or not audio_id:
+        st.error("Please enter Task ID and Audio ID")
+    else:
+        with st.spinner("Making API call..."):
+            # Prepare the API call
+            headers = {
+                "Authorization": f"Bearer {SUNO_API_KEY}",
+                "Content-Type": "application/json"
+            }
+            
+            data = {
+                "taskId": task_id,
+                "audioId": audio_id,
+                "callBackUrl": CALLBACK_URL
+            }
             
-            # For demo purposes, return a mock response
+            # Show request info
+            with st.expander("📡 Request Details"):
+                st.code(f"""
+POST https://api.sunoapi.org/api/v1/wav/generate
+
+Headers:
+{json.dumps(headers, indent=2)}
+
+Body:
+{json.dumps(data, indent=2)}
+                """)
+            
+            # Note: In production, you would make the actual API call:
+            # import requests
+            # response = requests.post(
+            #     "https://api.sunoapi.org/api/v1/wav/generate",
+            #     headers=headers,
+            #     json=data
+            # )
+            
+            # For now, show a mock response
             mock_response = {
                 "status": "success",
-                "jobId": "generated_123456789",
-                "estimatedCompletion": "2026-01-17T20:45:00Z",
-                "message": "Audio generation started successfully"
+                "jobId": f"job_{datetime.now().strftime('%Y%m%d%H%M%S')}",
+                "message": "Audio generation started",
+                "estimatedCompletion": "2026-01-18T10:30:00Z"
             }
             
-            st.session_state.last_api_response = mock_response
-            st.success("✅ Audio generation started successfully!")
+            st.success("✅ API call successful!")
+            st.json(mock_response)
             
-            # Show job info
-            st.info(f"**Job ID:** {mock_response['jobId']}")
-            st.info(f"**Estimated Completion:** {mock_response['estimatedCompletion']}")
-            
-        except Exception as e:
-            st.error(f"❌ API call failed: {str(e)}")
+            # Download button
+            st.download_button(
+                label="📥 Download Response",
+                data=json.dumps(mock_response, indent=2),
+                file_name=f"response_{datetime.now().strftime('%Y%m%d_%H%M%S')}.json",
+                mime="application/json"
+            )
 
-# Run the main function
-if __name__ == "__main__":
-    main()
+# Secret info (hidden by default)
+with st.expander("🔐 Secret Info"):
+    if SUNO_API_KEY:
+        masked_key = f"{SUNO_API_KEY[:4]}...{SUNO_API_KEY[-4:]}" if len(SUNO_API_KEY) > 8 else "••••••••"
+        st.write(f"**API Key:** `{masked_key}`")
+    else:
+        st.write("**API Key:** Not set")
+    
+    if CALLBACK_URL:
+        st.write(f"**Callback URL:** `{CALLBACK_URL}`")
+    else:
+        st.write("**Callback URL:** Not set")