Enhance application configuration: add MongoDB settings and directory management for audio, text, and temporary files, ensuring necessary directories are created at initialization.

app/config.py

@@ -5,8 +5,22 @@ from pydantic_settings import BaseSettings
 
 class Settings(BaseSettings):
     """Application settings."""
-    DATABASE_URL: str = os.getenv("DATABASE_URL", "sqlite:///app/storage/audiobooks.db")
-    STORAGE_PATH: str = os.getenv("STORAGE_PATH", "app/storage")
+    # MongoDB settings
+    MONGO_URI: str = os.getenv("MONGO_URI", "mongodb://localhost:27017")
+    DB_NAME: str = os.getenv("DB_NAME", "tts_api")
+    
+    # Storage paths
+    STORAGE_PATH: str = os.getenv("STORAGE_PATH", "/app/storage")
+    AUDIO_DIR: str = os.path.join(STORAGE_PATH, "audio")
+    TEXT_DIR: str = os.path.join(STORAGE_PATH, "text")
+    TEMP_DIR: str = os.path.join(STORAGE_PATH, "temp")
+    
+    # Ensure directories exist
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+        os.makedirs(self.AUDIO_DIR, exist_ok=True)
+        os.makedirs(self.TEXT_DIR, exist_ok=True)
+        os.makedirs(self.TEMP_DIR, exist_ok=True)
     
     class Config:
         """Pydantic config."""
@@ -15,4 +29,12 @@ class Settings(BaseSettings):
 @lru_cache()
 def get_settings() -> Settings:
     """Get cached settings."""
-    return Settings()
+    return Settings()
+
+# Create settings instance
+settings = get_settings()
+
+# Export directory paths
+AUDIO_DIR = settings.AUDIO_DIR
+TEXT_DIR = settings.TEXT_DIR
+TEMP_DIR = settings.TEMP_DIR

requirements.txt

@@ -1,6 +1,7 @@
 fastapi>=0.100.0
 uvicorn>=0.24.0
 pydantic>=2.4.0
+pydantic-settings>=2.1.0
 python-multipart>=0.0.6
 huggingface_hub>=0.20.0
 torch>=2.0.0