`react-dash/src/setup/steps/RunPredictions.jsx`

119 lines
import React, { useState, useEffect } from 'react';
import { post } from '../../api';

export default function RunPredictions({ data, update, next, back }) {
  const [loading, setLoading] = useState(true);
  const [result, setResult] = useState(null);
  const [error, setError] = useState(null);

  useEffect(() => {
    runPredictions();
  }, []);

  async function runPredictions() {
    setLoading(true);
    setError(null);
    try {
      const r = await post('/models', { action: 'run_predictions' });
      setResult(r);
      update({ predictionsRun: true });
    } catch {
      setError('Failed to generate predictions');
    }
    setLoading(false);
  }

  const predictions = result?.predictions || [];
  const buySignals = predictions.filter(p => p.signal === 'BUY' || p.edge > 0);

  return (
    <div className="space-y-6">
      <div className="text-center">
        <h2 className="text-2xl font-bold mb-2">AI Predictions</h2>
        <p className="text-gray-400 text-sm">Generating probability models and calculating edges</p>
      </div>

      <div className="card">
        {loading ? (
          <div className="flex flex-col items-center justify-center py-12 gap-4">
            <div className="animate-spin w-10 h-10 border-3 border-green-500 border-t-transparent rounded-full" />
            <p className="text-sm text-gray-400">Running prediction models...</p>
          </div>
        ) : error ? (
          <div className="text-center py-8">
            <div className="w-12 h-12 rounded-full bg-red-500/10 flex items-center justify-center mx-auto mb-3">
              <svg className="w-6 h-6 text-red-400" fill="none" stroke="currentColor" strokeWidth="1.5" viewBox="0 0 24 24">
                <path strokeLinecap="round" strokeLinejoin="round" d="M12 9v3.75m9-.75a9 9 0 11-18 0 9 9 0 0118 0zm-9 3.75h.008v.008H12v-.008z" />
              </svg>
            </div>
            <p className="text-red-400 text-sm mb-3">{error}</p>
            <button onClick={runPredictions} className="btn btn-s text-sm">Retry</button>
          </div>
        ) : (
          <div className="space-y-4">
            <div className="grid grid-cols-3 gap-4">
              <div className="text-center">
                <p className="stat-v text-purple-400">{predictions.length}</p>
                <p className="text-xs text-gray-500 mt-1">Predictions</p>
              </div>
              <div className="text-center">
                <p className="stat-v text-green-400">{buySignals.length}</p>
                <p className="text-xs text-gray-500 mt-1">Buy Signals</p>
              </div>
              <div className="text-center">
                <p className="stat-v text-blue-400">{result?.edges_found || buySignals.length}</p>
                <p className="text-xs text-gray-500 mt-1">Edges Found</p>
              </div>
            </div>

            {predictions.length > 0 && (
              <div className="border-t border-gray-800 pt-3">
                <p className="text-xs text-gray-500 mb-2 uppercase font-medium">Top Predictions</p>
                <table>
                  <thead>
                    <tr>
                      <th>Market</th>
                      <th>P(Yes)</th>
                      <th>Edge</th>
                      <th>Signal</th>
                    </tr>
                  </thead>
                  <tbody>
                    {predictions.slice(0, 6).map((p, i) => (
                      <tr key={i}>
                        <td className="max-w-[200px]"><span className="truncate block text-xs">{p.question || p.market}</span></td>
                        <td className="font-mono text-sm">{((p.p_yes || p.probability || 0) * 100).toFixed(1)}%</td>
                        <td className={`font-mono text-sm ${(p.edge || 0) > 0 ? 'text-green-400' : 'text-gray-500'}`}>
                          {((p.edge || 0) * 100).toFixed(1)}%
                        </td>
                        <td>
                          <span className={`badge ${p.signal === 'BUY' || (p.edge || 0) > 0 ? 'bg-green' : 'bg-red'}`}>
                            {p.signal || ((p.edge || 0) > 0 ? 'BUY' : 'HOLD')}
                          </span>
                        </td>
                      </tr>
                    ))}
                  </tbody>
                </table>
              </div>
            )}

            {predictions.length === 0 && !error && (
              <div className="text-center py-4">
                <p className="text-gray-500 text-sm">No predictions generated yet — models may need more market data</p>
              </div>
            )}
          </div>
        )}
      </div>

      <div className="flex items-center justify-between">
        <button onClick={back} className="text-sm text-gray-500 hover:text-gray-300">Back</button>
        <button onClick={next} disabled={loading} className="wizard-btn">
          {loading ? 'Processing...' : 'Continue'}
        </button>
      </div>
    </div>
  );
}