app/main.py

# --- Model download logic (Hugging Face Hub) ---
import os
import requests

def download_if_missing(url, dest):
    if not os.path.exists(dest):
        print(f"Downloading model from {url} to {dest}...")
        os.makedirs(os.path.dirname(dest), exist_ok=True)
        with requests.get(url, stream=True) as r:
            r.raise_for_status()
            with open(dest, "wb") as f:
                for chunk in r.iter_content(chunk_size=8192):
                    f.write(chunk)
        print("Download complete.")

# Hugging Face direct download links
# DAMAGE_MODEL_URL = "https://huggingface.co/AItoolstack/car_damage_detection/resolve/main/yolov8_models/damage/weights/weights/best.pt"
PARTS_MODEL_URL = "https://huggingface.co/AItoolstack/car_damage_detection/resolve/main/yolov8_models/parts/weights/weights/best.pt"

# DAMAGE_MODEL_PATH = os.path.join("/tmp", "models", "damage", "weights", "weights", "best.pt")
PARTS_MODEL_PATH = os.path.join("/tmp", "models", "parts", "weights", "weights", "best.pt")

# download_if_missing(DAMAGE_MODEL_URL, DAMAGE_MODEL_PATH)
download_if_missing(PARTS_MODEL_URL, PARTS_MODEL_PATH)

from fastapi import FastAPI, File, UploadFile, BackgroundTasks
from fastapi.staticfiles import StaticFiles
from fastapi.templating import Jinja2Templates
from fastapi.responses import JSONResponse, HTMLResponse
from fastapi.middleware.cors import CORSMiddleware
import uvicorn
from datetime import datetime
import aiofiles
from pathlib import Path
import uuid
from app.routers import inference
from .utils.model_loader import PARTS_MODEL_PATH

app = FastAPI(title="Car Damage Detection API")

# CORS middleware
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

# Get absolute paths for static and template directories
APP_DIR = Path(__file__).resolve().parent
STATIC_DIR = APP_DIR / "static"
TEMPLATES_DIR = APP_DIR / "templates"

# Mount static files using absolute path
app.mount("/static", StaticFiles(directory=str(STATIC_DIR)), name="static")
templates = Jinja2Templates(directory=str(TEMPLATES_DIR))

# Include routers
app.include_router(inference.router)

# Create required directories with absolute paths
UPLOAD_DIR = Path("/tmp/uploads")
RESULTS_DIR = Path("/tmp/results")

def create_dirs():
    # Ensure directories exist with proper permissions
    for dir_path in [UPLOAD_DIR, RESULTS_DIR]:
        try:
            dir_path.mkdir(parents=True, exist_ok=True)
            # Set permissions to 777 to ensure write access
            os.chmod(str(dir_path), 0o777)
        except Exception as e:
            print(f"Error creating directory {dir_path}: {e}")
            raise

# Add to startup
@app.on_event("startup")
async def startup_event():
    create_dirs()

if __name__ == "__main__":
    uvicorn.run("main:app", host="0.0.0.0", port=7860)