import axios from 'axios'

const api = axios.create({
  baseURL: '/api',
  timeout: 30000
})

const dockerApi = axios.create({
  baseURL: '/docker-api',
  timeout: 120000
})

// 视频检测专用，超时更长
const videoApi = axios.create({
  baseURL: '/api',
  timeout: 300000
})

export const detectionApi = {
  getModels() {
    return api.get('/models')
  },

  getAlgorithmConfig() {
    return api.get('/algorithms/config')
  },

  detectImage(formData, modelId, confidence, iou, algorithmConfig = null, regionPolygon = null) {
    const params = {
      model_id: modelId,
      confidence,
      iou
    }
    if (algorithmConfig) {
      params.algorithm_config = JSON.stringify(algorithmConfig)
    }
    if (regionPolygon) {
      params.region_polygon = JSON.stringify(regionPolygon)
    }

    return api.post('/detect/image', formData, {
      headers: {
        'Content-Type': 'multipart/form-data'
      },
      params
    })
  },

  detectVideo(formData) {
    return dockerApi.post('/api/fight_detect', formData, {
      headers: {
        'Content-Type': 'multipart/form-data'
      }
    })
  },

  // 本地 YOLO 视频检测（打架斗殴检测等）
  detectLocalVideo(formData, modelId = 'fight_detection', confidence = 0.5, iou = 0.45) {
    return videoApi.post('/detect/video', formData, {
      headers: {
        'Content-Type': 'multipart/form-data'
      },
      params: {
        model_id: modelId,
        confidence,
        iou
      }
    })
  }
}