@onairos/react-native 3.4.1 → 3.5.4

package/src/components/TrainingModal.tsx

@@ -1,737 +0,0 @@
-import React, { useEffect, useRef, useState } from 'react';
-import {
-  View,
-  Text,
-  StyleSheet,
-  TouchableOpacity,
-  ActivityIndicator,
-  Dimensions,
-  Modal,
-  Animated,
-  TouchableWithoutFeedback,
-  SafeAreaView,
-  Image,
-  Alert,
-} from 'react-native';
-import Icon from 'react-native-vector-icons/MaterialIcons';
-import { io, Socket } from 'socket.io-client';
-import { COLORS } from '../constants';
-import type { TrainingModalProps } from '../types';
-
-const { width, height } = Dimensions.get('window');
-
-export const TrainingModal: React.FC<TrainingModalProps> = ({
-  visible,
-  progress,
-  eta,
-  onCancel,
-  onComplete,
-  modelKey,
-  username,
-  test = false,
-}) => {
-  // Socket and training state
-  const socketRef = useRef<Socket | null>(null);
-  const [socketConnected, setSocketConnected] = useState(false);
-  const [trainingStatus, setTrainingStatus] = useState('Initializing...');
-  const [hasError, setHasError] = useState(false);
-  const [errorCode, setErrorCode] = useState<string | null>(null);
-  const [isTrainingComplete, setIsTrainingComplete] = useState(false);
-  const [userTraits, setUserTraits] = useState<any>(null);
-  const [inferenceResults, setInferenceResults] = useState<any>(null);
-  const [internalProgress, setInternalProgress] = useState(0);
-  const [isEnochSDK, setIsEnochSDK] = useState(false); // Determine if using Enoch SDK
-  const [trainingResults, setTrainingResults] = useState<any>(null);
-  
-  // Use internal progress if available, otherwise fall back to prop progress
-  const currentProgress = internalProgress > 0 ? internalProgress / 100 : progress;
-  const progressPercentage = Math.round(currentProgress * 100);
-  
-  // Mock user data for now - in real implementation this would come from props
-  const userToken = 'mock-token'; // TODO: Get from actual auth
-  const userInfo = {
-    username: username || 'mobile_user',
-    email: null,
-    id: null,
-    name: username || 'mobile_user'
-  };
-
-  // Check if we should use Enoch SDK (based on modelKey or other criteria)
-  useEffect(() => {
-    // For now, default to clean training unless specified
-    // In real implementation, this would be determined by SDK type
-    setIsEnochSDK(modelKey?.includes('enoch') || false);
-  }, [modelKey]);
-
-  // Start training with new API schema
-  const startTraining = async (socketId: string) => {
-    try {
-      setTrainingStatus('Starting training...');
-      setInternalProgress(10);
-      setHasError(false);
-      setErrorCode(null);
-      
-      if (!userToken) {
-        throw new Error('No authentication token available');
-      }
-      
-      // Determine endpoint based on SDK type
-      const endpoint = isEnochSDK ? '/mobile-training/enoch' : '/mobile-training/clean';
-      
-      console.log(`🚀 Starting training with endpoint: ${endpoint}`);
-      console.log('🔍 Production mode: Using new training API schema');
-      
-      // Prepare training data in new format
-      const trainingData = {
-        Info: {
-          username: userInfo?.username || userInfo?.name || username || 'mobile_user',
-          // Mock connected platforms - in real implementation this would come from props
-          connectedPlatforms: ['youtube', 'instagram', 'reddit'] // TODO: Get from actual connections
-        }
-      };
-      
-      console.log('📤 Sending training data:', trainingData);
-      
-      // Call the new training API
-      const response = await fetch(`https://api2.onairos.uk${endpoint}`, {
-        method: 'POST',
-        headers: {
-          'Content-Type': 'application/json',
-          'Authorization': `Bearer ${userToken}` // JWT token authentication
-        },
-        body: JSON.stringify(trainingData)
-      });
-      
-      const result = await response.json();
-      
-      if (result.success) {
-        console.log('🚀 Training Started:', result.message);
-        setTrainingStatus('Training initiated...');
-        setInternalProgress(20);
-      } else {
-        console.error('Training start failed:', result.error);
-        
-        // Handle specific error codes from new schema
-        if (result.code) {
-          setErrorCode(result.code);
-          handleTrainingError(result.code, result.error);
-        } else {
-          setTrainingStatus(`Error: ${result.error}`);
-          setHasError(true);
-        }
-      }
-      
-    } catch (error) {
-      console.error('Training start error:', error);
-      setTrainingStatus(`Error: ${error instanceof Error ? error.message : 'Unknown error'}`);
-      setHasError(true);
-    }
-  };
-
-  // Handle training errors with specific error codes
-  const handleTrainingError = (code: string, message: string) => {
-    console.error(`Training Error [${code}]:`, message);
-    setHasError(true);
-    
-    switch (code) {
-      case 'PIN_REQUIRED':
-        setTrainingStatus('PIN setup required');
-        Alert.alert(
-          'PIN Required',
-          'Please set up your PIN first before training.',
-          [
-            { text: 'Cancel', style: 'cancel' },
-            { text: 'Set PIN', onPress: () => {
-              // TODO: Navigate to PIN setup
-              console.log('Navigate to PIN setup');
-            }}
-          ]
-        );
-        break;
-        
-      case 'CONNECTIONS_REQUIRED':
-        setTrainingStatus('No connected accounts');
-        Alert.alert(
-          'Connections Required',
-          'Please connect at least one social media account before training.',
-          [
-            { text: 'Cancel', style: 'cancel' },
-            { text: 'Connect', onPress: () => {
-              // TODO: Navigate to connections
-              console.log('Navigate to connections');
-            }}
-          ]
-        );
-        break;
-        
-      case 'INSUFFICIENT_DATA':
-        setTrainingStatus('Insufficient data for training');
-        Alert.alert(
-          'Insufficient Data',
-          'We need more interaction data to train your model effectively. Please use your connected platforms more and try again.',
-          [{ text: 'OK', style: 'default' }]
-        );
-        break;
-        
-      case 'ENCRYPTION_REQUIRED':
-        setTrainingStatus('Model encryption failed');
-        Alert.alert(
-          'Encryption Error',
-          'Failed to encrypt your model. Please try again.',
-          [{ text: 'OK', style: 'default' }]
-        );
-        break;
-        
-      default:
-        setTrainingStatus(`Error: ${message}`);
-        Alert.alert(
-          'Training Error',
-          message || 'An unexpected error occurred during training.',
-          [{ text: 'OK', style: 'default' }]
-        );
-    }
-  };
-
-  // Simulate training progress for test mode
-  const simulateTraining = () => {
-    console.log('🧪 Starting simulated training with new schema phases...');
-    setTrainingStatus('Validating Requirements');
-    setInternalProgress(5);
-    
-    // Simulate all phases from the new schema
-    const phases = [
-      { progress: 10, status: 'Validating Requirements', delay: 1000 },
-      { progress: 25, status: 'Training Model', delay: 2000 },
-      { progress: 50, status: 'Training Model', delay: 2000 },
-      { progress: 65, status: 'Model Trained - Running Test Inference', delay: 1500 },
-      { progress: 80, status: 'Encrypting and Compressing Model', delay: 1000 },
-      { progress: 95, status: 'Uploading Encrypted Model to ARDrive', delay: 1500 },
-      { progress: 100, status: 'Complete!', delay: 500 },
-    ];
-    
-    let currentPhase = 0;
-    
-    const progressInterval = setInterval(() => {
-      if (currentPhase < phases.length) {
-        const phase = phases[currentPhase];
-        setInternalProgress(phase.progress);
-        setTrainingStatus(phase.status);
-        
-        if (phase.progress === 100) {
-          setIsTrainingComplete(true);
-          clearInterval(progressInterval);
-          
-          // Mock completion results
-          const mockResults = {
-            completed: true,
-            message: "Model trained, encrypted, and stored successfully",
-            storage: "ARDrive",
-            encryption: true,
-            inference: isEnochSDK, // Only Enoch SDK runs inference
-            modelUrl: `https://arweave.net/mock_tx_${Date.now()}`
-          };
-          
-          setTrainingResults(mockResults);
-          
-          // Auto-complete after a short delay
-          setTimeout(() => {
-            onComplete && onComplete();
-          }, 1000);
-        }
-        
-        currentPhase++;
-      }
-    }, phases[currentPhase]?.delay || 1000);
-    
-    // Cleanup interval on unmount
-    return () => clearInterval(progressInterval);
-  };
-  
-  // Setup socket connection and training
-  useEffect(() => {
-    if (!visible || !userInfo) return;
-    
-    console.log('Setting up socket connection for training...');
-    console.log('🧑‍💻 User info available:', userInfo);
-    console.log('🧪 Test mode:', test);
-    console.log('🔧 Using Enoch SDK:', isEnochSDK);
-    
-    // If test mode is enabled, use simulation instead of real API
-    if (test) {
-      console.log('🧪 Test mode enabled - Using simulated training with new schema');
-      setSocketConnected(true);
-      simulateTraining();
-      return;
-    }
-    
-    // Initialize real socket connection
-    try {
-      console.log('🔌 Production mode: Initializing real socket connection');
-      
-      // Initialize socket connection
-      socketRef.current = io('https://api2.onairos.uk', {
-        transports: ['websocket'],
-        autoConnect: false
-      });
-      
-      // Socket event listeners for new schema
-      socketRef.current.on('connect', () => {
-        console.log('✅ Socket connected for training');
-        setSocketConnected(true);
-        const socketId = socketRef.current?.id;
-        if (socketId) {
-          startTraining(socketId);
-        }
-      });
-      
-      socketRef.current.on('disconnect', () => {
-        console.log('❌ Socket disconnected');
-        setSocketConnected(false);
-      });
-      
-      socketRef.current.on('connect_error', (error) => {
-        console.error('❌ Socket connection error:', error);
-        setTrainingStatus('Connection failed');
-        setHasError(true);
-      });
-
-      // Handle training updates with new schema phases
-      socketRef.current.on('trainingUpdate', (data) => {
-        console.log('📡 Training update received:', data);
-        
-        if (data.error) {
-          if (data.code) {
-            setErrorCode(data.code);
-            handleTrainingError(data.code, data.error);
-          } else {
-            console.error('Training update error:', data.error);
-            setTrainingStatus(`Error: ${data.error}`);
-            setHasError(true);
-          }
-        } else {
-          // Handle different status phases
-          if (data.status) {
-            setTrainingStatus(data.status);
-            
-            // Update progress based on phase
-            switch (data.status) {
-              case 'Validating Requirements':
-                setInternalProgress(10);
-                break;
-              case 'Training Model':
-                setInternalProgress(data.percent || 40);
-                break;
-              case 'Model Trained - Running Test Inference':
-                setInternalProgress(65);
-                break;
-              case 'Encrypting and Compressing Model':
-                setInternalProgress(80);
-                break;
-              case 'Uploading Encrypted Model to ARDrive':
-                setInternalProgress(90);
-                break;
-              default:
-                if (data.percent) {
-                  setInternalProgress(data.percent);
-                }
-            }
-          }
-          
-          if (data.progress) {
-            setInternalProgress(data.progress);
-          }
-        }
-      });
-      
-      // Handle training completion with new schema
-      socketRef.current.on('trainingCompleted', (data) => {
-        console.log('✅ Training Complete:', data);
-        
-        if (data.completed && data.storage === 'ARDrive' && data.encryption === true) {
-          setTrainingStatus('Training completed successfully!');
-          setIsTrainingComplete(true);
-          setInternalProgress(100);
-          setTrainingResults(data);
-          
-          // Display results
-          console.log('✅ Model URL:', data.modelUrl);
-          console.log('✅ Storage:', data.storage);
-          console.log('✅ Encrypted:', data.encryption);
-          console.log('✅ Inference:', data.inference);
-          
-          // Auto-complete after a short delay
-          setTimeout(() => {
-            onComplete && onComplete();
-          }, 1500);
-        } else {
-          console.error('Training completion missing required fields:', data);
-          setTrainingStatus('Training completed with warnings');
-          setHasError(true);
-        }
-      });
-
-      // Legacy handlers for backward compatibility
-      socketRef.current.on('inferenceCompleted', (data) => {
-        console.log('🧠 Inference Complete:', data);
-        setUserTraits(data.traits);
-        setInferenceResults(data.inferenceResults);
-      });
-      
-      socketRef.current.on('modelStandby', (data) => {
-        console.log('🎉 Model Ready:', data);
-        // This is handled by trainingCompleted now
-      });
-      
-      // Connect to socket
-      socketRef.current.connect();
-      
-    } catch (socketError) {
-      console.error('🚨 Production mode: Socket connection failed:', socketError);
-      setTrainingStatus('Failed to connect to training service');
-      setHasError(true);
-    }
-    
-    // Cleanup function
-    return () => {
-      if (socketRef.current) {
-        console.log('🔌 Disconnecting training socket...');
-        socketRef.current.disconnect();
-        socketRef.current = null;
-      }
-    };
-  }, [visible, userInfo, test, isEnochSDK]);
-
-  // Render error details for specific error codes
-  const renderErrorDetails = () => {
-    if (!hasError || !errorCode) return null;
-    
-    const errorDetails = {
-      'PIN_REQUIRED': {
-        icon: 'lock',
-        title: 'PIN Setup Required',
-        description: 'Set up your PIN to secure your AI model'
-      },
-      'CONNECTIONS_REQUIRED': {
-        icon: 'link',
-        title: 'Connect Accounts',
-        description: 'Connect at least one social media account'
-      },
-      'INSUFFICIENT_DATA': {
-        icon: 'data-usage',
-        title: 'More Data Needed',
-        description: 'We need more interaction data to train effectively'
-      },
-      'ENCRYPTION_REQUIRED': {
-        icon: 'security',
-        title: 'Encryption Failed',
-        description: 'Failed to encrypt your model securely'
-      }
-    };
-    
-    const details = errorDetails[errorCode];
-    if (!details) return null;
-    
-    return (
-      <View style={styles.errorDetailsContainer}>
-        <Icon name={details.icon} size={32} color="#FF6B6B" />
-        <Text style={styles.errorTitle}>{details.title}</Text>
-        <Text style={styles.errorDescription}>{details.description}</Text>
-      </View>
-    );
-  };
-
-  return (
-    <Modal
-      visible={visible}
-      transparent
-      animationType="none"
-      statusBarTranslucent
-      onRequestClose={onCancel}
-    >
-      <TouchableWithoutFeedback>
-        <View style={styles.modalOverlay}>
-          <TouchableWithoutFeedback>
-            <View style={styles.bottomSheet}>
-              <View style={styles.handleContainer}>
-                <View style={styles.handle} />
-              </View>
-              
-              <SafeAreaView style={styles.container}>
-                <View style={styles.content}>
-                  <Image 
-                    source={require('../assets/images/onairos_logo.png')} 
-                    style={styles.onairosLogo}
-                    resizeMode="contain"
-                  />
-                  
-                  <Text style={styles.title}>
-                    {isEnochSDK ? 'Training AI with Inference' : 'Training Your AI'}
-                  </Text>
-                  
-                  <View style={styles.progressContainer}>
-                    <View style={styles.progressBar}>
-                      <View
-                        style={[
-                          styles.progressFill,
-                          { width: `${progressPercentage}%` },
-                        ]}
-                      />
-                    </View>
-                    <Text style={styles.progressText}>{progressPercentage}%</Text>
-                  </View>
-
-                  <Text style={styles.statusText}>
-                    {trainingStatus}
-                  </Text>
-                  
-                  {hasError && (
-                    <View style={styles.errorContainer}>
-                      <Icon name="error-outline" size={24} color="#FF6B6B" />
-                      <Text style={styles.errorText}>Training encountered an error</Text>
-                    </View>
-                  )}
-
-                  {renderErrorDetails()}
-                  
-                  {isTrainingComplete && trainingResults && (
-                    <View style={styles.resultsContainer}>
-                      <Icon name="check-circle" size={32} color="#4CAF50" />
-                      <Text style={styles.resultsTitle}>Training Complete!</Text>
-                      <View style={styles.resultsDetails}>
-                        <View style={styles.resultItem}>
-                          <Icon name="cloud-done" size={16} color="#666" />
-                          <Text style={styles.resultText}>Storage: {trainingResults.storage}</Text>
-                        </View>
-                        <View style={styles.resultItem}>
-                          <Icon name="security" size={16} color="#666" />
-                          <Text style={styles.resultText}>Encrypted: {trainingResults.encryption ? 'Yes' : 'No'}</Text>
-                        </View>
-                        {trainingResults.inference && (
-                          <View style={styles.resultItem}>
-                            <Icon name="psychology" size={16} color="#666" />
-                            <Text style={styles.resultText}>Inference: Completed</Text>
-                          </View>
-                        )}
-                      </View>
-                    </View>
-                  )}
-
-                  <View style={styles.footer}>
-                    <TouchableOpacity
-                      style={styles.cancelButton}
-                      onPress={onCancel}
-                    >
-                      <Text style={styles.cancelButtonText}>Cancel</Text>
-                    </TouchableOpacity>
-                    
-                    {(currentProgress >= 1 || isTrainingComplete) && (
-                      <TouchableOpacity
-                        style={styles.completeButton}
-                        onPress={onComplete}
-                      >
-                        <Text style={styles.completeButtonText}>Complete</Text>
-                      </TouchableOpacity>
-                    )}
-                  </View>
-                </View>
-              </SafeAreaView>
-            </View>
-          </TouchableWithoutFeedback>
-        </View>
-      </TouchableWithoutFeedback>
-    </Modal>
-  );
-};
-
-const styles = StyleSheet.create({
-  modalOverlay: {
-    flex: 1,
-    backgroundColor: 'rgba(0, 0, 0, 0.5)',
-    justifyContent: 'flex-end',
-    alignItems: 'center',
-  },
-  bottomSheet: {
-    backgroundColor: '#fff',
-    width: width,
-    height: height * 0.8,
-    borderTopLeftRadius: 24,
-    borderTopRightRadius: 24,
-    overflow: 'hidden',
-  },
-  handleContainer: {
-    width: '100%',
-    alignItems: 'center',
-    paddingTop: 12,
-    paddingBottom: 8,
-  },
-  handle: {
-    width: 40,
-    height: 5,
-    borderRadius: 3,
-    backgroundColor: '#E0E0E0',
-  },
-  container: {
-    flex: 1,
-    backgroundColor: '#fff',
-  },
-  content: {
-    flex: 1,
-    width: '100%',
-    alignItems: 'center',
-    justifyContent: 'center',
-    padding: 24,
-  },
-  title: {
-    fontSize: 24,
-    fontWeight: '600',
-    marginTop: 24,
-    marginBottom: 8,
-    color: COLORS.text.primary,
-    textAlign: 'center',
-  },
-  subtitle: {
-    fontSize: 16,
-    color: COLORS.text.secondary,
-    textAlign: 'center',
-    marginBottom: 32,
-  },
-  statusText: {
-    fontSize: 14,
-    color: COLORS.text.secondary,
-    marginBottom: 24,
-    textAlign: 'center',
-  },
-  progressContainer: {
-    width: '100%',
-    marginBottom: 16,
-  },
-  progressBar: {
-    height: 8,
-    backgroundColor: COLORS.border,
-    borderRadius: 4,
-    overflow: 'hidden',
-  },
-  progressFill: {
-    height: '100%',
-    backgroundColor: COLORS.primary,
-    borderRadius: 4,
-  },
-  progressText: {
-    fontSize: 14,
-    color: COLORS.text.secondary,
-    textAlign: 'center',
-    marginTop: 8,
-  },
-  etaText: {
-    fontSize: 14,
-    color: COLORS.text.secondary,
-    marginBottom: 24,
-  },
-  loadingContainer: {
-    flexDirection: 'row',
-    alignItems: 'center',
-    marginBottom: 32,
-  },
-  loadingText: {
-    fontSize: 14,
-    color: COLORS.text.secondary,
-    marginLeft: 8,
-  },
-  footer: {
-    flexDirection: 'row',
-    alignItems: 'center',
-    justifyContent: 'space-between',
-    width: '100%',
-    borderTopWidth: 1,
-    borderTopColor: '#eee',
-    paddingTop: 16,
-    paddingHorizontal: 24,
-    paddingBottom: 24,
-  },
-  cancelButton: {
-    paddingVertical: 8,
-    paddingHorizontal: 16,
-  },
-  cancelButtonText: {
-    fontSize: 16,
-    color: '#666',
-  },
-  completeButton: {
-    paddingVertical: 16,
-    paddingHorizontal: 32,
-    borderRadius: 16,
-    backgroundColor: '#fff',
-    borderWidth: 1,
-    borderColor: '#000',
-  },
-  completeButtonText: {
-    fontSize: 16,
-    fontWeight: '600',
-    color: '#000',
-  },
-  onairosLogo: {
-    width: 48,
-    height: 48,
-  },
-  errorContainer: {
-    flexDirection: 'row',
-    alignItems: 'center',
-    marginBottom: 16,
-  },
-  errorText: {
-    fontSize: 14,
-    color: '#FF6B6B',
-    marginLeft: 8,
-  },
-  errorDetailsContainer: {
-    alignItems: 'center',
-    marginBottom: 24,
-    padding: 16,
-    backgroundColor: '#FFF5F5',
-    borderRadius: 12,
-    borderWidth: 1,
-    borderColor: '#FFEBEE',
-  },
-  errorTitle: {
-    fontSize: 16,
-    fontWeight: '600',
-    color: '#FF6B6B',
-    marginTop: 8,
-    marginBottom: 4,
-  },
-  errorDescription: {
-    fontSize: 14,
-    color: '#666',
-    textAlign: 'center',
-  },
-  resultsContainer: {
-    alignItems: 'center',
-    marginBottom: 24,
-    padding: 16,
-    backgroundColor: '#F5F5F5',
-    borderRadius: 12,
-    borderWidth: 1,
-    borderColor: '#E0E0E0',
-  },
-  resultsTitle: {
-    fontSize: 16,
-    fontWeight: '600',
-    color: '#4CAF50',
-    marginTop: 8,
-    marginBottom: 12,
-  },
-  resultsDetails: {
-    alignItems: 'flex-start',
-    width: '100%',
-  },
-  resultItem: {
-    flexDirection: 'row',
-    alignItems: 'center',
-    marginBottom: 4,
-  },
-  resultText: {
-    fontSize: 14,
-    color: '#666',
-    marginLeft: 8,
-  },
-});