audial-sdk 1.0.0__py3-none-any.whl

audial/__init__.py

@@ -0,0 +1,32 @@
+"""
+Audial SDK: A Python package for interacting with the Audial audio processing API.
+"""
+
+__version__ = "1.0.0"
+
+# Import and re-export public functions
+from audial.functions.stem_split import stem_split
+from audial.functions.analyze import analyze
+from audial.functions.segment import segment
+from audial.functions.master import master
+from audial.functions.samples import generate_samples
+from audial.functions.midi import generate_midi
+
+# Import config module
+from audial.utils import config
+
+# Import exceptions
+from audial.api.exceptions import AudialError, AudialAuthError, AudialAPIError
+
+__all__ = [
+    "stem_split",
+    "analyze",
+    "segment",
+    "master",
+    "generate_samples",
+    "generate_midi",
+    "config",
+    "AudialError",
+    "AudialAuthError",
+    "AudialAPIError",
+]

audial/api/__init__.py

@@ -0,0 +1,15 @@
+"""
+API client for the Audial SDK.
+"""
+
+from audial.api.proxy import AudialProxy
+from audial.api.auth import get_auth_headers
+from audial.api.exceptions import AudialError, AudialAuthError, AudialAPIError
+
+__all__ = [
+    "AudialProxy",
+    "get_auth_headers",
+    "AudialError",
+    "AudialAuthError",
+    "AudialAPIError",
+]

audial/api/auth.py

@@ -0,0 +1,117 @@
+"""
+Authentication utilities for the Audial SDK.
+"""
+
+from typing import Dict
+import uuid
+import os
+import json
+
+from audial.api.exceptions import AudialAuthError
+from audial.utils.config import get_user_id
+
+def load_dotenv():
+    """
+    Load environment variables from .env file.
+    """
+    try:
+        # Check if .env file exists
+        if os.path.exists('.env'):
+            with open('.env', 'r') as f:
+                for line in f:
+                    line = line.strip()
+                    if line and not line.startswith('#'):
+                        key, value = line.split('=', 1)
+                        # Remove quotes if present
+                        value = value.strip().strip('"\'')
+                        os.environ[key.strip()] = value
+    except Exception as e:
+        print(f"Warning: Could not load .env file: {str(e)}")
+
+def get_api_key():
+    """
+    Get the API key from environment or config file.
+    
+    Returns:
+        str: The API key.
+        
+    Raises:
+        AudialAuthError: If no API key is available.
+    """
+    # Try to load from .env file first
+    load_dotenv()
+    
+    # Check environment variable
+    api_key = os.environ.get("AUDIAL_API_KEY")
+    
+    if not api_key:
+        # Check config file
+        config_dir = os.path.expanduser("~/.audial")
+        config_file = os.path.join(config_dir, "config.json")
+        
+        if os.path.exists(config_file):
+            try:
+                with open(config_file, 'r') as f:
+                    config = json.load(f)
+                    api_key = config.get("api_key")
+            except Exception as e:
+                pass
+    
+    # If still no API key, raise an error
+    if not api_key:
+        raise AudialAuthError(
+            "No API key available. Please set your API key using "
+            "audial.config.set_api_key() or by setting the AUDIAL_API_KEY "
+            "environment variable."
+        )
+    
+    return api_key
+
+def get_auth_headers(api_key: str = None) -> Dict[str, str]:
+    """
+    Get the authentication headers for API requests.
+    
+    Args:
+        api_key (str, optional): The API key to use. If not provided, it will be loaded from the configuration.
+        
+    Returns:
+        Dict[str, str]: The authentication headers.
+        
+    Raises:
+        AudialAuthError: If no API key is available.
+    """
+    # Get API key from config if not provided
+    if api_key is None:
+        api_key = get_api_key()
+    
+    # Check if API key is available
+    if not api_key:
+        raise AudialAuthError(
+            "No API key available. Please set your API key using "
+            "audial.config.set_api_key() or by setting the AUDIAL_API_KEY "
+            "environment variable."
+        )
+    
+    # Get user ID from configuration
+    user_id = get_user_id()
+    if not user_id:
+        raise AudialAuthError(
+            "No user ID available. Please set your user ID using "
+            "audial.config.set_user_id() or by setting the AUDIAL_USER_ID "
+            "environment variable."
+        )
+    
+    # Generate request ID for tracking
+    request_id = str(uuid.uuid4())
+    
+    # Return headers matching exactly what the frontend sends
+    # This is critical - including both x-api-key and Authorization headers,
+    # as well as the required x-user-id header
+    # print(f"Using API Key: {api_key}, Using x-user-id: {user_id}")
+    return {
+        # "Authorization": f"Bearer {api_key}",
+        "x-api-key": api_key,
+        "x-user-id": user_id,  
+        "Content-Type": "application/json",
+        "X-Request-ID": request_id
+    }

audial/api/constants.py

@@ -0,0 +1,59 @@
+"""
+Constants for the Audial API.
+"""
+
+# Base API URL - using the auth server as proxy
+# This endpoint will be responsible for routing to the actual API endpoints
+API_BASE_URL = "https://starfish-app-2x28e.ondigitalocean.app/api"
+AUTH_SERVER_URL = "https://starfish-app-2x28e.ondigitalocean.app/api/proxy"
+
+# Function names as defined in the API
+FUNCTION_STEM_SPLITTER = "stem-splitter"
+FUNCTION_PRIMARY_ANALYSIS = "primary-analysis"
+FUNCTION_SEGMENTATION = "segmentation"
+FUNCTION_MASTERING = "mastering"
+FUNCTION_SAMPLE_PACK = "sample-pack"
+FUNCTION_GENERATE_MIDI = "generate-midi"
+
+# Execution types
+EXECUTION_TYPE_STEM = "stem"
+EXECUTION_TYPE_MODIFIED = "modified"
+EXECUTION_TYPE_MASTER = "master"
+EXECUTION_TYPE_MIDI = "midi"
+EXECUTION_TYPE_SAMPLES = "samples"
+EXECUTION_TYPE_SEGMENTATION = "segmentation"
+
+# Execution states
+EXECUTION_STATE_CREATED = "created"
+EXECUTION_STATE_INITIALIZED = "initialized"
+EXECUTION_STATE_PROCESSING = "processing"
+EXECUTION_STATE_COMPLETED = "completed"
+EXECUTION_STATE_FAILED = "failed"
+
+# Default stem options
+DEFAULT_STEM_OPTIONS = ["vocals", "drums", "bass", "other"]
+
+# Available stem options
+ALL_STEM_OPTIONS = [
+    "vocals", 
+    "drums", 
+    "bass", 
+    "other",
+    "full_song_without_vocals",
+    "full_song_without_drums",
+    "full_song_without_bass",
+    "full_song_without_other"
+]
+
+# Default polling interval for checking execution status (in seconds)
+DEFAULT_POLLING_INTERVAL = 2
+
+# Maximum number of retries for API requests
+MAX_RETRIES = 3
+
+# Timeout for API requests (in seconds)
+REQUEST_TIMEOUT = 60
+
+# Default values for segmentation
+DEFAULT_SEGMENTATION_COMPONENTS = ["intro", "verse", "chorus", "outro"]
+DEFAULT_SEGMENTATION_FEATURES = ["energy", "tempo", "loudness"]

audial/api/exceptions.py

@@ -0,0 +1,18 @@
+"""
+Exceptions for the Audial SDK.
+"""
+
+class AudialError(Exception):
+    """Base exception for all Audial SDK errors."""
+    pass
+
+class AudialAuthError(AudialError):
+    """Exception raised for authentication errors."""
+    pass
+
+class AudialAPIError(AudialError):
+    """Exception raised for API errors."""
+    def __init__(self, message, status_code=None, response=None):
+        self.status_code = status_code
+        self.response = response
+        super().__init__(message)