updated the path to the APP_DIR in the main file

app/main.py

@@ -54,18 +54,19 @@ async def lifespan(app: FastAPI):
     app.state.logger = logger  # Make logger available to routes
     
     # Create necessary directories - use persistent locations
-    os.makedirs("/app/models", exist_ok=True)
-    os.makedirs("/app/tokenizers", exist_ok=True)
-    os.makedirs("/app/voice_memories", exist_ok=True)
-    os.makedirs("/app/voice_references", exist_ok=True)
-    os.makedirs("/app/voice_profiles", exist_ok=True)
-    os.makedirs("/app/cloned_voices", exist_ok=True)
-    os.makedirs("/app/audio_cache", exist_ok=True)
-    os.makedirs("/app/static", exist_ok=True)
+    APP_DIR = os.path.join(os.environ['HOME'], 'app')
+    os.makedirs(os.path.join(APP_DIR, "models"), exist_ok=True)
+    os.makedirs(os.path.join(APP_DIR, "tokenizers"), exist_ok=True)
+    os.makedirs(os.path.join(APP_DIR, "voice_memories"), exist_ok=True)
+    os.makedirs(os.path.join(APP_DIR, "voice_references"), exist_ok=True)
+    os.makedirs(os.path.join(APP_DIR, "voice_profiles"), exist_ok=True)
+    os.makedirs(os.path.join(APP_DIR, "cloned_voices"), exist_ok=True)
+    os.makedirs(os.path.join(APP_DIR, "audio_cache"), exist_ok=True)
+    os.makedirs(os.path.join(APP_DIR, "static"), exist_ok=True)
     
     # Set tokenizer cache
     try:
-        os.environ["TRANSFORMERS_CACHE"] = "/app/tokenizers"
+        os.environ["TRANSFORMERS_CACHE"] = os.path.join(APP_DIR, "tokenizers")
         logger.info(f"Set tokenizer cache to: {os.environ['TRANSFORMERS_CACHE']}")
     except Exception as e:
         logger.error(f"Error setting tokenizer cache: {e}")
@@ -109,7 +110,7 @@ async def lifespan(app: FastAPI):
     app.state.device_map = device_map
     
     # Check if model file exists
-    model_path = os.path.join("/app/models", "ckpt.pt")
+    model_path = os.path.join(APP_DIR, "models", "ckpt.pt")
     if not os.path.exists(model_path):
         # Try to download at runtime if not present
         logger.info("Model not found. Attempting to download...")
@@ -125,7 +126,7 @@ async def lifespan(app: FastAPI):
             model_path = hf_hub_download(
                 repo_id="sesame/csm-1b", 
                 filename="ckpt.pt", 
-                local_dir="/app/models"
+                local_dir=APP_DIR
             )
             download_time = time.time() - download_start
             logger.info(f"Model downloaded to {model_path} in {download_time:.2f} seconds")
@@ -180,7 +181,7 @@ async def lifespan(app: FastAPI):
             logger.info("Initializing voice cloning system...")
             from app.voice_cloning import VoiceCloner, CLONED_VOICES_DIR
             # Update the cloned voices directory to use the persistent volume
-            app.state.cloned_voices_dir = "/app/cloned_voices"  # Store path in app state for access
+            app.state.cloned_voices_dir = os.path.join(APP_DIR, "cloned_voices")  # Store path in app state for access
             os.makedirs(app.state.cloned_voices_dir, exist_ok=True)
             CLONED_VOICES_DIR = app.state.cloned_voices_dir  # Update the module constant
             
@@ -362,7 +363,7 @@ async def lifespan(app: FastAPI):
         # Set up audio cache
         app.state.audio_cache_enabled = os.environ.get("ENABLE_AUDIO_CACHE", "true").lower() == "true"
         if app.state.audio_cache_enabled:
-            app.state.audio_cache_dir = "/app/audio_cache"
+            app.state.audio_cache_dir = os.path.join(APP_DIR, "audio_cache")
             logger.info(f"Audio cache enabled, cache dir: {app.state.audio_cache_dir}")
         
         # Log GPU utilization after model loading
@@ -497,11 +498,12 @@ app.add_middleware(
 )
 
 # Create static and other required directories
-os.makedirs("/app/static", exist_ok=True)
-os.makedirs("/app/cloned_voices", exist_ok=True)
+APP_DIR = os.path.join(os.environ['HOME'], 'app')
+os.makedirs(os.path.join(APP_DIR, "static"), exist_ok=True)
+os.makedirs(os.path.join(APP_DIR, "cloned_voices"), exist_ok=True)
 
 # Mount the static files directory
-app.mount("/static", StaticFiles(directory="/app/static"), name="static")
+app.mount("/static", StaticFiles(directory=os.path.join(APP_DIR, "static")), name="static")
 
 # Include routers
 app.include_router(api_router, prefix="/api/v1")
@@ -588,13 +590,13 @@ async def version():
 @app.get("/voice-cloning", include_in_schema=False)
 async def voice_cloning_ui():
     """Voice cloning UI endpoint."""
-    return FileResponse("/app/static/voice-cloning.html")
+    return FileResponse(os.path.join(APP_DIR, "static/voice-cloning.html"))
 
 # Streaming demo endpoint
 @app.get("/streaming-demo", include_in_schema=False)
 async def streaming_demo():
     """Streaming TTS demo endpoint."""
-    return FileResponse("/app/static/streaming-demo.html")
+    return FileResponse(os.path.join(APP_DIR, "static/streaming-demo.html"))
 
 @app.get("/", include_in_schema=False)
 async def root():