from flask import Flask, jsonify, request, render_template_string
from flask_sqlalchemy import SQLAlchemy
from flask_migrate import Migrate
from flask_jwt_extended import JWTManager, create_access_token, jwt_required, get_jwt_identity
import uuid
from datetime import datetime

# Initialize Flask app
app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///flowforge.db'
app.config['JWT_SECRET_KEY'] = 'super-secret-2023'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False

# Initialize extensions
db = SQLAlchemy(app)
migrate = Migrate(app, db)
jwt = JWTManager(app)

# Database Models
class User(db.Model):
    id = db.Column(db.String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    email = db.Column(db.String(120), unique=True, nullable=False)
    password = db.Column(db.String(120), nullable=False)
    tier = db.Column(db.String(20), default='free')  # free, pro, enterprise

class Project(db.Model):
    id = db.Column(db.String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    name = db.Column(db.String(120), nullable=False)
    owner_id = db.Column(db.String(36), db.ForeignKey('user.id'), nullable=False)
    deadline = db.Column(db.DateTime)
    
class Task(db.Model):
    id = db.Column(db.String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    title = db.Column(db.String(120), nullable=False)
    project_id = db.Column(db.String(36), db.ForeignKey('project.id'), nullable=False)
    status = db.Column(db.String(20), default='todo')
    estimated_duration = db.Column(db.Integer)
    assigned_to = db.Column(db.String(36), db.ForeignKey('user.id'))
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    due_date = db.Column(db.DateTime)

# AI Simulation Engine
class AIService:
    @staticmethod
    def predict_duration(task_title):
        """Simulate AI duration prediction"""
        keyword_weights = {'design': 8, 'develop': 16, 'test': 4, 
                          'review': 2, 'meeting': 1, 'document': 6}
        base_hours = 4
        for kw, hours in keyword_weights.items():
            if kw in task_title.lower():
                return hours
        return base_hours

    @staticmethod
    def optimize_schedule(tasks):
        """Simulate AI schedule optimization"""
        return sorted(tasks, key=lambda x: x.get('estimated_duration', 0))

# Authentication Routes
@app.route('/api/register', methods=['POST'])
def register():
    data = request.get_json()
    if User.query.filter_by(email=data['email']).first():
        return jsonify({'error': 'Email already exists'}), 400
        
    new_user = User(
        email=data['email'],
        password=data['password'],  # In production: use bcrypt
        tier=data.get('tier', 'free')
    )
    db.session.add(new_user)
    db.session.commit()
    return jsonify({'message': 'User created'}), 201

@app.route('/api/login', methods=['POST'])
def login():
    data = request.get_json()
    user = User.query.filter_by(email=data['email']).first()
    
    if not user or user.password != data['password']:
        return jsonify({'error': 'Invalid credentials'}), 401
        
    access_token = create_access_token(identity=user.id)
    return jsonify(access_token=access_token, tier=user.tier), 200

# Project Management Routes
@app.route('/api/projects', methods=['POST'])
@jwt_required()
def create_project():
    current_user = get_jwt_identity()
    data = request.get_json()
    
    # Check project limit for free tier
    if User.query.get(current_user).tier == 'free' and \
       Project.query.filter_by(owner_id=current_user).count() >= 3:
        return jsonify({'error': 'Free tier limited to 3 projects'}), 403
        
    new_project = Project(
        name=data['name'],
        owner_id=current_user,
        deadline=datetime.fromisoformat(data['deadline']) if data.get('deadline') else None
    )
    db.session.add(new_project)
    db.session.commit()
    return jsonify({'id': new_project.id, 'name': new_project.name}), 201

@app.route('/api/projects', methods=['GET'])
@jwt_required()
def get_projects():
    current_user = get_jwt_identity()
    projects = Project.query.filter_by(owner_id=current_user).all()
    return jsonify([{'id': p.id, 'name': p.name} for p in projects]), 200

# Task Management Routes
@app.route('/api/tasks', methods=['POST'])
@jwt_required()
def create_task():
    current_user = get_jwt_identity()
    data = request.get_json()
    
    # AI prediction
    predicted_duration = AIService.predict_duration(data['title'])
    
    new_task = Task(
        title=data['title'],
        project_id=data['project_id'],
        estimated_duration=predicted_duration,
        due_date=datetime.fromisoformat(data['due_date']) if data.get('due_date') else None
    )
    db.session.add(new_task)
    db.session.commit()
    return jsonify({
        'id': new_task.id,
        'title': new_task.title,
        'predicted_duration': predicted_duration
    }), 201

@app.route('/api/ai-schedule/<project_id>', methods=['GET'])
@jwt_required()
def ai_schedule(project_id):
    tasks = Task.query.filter_by(project_id=project_id).all()
    if not tasks:
        return jsonify({'error': 'No tasks found'}), 404
    
    optimized = AIService.optimize_schedule([
        {'id': t.id, 'title': t.title, 'estimated_duration': t.estimated_duration}
        for t in tasks
    ])
    return jsonify(optimized), 200

# Frontend Route
@app.route('/')
def index():
    return render_template_string('''
    <!DOCTYPE html>
    <html>
    <head>
        <title>FlowForge SaaS</title>
        <script src="https://unpkg.com/axios/dist/axios.min.js"></script>
        <style>
            :root {
                --primary: #4361ee;
                --secondary: #3f37c9;
                --success: #4cc9f0;
                --dark: #1d3557;
            }
            * {
                box-sizing: border-box;
                margin: 0;
                padding: 0;
                font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
            }
            body {
                background-color: #f5f7fa;
                color: #333;
                line-height: 1.6;
                padding: 20px;
            }
            .container {
                max-width: 1200px;
                margin: 0 auto;
            }
            header {
                background: linear-gradient(135deg, var(--primary), var(--secondary));
                color: white;
                padding: 1rem 2rem;
                border-radius: 10px;
                margin-bottom: 2rem;
                box-shadow: 0 4px 6px rgba(0,0,0,0.1);
            }
            .card {
                background: white;
                border-radius: 10px;
                box-shadow: 0 4px 6px rgba(0,0,0,0.05);
                padding: 1.5rem;
                margin-bottom: 1.5rem;
                transition: transform 0.3s ease, box-shadow 0.3s ease;
            }
            .card:hover {
                transform: translateY(-5px);
                box-shadow: 0 10px 20px rgba(0,0,0,0.1);
            }
            .grid {
                display: grid;
                grid-template-columns: repeat(auto-fit, minmax(300px, 1fr));
                gap: 1.5rem;
            }
            input, select, button {
                padding: 0.8rem;
                border: 1px solid #ddd;
                border-radius: 4px;
                font-size: 1rem;
                margin: 0.5rem 0;
                width: 100%;
            }
            button {
                background-color: var(--primary);
                color: white;
                border: none;
                cursor: pointer;
                font-weight: 600;
                letter-spacing: 0.5px;
                transition: background-color 0.3s;
            }
            button:hover {
                background-color: var(--secondary);
            }
            .task-card {
                border-left: 4px solid var(--success);
                padding: 1rem;
                margin: 1rem 0;
                background-color: #f8f9fa;
            }
            .view-section {
                display: none;
            }
            .active {
                display: block;
            }
            .nav {
                display: flex;
                gap: 1rem;
                margin-bottom: 1rem;
            }
            .nav-btn {
                background: none;
                color: var(--dark);
                border-bottom: 3px solid transparent;
                border-radius: 0;
                width: auto;
            }
            .nav-btn.active {
                border-bottom: 3px solid var(--primary);
                color: var(--primary);
            }
            @media (max-width: 768px) {
                .grid {
                    grid-template-columns: 1fr;
                }
            }
        </style>
    <link rel="me" href="https://www.blogger.com/profile/11021217223971810927" />
<meta name='google-adsense-platform-account' content='ca-host-pub-1556223355139109'/>
<meta name='google-adsense-platform-domain' content='blogspot.com'/>
</head>
    <body>
        <div class="container">
            <header>
                <h1>FlowForge SaaS</h1>
                <p>AI-Powered Project Management</p>
            </header>
            
            <div id="auth-view" class="view-section active">
                <div class="card">
                    <h2>Login to Your Account</h2>
                    <input id="login-email" type="email" placeholder="Email" />
                    <input id="login-password" type="password" placeholder="Password" />
                    <button onclick="login()">Login</button>
                    <p style="text-align: center; margin-top: 1rem;">
                        Don't have an account? <a href="#" onclick="showView('register-view')">Register</a>
                    </p>
                </div>
            </div>
            
            <div id="register-view" class="view-section">
                <div class="card">
                    <h2>Create New Account</h2>
                    <input id="register-email" type="email" placeholder="Email" />
                    <input id="register-password" type="password" placeholder="Password" />
                    <button onclick="register()">Create Account</button>
                    <p style="text-align: center; margin-top: 1rem;">
                        Already have an account? <a href="#" onclick="showView('auth-view')">Login</a>
                    </p>
                </div>
            </div>
            
            <div id="app-view" class="view-section">
                <div class="nav">
                    <button class="nav-btn" onclick="showView('projects-view')">Projects</button>
                    <button class="nav-btn" onclick="showView('tasks-view')">Tasks</button>
                    <button class="nav-btn" onclick="showView('ai-view')">AI Tools</button>
                    <button class="nav-btn" style="margin-left: auto;" onclick="logout()">Logout</button>
                </div>
                
                <div id="projects-view" class="view-section active">
                    <div class="card">
                        <h2>Create New Project</h2>
                        <input id="project-name" placeholder="Project name" />
                        <button onclick="createProject()">Create Project</button>
                    </div>
                    
                    <div class="card">
                        <h2>Your Projects</h2>
                        <div id="projects-list" class="grid"></div>
                    </div>
                </div>
                
                <div id="tasks-view" class="view-section">
                    <div class="card">
                        <h2>Create New Task</h2>
                        <select id="project-select"></select>
                        <input id="task-title" placeholder="Task title" />
                        <button onclick="createTask()">Create Task</button>
                        <div id="ai-prediction" style="margin-top: 1rem; padding: 1rem; background: #e8f4fc; border-radius: 4px;"></div>
                    </div>
                    
                    <div class="card">
                        <h2>Project Tasks</h2>
                        <select id="tasks-project-select" onchange="loadProjectTasks(this.value)"></select>
                        <div id="tasks-list" style="margin-top: 1rem;"></div>
                    </div>
                </div>
                
                <div id="ai-view" class="view-section">
                    <div class="card">
                        <h2>AI Schedule Optimization</h2>
                        <select id="ai-project-select"></select>
                        <button onclick="optimizeSchedule()">Optimize Schedule</button>
                        <div id="optimized-tasks" style="margin-top: 1rem;"></div>
                    </div>
                </div>
            </div>
        </div>

        <script>
            // State management
            let authToken = null;
            let currentUser = null;
            
            // DOM Helper functions
            function showView(viewId) {
                // Hide all views
                document.querySelectorAll('.view-section').forEach(view => {
                    view.classList.remove('active');
                });
                
                // Show requested view
                const view = document.getElementById(viewId);
                if (view) view.classList.add('active');
                
                // Handle specific view initialization
                if (viewId === 'app-view') {
                    loadProjects();
                    initProjectSelects();
                }
            }
            
            // Auth functions
            async function login() {
                const email = document.getElementById('login-email').value;
                const password = document.getElementById('login-password').value;
                
                try {
                    const response = await axios.post('/api/login', { email, password });
                    authToken = response.data.access_token;
                    currentUser = { tier: response.data.tier };
                    showView('app-view');
                } catch (error) {
                    alert('Login failed: ' + (error.response?.data?.error || 'Server error'));
                }
            }
            
            async function register() {
                const email = document.getElementById('register-email').value;
                const password = document.getElementById('register-password').value;
                
                try {
                    await axios.post('/api/register', { email, password });
                    alert('Account created successfully! Please login.');
                    showView('auth-view');
                } catch (error) {
                    alert('Registration failed: ' + (error.response?.data?.error || 'Server error'));
                }
            }
            
            function logout() {
                authToken = null;
                currentUser = null;
                showView('auth-view');
            }
            
            // Project functions
            async function createProject() {
                const name = document.getElementById('project-name').value;
                
                try {
                    const response = await axios.post('/api/projects', { name }, {
                        headers: { Authorization: `Bearer ${authToken}` }
                    });
                    loadProjects();
                    initProjectSelects();
                    alert(`Project "${response.data.name}" created!`);
                } catch (error) {
                    alert('Error: ' + (error.response?.data?.error || 'Server error'));
                }
            }
            
            async function loadProjects() {
                try {
                    const response = await axios.get('/api/projects', {
                        headers: { Authorization: `Bearer ${authToken}` }
                    });
                    
                    const projectsList = document.getElementById('projects-list');
                    projectsList.innerHTML = '';
                    
                    response.data.forEach(project => {
                        const projectCard = document.createElement('div');
                        projectCard.className = 'card';
                        projectCard.innerHTML = `
                            <h3>${project.name}</h3>
                            <p>Project ID: ${project.id}</p>
                            <button onclick="loadProjectTasks('${project.id}')">View Tasks</button>
                        `;
                        projectsList.appendChild(projectCard);
                    });
                } catch (error) {
                    console.error('Error loading projects:', error);
                }
            }
            
            function initProjectSelects() {
                ['project-select', 'tasks-project-select', 'ai-project-select'].forEach(selectId => {
                    const select = document.getElementById(selectId);
                    select.innerHTML = '<option value="">Select Project</option>';
                    
                    // Add projects from server (in real app we'd fetch)
                    axios.get('/api/projects', {
                        headers: { Authorization: `Bearer ${authToken}` }
                    }).then(response => {
                        response.data.forEach(project => {
                            const option = document.createElement('option');
                            option.value = project.id;
                            option.textContent = project.name;
                            select.appendChild(option);
                        });
                    });
                });
            }
            
            // Task functions
            async function createTask() {
                const projectId = document.getElementById('project-select').value;
                const title = document.getElementById('task-title').value;
                
                if (!projectId || !title) {
                    alert('Please select a project and enter a task title');
                    return;
                }
                
                try {
                    const response = await axios.post('/api/tasks', {
                        project_id: projectId,
                        title: title
                    }, {
                        headers: { Authorization: `Bearer ${authToken}` }
                    });
                    
                    document.getElementById('ai-prediction').innerHTML = `
                        <p>AI Prediction: This task will take approximately <strong>${response.data.predicted_duration} hours</strong></p>
                    `;
                    
                    // Refresh tasks list
                    loadProjectTasks(projectId);
                } catch (error) {
                    alert('Error creating task: ' + (error.response?.data?.error || 'Server error'));
                }
            }
            
            async function loadProjectTasks(projectId) {
                if (!projectId) return;
                
                try {
                    const response = await axios.get(`/api/ai-schedule/${projectId}`, {
                        headers: { Authorization: `Bearer ${authToken}` }
                    });
                    
                    const tasksList = document.getElementById('tasks-list');
                    tasksList.innerHTML = '';
                    
                    response.data.forEach(task => {
                        const taskCard = document.createElement('div');
                        taskCard.className = 'task-card';
                        taskCard.innerHTML = `
                            <h4>${task.title}</h4>
                            <p>Estimated Duration: ${task.estimated_duration} hours</p>
                        `;
                        tasksList.appendChild(taskCard);
                    });
                } catch (error) {
                    console.error('Error loading tasks:', error);
                }
            }
            
            // AI Functions
            async function optimizeSchedule() {
                const projectId = document.getElementById('ai-project-select').value;
                
                if (!projectId) {
                    alert('Please select a project');
                    return;
                }
                
                try {
                    const response = await axios.get(`/api/ai-schedule/${projectId}`, {
                        headers: { Authorization: `Bearer ${authToken}` }
                    });
                    
                    const container = document.getElementById('optimized-tasks');
                    container.innerHTML = '<h3>Optimized Task Order:</h3>';
                    
                    response.data.forEach((task, index) => {
                        const taskEl = document.createElement('div');
                        taskEl.className = 'task-card';
                        taskEl.innerHTML = `
                            <h4>#${index + 1}: ${task.title}</h4>
                            <p>Estimated Duration: ${task.estimated_duration} hours</p>
                        `;
                        container.appendChild(taskEl);
                    });
                } catch (error) {
                    alert('Error optimizing schedule: ' + (error.response?.data?.error || 'Server error'));
                }
            }
        </script>
    </body>
    </html>
    ''')

# Initialize Database
@app.before_first_request
def initialize_database():
    db.create_all()
    
    # Create sample user for testing
    if not User.query.first():
        sample_user = User(
            email="test@flowforge.com",
            password="password123",
            tier="pro"
        )
        db.session.add(sample_user)
        db.session.commit()

# Run the application
if __name__ == '__main__':
    app.run(debug=True)