Spaces:
Sleeping
Sleeping
File size: 1,590 Bytes
c840ad0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
"""Storage service for managing generated models."""
from typing import Dict, Optional
from datetime import datetime
import uuid
class StorageService:
"""In-memory storage service for generated models.
In production, this would be replaced with a database like PostgreSQL or MongoDB.
"""
def __init__(self):
self._models: Dict[str, dict] = {}
def create_model_record(self, prompt: str) -> str:
"""Create a new model record and return its ID."""
model_id = str(uuid.uuid4())
self._models[model_id] = {
"model_id": model_id,
"prompt": prompt,
"created_at": datetime.now().isoformat(),
"status": "processing",
"file_path": None,
"available_formats": []
}
return model_id
def get_model(self, model_id: str) -> Optional[dict]:
"""Get a model by ID."""
return self._models.get(model_id)
def update_model_status(self, model_id: str, status: str):
"""Update the status of a model."""
if model_id in self._models:
self._models[model_id]["status"] = status
def set_model_file(self, model_id: str, file_path: str, fmt: str = "glb"):
"""Set the file path and available format for a model."""
if model_id in self._models:
self._models[model_id]["file_path"] = file_path
formats = set(self._models[model_id].get("available_formats", []))
formats.add(fmt)
self._models[model_id]["available_formats"] = list(formats)
|