Add download lora

utils/model_utils.py

@@ -7,6 +7,7 @@ BASE_MODEL_MAPPING = {
     "sdxl": "SDXL",
     "sd-v2": "SD2.0",
     "flux1": "Flux.1 D",
+    "Illustrious": "IL"
 }
 
 def determine_base_model(version_string: Optional[str]) -> str:

utils/models.py

@@ -1,6 +1,8 @@
 from dataclasses import dataclass, asdict
 from typing import Dict, Optional
 from datetime import datetime
+import os
+from .model_utils import determine_base_model
 
 @dataclass
 class LoraMetadata:
@@ -23,6 +25,25 @@ class LoraMetadata:
         data_copy = data.copy()
         return cls(**data_copy)
 
+    @classmethod
+    def from_civitai_info(cls, version_info: Dict, file_info: Dict, save_path: str) -> 'LoraMetadata':
+        """Create LoraMetadata instance from Civitai version info"""
+        file_name = file_info['name']
+        base_model = determine_base_model(version_info.get('baseModel', ''))
+        
+        return cls(
+            file_name=os.path.splitext(file_name)[0],
+            model_name=version_info.get('model').get('name', os.path.splitext(file_name)[0]),
+            file_path=save_path.replace(os.sep, '/'),
+            size=file_info.get('sizeKB', 0) * 1024,
+            modified=datetime.now().timestamp(),
+            sha256=file_info['hashes'].get('SHA256', ''),
+            base_model=base_model,
+            preview_url=None,  # Will be updated after preview download
+            from_civitai=True,
+            civitai=version_info
+        )
+
     def to_dict(self) -> Dict:
         """Convert to dictionary for JSON serialization"""
         return asdict(self)
@@ -36,4 +57,10 @@ class LoraMetadata:
         """Update Civitai information"""
         self.civitai = civitai_data
 
-     
+    def update_file_info(self, file_path: str) -> None:
+        """Update metadata with actual file information"""
+        if os.path.exists(file_path):
+            self.size = os.path.getsize(file_path)
+            self.modified = os.path.getmtime(file_path)
+            self.file_path = file_path.replace(os.sep, '/')
+