import React, { useState, useEffect } from 'react';
import { motion } from 'framer-motion';
import { TrendingUp } from 'lucide-react';

interface LearningData {
  learningSystem: {
    enabled: boolean;
    learningActive?: boolean;
    activeDecisions?: number;
    message?: string;
    recommendation?: string;
    persistentStats?: {
      totalTrades: number;
      successRate: number;
      totalPnl: number;
      winRate: number;
    };
    recentTrades?: Array<{
      symbol: string;
      type: string;
      pnl: string;
      updatedAt: string;
    }>;
    systemHealth?: {
      totalDecisions: number;
      recentDecisions: number;
      lastActivity: string;
    };
    report?: {
      summary?: {
        totalDecisions?: number;
        successRate?: number;
        systemConfidence?: number;
      };
      insights?: {
        thresholds?: any;
        confidenceLevel?: number;
      };
      recommendations?: Array<{
        type: string;
        message: string;
        priority: string;
      } | string>;
    };
  };
  visibility?: {
    decisionTrackingActive?: boolean;
    learningDatabaseConnected?: boolean;
    aiEnhancementsActive?: boolean;
    lastUpdateTime?: string;
  };
  automationStatus?: any;
}

const EnhancedAILearningPanel = () => {
  const [learningData, setLearningData] = useState<LearningData | null>(null);
  const [loading, setLoading] = useState(true);
  const [error, setError] = useState<string | null>(null);

  const fetchLearningStatus = async () => {
    try {
      setLoading(true);
      
      // Get both learning status and persistent data (regardless of automation status)
      const [learningResponse, statusResponse, persistentResponse] = await Promise.all([
        fetch('/api/automation/learning-status'),
        fetch('/api/automation/status'),
        fetch('/api/learning/persistent-status')
      ]);
      
      const learningData = await learningResponse.json();
      const statusData = await statusResponse.json();
      const persistentData = await persistentResponse.json();
      
      // Merge persistent data with current learning status
      const safeData = {
        learningSystem: {
          ...learningData.learningSystem,
          // Always include persistent statistics
          persistentStats: persistentData.success ? persistentData.statistics : null,
          recentTrades: persistentData.success ? persistentData.recentTrades : [],
          systemHealth: persistentData.success ? persistentData.systemHealth : null
        },
        visibility: learningData.visibility || {
          decisionTrackingActive: false,
          learningDatabaseConnected: persistentData.success,
          aiEnhancementsActive: false,
          lastUpdateTime: new Date().toISOString()
        },
        automationStatus: statusData
      };
      
      setLearningData(safeData);
      setError(null);
    } catch (err) {
      console.error('Error fetching learning status:', err);
      setError(err instanceof Error ? err.message : 'Unknown error');
      
      // Set default data structure on error
      setLearningData({
        learningSystem: {
          enabled: false,
          message: 'Failed to fetch learning status',
          activeDecisions: 0
        },
        visibility: {
          decisionTrackingActive: false,
          learningDatabaseConnected: false,
          aiEnhancementsActive: false,
          lastUpdateTime: new Date().toISOString()
        },
        automationStatus: null
      });
    } finally {
      setLoading(false);
    }
  };

  useEffect(() => {
    fetchLearningStatus();
    
    // Refresh every 30 seconds
    const interval = setInterval(fetchLearningStatus, 30000);
    return () => clearInterval(interval);
  }, []);

  if (loading) {
    return (
      <div className="bg-gradient-to-br from-purple-900/20 to-blue-900/20 rounded-xl p-6 border border-purple-500/30">
        <div className="flex items-center space-x-3 mb-4">
          <div className="w-3 h-3 bg-purple-500 rounded-full animate-pulse"></div>
          <h3 className="text-lg font-semibold bg-gradient-to-r from-purple-400 to-blue-400 bg-clip-text text-transparent">
            🧠 AI Learning System
          </h3>
        </div>
        <div className="text-gray-400">Loading learning status...</div>
      </div>
    );
  }

  if (error) {
    return (
      <div className="bg-gradient-to-br from-red-900/20 to-orange-900/20 rounded-xl p-6 border border-red-500/30">
        <div className="flex items-center space-x-3 mb-4">
          <div className="w-3 h-3 bg-red-500 rounded-full"></div>
          <h3 className="text-lg font-semibold text-red-400">
            🧠 AI Learning System
          </h3>
        </div>
        <div className="text-red-400">Error: {error}</div>
        <button 
          onClick={fetchLearningStatus}
          className="mt-3 px-4 py-2 bg-red-600/20 hover:bg-red-600/30 text-red-400 rounded-lg transition-colors"
        >
          Retry
        </button>
      </div>
    );
  }

  if (!learningData) {
    return null;
  }

  const { learningSystem, visibility } = learningData;

  // Safety check for learningSystem
  if (!learningSystem) {
    return (
      <div className="bg-gradient-to-br from-gray-900/20 to-gray-800/20 rounded-xl p-6 border border-gray-500/30">
        <div className="flex items-center space-x-3 mb-4">
          <div className="w-3 h-3 bg-gray-500 rounded-full"></div>
          <h3 className="text-lg font-semibold bg-gradient-to-r from-purple-400 to-blue-400 bg-clip-text text-transparent">
            🧠 AI Learning System
          </h3>
        </div>
        <div className="text-gray-400">Loading learning system data...</div>
      </div>
    );
  }

  const renderLearningStatus = () => {
    if (!learningSystem || !learningSystem.enabled) {
      return (
        <div className="space-y-4">
          <div className="flex items-center space-x-2">
            <div className="w-3 h-3 bg-yellow-500 rounded-full"></div>
            <span className="text-yellow-400 font-medium">Learning System Not Active</span>
          </div>
          
          <div className="bg-yellow-900/20 rounded-lg p-4 border border-yellow-500/30">
            <div className="text-yellow-300 text-sm">
              {learningSystem?.message || 'The AI learning system is not currently integrated with the automation.'}
            </div>
            {learningSystem?.recommendation && (
              <div className="text-yellow-400 text-sm mt-2 font-medium">
                💡 {learningSystem.recommendation}
              </div>
            )}
          </div>
          
          <div className="text-gray-400 text-sm">
            • Decision recording: Not active<br/>
            • Learning database: Not connected<br/>
            • AI enhancements: Not applied
          </div>
        </div>
      );
    }

    return (
      <div className="space-y-4">
        <div className="flex items-center space-x-2">
          <div className="w-3 h-3 bg-green-500 rounded-full animate-pulse"></div>
          <span className="text-green-400 font-medium">AI Learning Active</span>
        </div>
        
        {learningSystem?.report && (
          <div className="grid grid-cols-1 md:grid-cols-3 gap-4">
            <div className="bg-blue-900/20 rounded-lg p-4 border border-blue-500/30">
              <div className="text-blue-300 text-sm font-medium">Total Decisions</div>
              <div className="text-2xl font-bold text-blue-400">
                {learningSystem.report.summary?.totalDecisions || 0}
              </div>
            </div>
            
            <div className="bg-purple-900/20 rounded-lg p-4 border border-purple-500/30">
              <div className="text-purple-300 text-sm font-medium">Success Rate</div>
              <div className="text-2xl font-bold text-purple-400">
                {learningSystem.report.summary?.successRate || 0}%
              </div>
            </div>
            
            <div className="bg-green-900/20 rounded-lg p-4 border border-green-500/30">
              <div className="text-green-300 text-sm font-medium">Confidence</div>
              <div className="text-2xl font-bold text-green-400">
                {learningSystem.report.summary?.systemConfidence || 0}%
              </div>
            </div>
          </div>
        )}
        
        <div className="bg-gray-800/50 rounded-lg p-4 border border-gray-600/30">
          <div className="text-gray-300 text-sm font-medium mb-2">🔍 Learning Visibility</div>
          <div className="grid grid-cols-2 gap-2 text-sm">
            <div className="flex items-center space-x-2">
              <div className={`w-2 h-2 rounded-full ${visibility?.decisionTrackingActive ? 'bg-green-500' : 'bg-gray-500'}`}></div>
              <span className={visibility?.decisionTrackingActive ? 'text-green-400' : 'text-gray-400'}>
                Decision Tracking
              </span>
            </div>
            <div className="flex items-center space-x-2">
              <div className={`w-2 h-2 rounded-full ${visibility?.learningDatabaseConnected ? 'bg-green-500' : 'bg-gray-500'}`}></div>
              <span className={visibility?.learningDatabaseConnected ? 'text-green-400' : 'text-gray-400'}>
                Database Connected
              </span>
            </div>
            <div className="flex items-center space-x-2">
              <div className={`w-2 h-2 rounded-full ${visibility?.aiEnhancementsActive ? 'bg-green-500' : 'bg-gray-500'}`}></div>
              <span className={visibility?.aiEnhancementsActive ? 'text-green-400' : 'text-gray-400'}>
                AI Enhancements
              </span>
            </div>
            <div className="flex items-center space-x-2">
              <div className={`w-2 h-2 rounded-full ${(learningSystem?.activeDecisions || 0) > 0 ? 'bg-blue-500' : 'bg-gray-500'}`}></div>
              <span className={(learningSystem?.activeDecisions || 0) > 0 ? 'text-blue-400' : 'text-gray-400'}>
                Active Decisions ({learningSystem?.activeDecisions || 0})
              </span>
            </div>
          </div>
        </div>
        
        {learningSystem?.report?.insights && (
          <div className="bg-purple-900/20 rounded-lg p-4 border border-purple-500/30">
            <div className="text-purple-300 text-sm font-medium mb-2">🎯 Learning Insights</div>
            <div className="text-sm text-gray-300">
              {learningSystem.report.insights.thresholds && (
                <div>Current Thresholds: {JSON.stringify(learningSystem.report.insights.thresholds)}</div>
              )}
              {learningSystem.report.insights.confidenceLevel && (
                <div>Confidence Level: {learningSystem.report.insights.confidenceLevel}%</div>
              )}
            </div>
          </div>
        )}
        
        {learningSystem?.report?.recommendations && learningSystem.report.recommendations.length > 0 && (
          <div className="bg-yellow-900/20 rounded-lg p-4 border border-yellow-500/30">
            <div className="text-yellow-300 text-sm font-medium mb-2">💡 AI Recommendations</div>
            <div className="space-y-1">
              {learningSystem.report.recommendations.map((rec: any, index: number) => (
                <div key={index} className="text-sm text-yellow-400">
                  • {typeof rec === 'string' ? rec : rec.message || rec.type || 'No message'}
                </div>
              ))}
            </div>
          </div>
        )}
      </div>
    );
  };

  return (
    <div className="bg-gradient-to-br from-purple-900/20 to-blue-900/20 rounded-xl p-6 border border-purple-500/30">
      <div className="flex items-center justify-between mb-6">
        <div className="flex items-center space-x-3">
          <div className={`w-3 h-3 rounded-full ${learningSystem?.enabled ? 'bg-green-500 animate-pulse' : 'bg-yellow-500'}`}></div>
          <h3 className="text-lg font-semibold bg-gradient-to-r from-purple-400 to-blue-400 bg-clip-text text-transparent">
            🧠 AI Learning System
          </h3>
        </div>
        
        <button 
          onClick={fetchLearningStatus}
          className="px-3 py-1 bg-purple-600/20 hover:bg-purple-600/30 text-purple-400 rounded-lg transition-colors text-sm"
        >
          Refresh
        </button>
      </div>
      
      {/* Trading Performance Section - Always show if we have persistent data */}
      {learningData?.learningSystem?.persistentStats && (
        <motion.div 
          initial={{ opacity: 0, y: 20 }}
          animate={{ opacity: 1, y: 0 }}
          transition={{ delay: 0.3 }}
          className="bg-gradient-to-br from-blue-900/30 to-purple-900/30 rounded-xl p-6 border border-blue-500/30 mb-6"
        >
          <h4 className="text-xl font-bold text-blue-300 mb-4 flex items-center gap-2">
            <TrendingUp className="w-5 h-5" />
            Trading Performance
          </h4>
          
          <div className="grid grid-cols-2 md:grid-cols-4 gap-4 mb-6">
            <div className="bg-black/30 rounded-lg p-4 text-center">
              <div className="text-2xl font-bold text-green-400">
                {learningData.learningSystem.persistentStats.totalTrades}
              </div>
              <div className="text-sm text-gray-400">Total Trades</div>
            </div>
            
            <div className="bg-black/30 rounded-lg p-4 text-center">
              <div className="text-2xl font-bold text-blue-400">
                {learningData.learningSystem.persistentStats.successRate?.toFixed(1)}%
              </div>
              <div className="text-sm text-gray-400">Success Rate</div>
            </div>
            
            <div className="bg-black/30 rounded-lg p-4 text-center">
              <div className={`text-2xl font-bold ${
                learningData.learningSystem.persistentStats.totalPnl >= 0 ? 'text-green-400' : 'text-red-400'
              }`}>
                ${learningData.learningSystem.persistentStats.totalPnl?.toFixed(2)}
              </div>
              <div className="text-sm text-gray-400">Total P&L</div>
            </div>
            
            <div className="bg-black/30 rounded-lg p-4 text-center">
              <div className="text-2xl font-bold text-yellow-400">
                {learningData.learningSystem.persistentStats.winRate?.toFixed(0)}%
              </div>
              <div className="text-sm text-gray-400">Win Rate</div>
            </div>
          </div>

          {/* Recent Trades */}
          {learningData.learningSystem.recentTrades && learningData.learningSystem.recentTrades.length > 0 && (
            <div className="mt-6">
              <h5 className="text-lg font-semibold text-blue-300 mb-3">Recent Trades</h5>
              <div className="space-y-2 max-h-64 overflow-y-auto">
                {learningData.learningSystem.recentTrades.map((trade: any, index: number) => (
                  <div key={index} className="bg-black/20 rounded-lg p-3 flex justify-between items-center">
                    <div className="flex items-center gap-3">
                      <span className="text-sm font-medium text-gray-300">{trade.symbol}</span>
                      <span className={`text-xs px-2 py-1 rounded ${
                        trade.type === 'long' ? 'bg-green-900/50 text-green-300' : 'bg-red-900/50 text-red-300'
                      }`}>
                        {trade.type?.toUpperCase()}
                      </span>
                    </div>
                    <div className="text-right">
                      <div className={`text-sm font-semibold ${
                        parseFloat(trade.pnl) >= 0 ? 'text-green-400' : 'text-red-400'
                      }`}>
                        ${parseFloat(trade.pnl).toFixed(2)}
                      </div>
                      <div className="text-xs text-gray-500">
                        {new Date(trade.updatedAt).toLocaleDateString()}
                      </div>
                    </div>
                  </div>
                ))}
              </div>
            </div>
          )}

          {/* System Health */}
          {learningData.learningSystem.systemHealth && (
            <div className="mt-6 p-4 bg-black/20 rounded-lg">
              <h5 className="text-lg font-semibold text-blue-300 mb-2">System Health</h5>
              <div className="grid grid-cols-1 md:grid-cols-3 gap-4 text-sm">
                <div>
                  <span className="text-gray-400">AI Decisions:</span>
                  <span className="ml-2 text-white font-medium">
                    {learningData.learningSystem.systemHealth.totalDecisions?.toLocaleString()}
                  </span>
                </div>
                <div>
                  <span className="text-gray-400">Recent Activity:</span>
                  <span className="ml-2 text-white font-medium">
                    {learningData.learningSystem.systemHealth.recentDecisions} decisions
                  </span>
                </div>
                <div>
                  <span className="text-gray-400">Last Updated:</span>
                  <span className="ml-2 text-white font-medium">
                    {new Date(learningData.learningSystem.systemHealth.lastActivity).toLocaleTimeString()}
                  </span>
                </div>
              </div>
            </div>
          )}
        </motion.div>
      )}
      
      {renderLearningStatus()}
      
      {visibility?.lastUpdateTime && (
        <div className="mt-4 pt-4 border-t border-gray-600/30">
          <div className="text-gray-500 text-xs">
            Last updated: {new Date(visibility.lastUpdateTime).toLocaleTimeString()}
          </div>
        </div>
      )}
    </div>
  );
};

export default EnhancedAILearningPanel;