import MultiSourceCaptioningView from "./components/MultiSourceCaptioningView";
import { useVLMContext } from "./context/useVLMContext";
import { useState } from "react";

export default function App() {
  const { isLoaded, isLoading, error, loadModel } = useVLMContext();
  const [started, setStarted] = useState(false);

  const handleLoadModel = async () => {
    try {
      await loadModel();
      setStarted(true);
    } catch (e) {
      // error is handled by context
    }
  };

  if (!started || !isLoaded) {
    return (
      <div className="flex flex-col items-center justify-center h-screen bg-gray-900 text-white">
        <h1 className="text-2xl font-bold mb-4">FastVLM WebGPU</h1>
        <button
          className="px-6 py-3 rounded-lg bg-blue-600 text-white font-semibold text-lg mb-4"
          onClick={handleLoadModel}
          disabled={isLoading}
        >
          {isLoading ? "Loading Model..." : "Load Model"}
        </button>
        {error && <div className="text-red-400 mt-2">Model error: {error}</div>}
      </div>
    );
  }

  return (
    <div className="App relative h-screen overflow-hidden">
      <div className="absolute inset-0 bg-gray-900" />
      <MultiSourceCaptioningView />
    </div>
  );
}