defmodule MLLearning.MixProject do
  use Mix.Project

  def project do
    [
      app: :ml_elixir_learning,
      version: "0.1.0",
      elixir: "~> 1.16",
      start_permanent: Mix.env() == :prod,
      deps: deps(),
      aliases: aliases()
    ]
  end

  def application do
    [
      extra_applications: [:logger]
    ]
  end

  defp deps do
    [
      # Core ML
      {:nx, "~> 0.10"},
      {:axon, "~> 0.7"},
      {:exla, "~> 0.10"},

      # Pre-trained models (Hugging Face Hub)
      {:bumblebee, "~> 0.6"},

      # Data
      {:explorer, "~> 0.10"},
      {:scidata, "~> 0.1"},

      # Image / Audio IO
      {:stb_image, "~> 0.6"},

      # Visualization
      {:kino, "~> 0.15"},
      {:kino_vega_lite, "~> 0.1"},
      {:vega_lite, "~> 0.1"},

      # HTTP
      {:req, "~> 0.5"},

      # Web framework (for LiveView deployment section)
      {:phoenix, "~> 1.7"},
      {:phoenix_live_view, "~> 1.0"},
      {:bandit, "~> 1.0"},

      # Dev / Test
      {:benchee, "~> 1.0", only: :dev},
      {:ex_doc, "~> 0.34", only: :dev, runtime: false}
    ]
  end

  defp aliases do
    [
      setup: ["deps.get", "deps.compile"],
      nb: ["livebook", "server"]
    ]
  end
end