Implement early access handling and UI enhancements for LoRA downloads

- Added error handling for early access restrictions in the API routes, returning appropriate status codes and messages.
- Enhanced the Civitai client to log unauthorized access attempts and provide user-friendly error messages.
- Updated the download manager to check for early access requirements and log warnings accordingly.
- Introduced UI elements to indicate early access status for LoRAs, including badges and warning messages in the import manager.
- Improved toast notifications to inform users about early access download failures and provide relevant information.

py/services/civitai_client.py

@@ -76,6 +76,17 @@ class CivitaiClient:
             headers = self._get_request_headers()
             async with session.get(url, headers=headers, allow_redirects=True) as response:
                 if response.status != 200:
+                    # Handle early access 401 unauthorized responses
+                    if response.status == 401:
+                        logger.warning(f"Unauthorized access to resource: {url} (Status 401)")
+                        return False, "Early access restriction: You must purchase early access to download this LoRA."
+                    
+                    # Handle other client errors that might be permission-related
+                    if response.status == 403:
+                        logger.warning(f"Forbidden access to resource: {url} (Status 403)")
+                        return False, "Access forbidden: You don't have permission to download this file."
+                    
+                    # Generic error response for other status codes
                     return False, f"Download failed with status {response.status}"
 
                 # Get filename from content-disposition header

py/services/download_manager.py

@@ -28,6 +28,25 @@ class DownloadManager:
             if not version_info:
                 return {'success': False, 'error': 'Failed to fetch model metadata'}
 
+            # Check if this is an early access LoRA
+            if 'earlyAccessEndsAt' in version_info:
+                early_access_date = version_info.get('earlyAccessEndsAt', '')
+                # Convert to a readable date if possible
+                try:
+                    from datetime import datetime
+                    date_obj = datetime.fromisoformat(early_access_date.replace('Z', '+00:00'))
+                    formatted_date = date_obj.strftime('%Y-%m-%d')
+                    early_access_msg = f"This LoRA requires early access payment (until {formatted_date}). "
+                except:
+                    early_access_msg = "This LoRA requires early access payment. "
+                
+                early_access_msg += "Please ensure you have purchased early access and are logged in to Civitai."
+                logger.warning(f"Early access LoRA detected: {version_info.get('name', 'Unknown')}")
+                
+                # We'll still try to download, but log a warning and prepare for potential failure
+                if progress_callback:
+                    await progress_callback(1)  # Show minimal progress to indicate we're trying
+
             # Report initial progress
             if progress_callback:
                 await progress_callback(0)
@@ -82,6 +101,10 @@ class DownloadManager:
 
         except Exception as e:
             logger.error(f"Error in download_from_civitai: {e}", exc_info=True)
+            # Check if this might be an early access error
+            error_str = str(e).lower()
+            if "403" in error_str or "401" in error_str or "unauthorized" in error_str or "early access" in error_str:
+                return {'success': False, 'error': f"Early access restriction: {str(e)}. Please ensure you have purchased early access and are logged in to Civitai."}
             return {'success': False, 'error': str(e)}
 
     async def _execute_download(self, download_url: str, save_dir: str,