from flask import Blueprint, jsonify, render_template, request, redirect, url_for, flash
from flask_login import login_required, current_user
from werkzeug.utils import secure_filename
from models import Comment, Follow, User, Like, Message, Notification, Post, db
from datetime import datetime
import os
import google.generativeai as genai
from dotenv import load_dotenv
from cache_utils import cache


load_dotenv()

profile = Blueprint('prof', __name__)

# Configure Gemini AI
genai.configure(api_key=os.getenv('GEMINI_API_KEY'))
model = genai.GenerativeModel('gemini-pro')

ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif'}

def allowed_file(filename):
    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS

@profile.route('/profile/<username>')
@login_required
@cache.cached(timeout=300) 
def user_profile(username):
    user = User.query.filter_by(username=username).first_or_404()
    posts = Post.query.filter_by(author=user).order_by(Post.timestamp.desc()).all()
    followers_count = user.followers.count()
    following_count = user.followed.count()
    posts_count = user.posts.count()

    return render_template(
        'profile.html',
        user=user,
        current_user=current_user,
        posts=posts,
        followers_count=followers_count,
        following_count=following_count,
        posts_count=posts_count
    )

@profile.route('/edit_profile', methods=['GET', 'POST'])
@login_required
def edit_profile():
    if request.method == 'POST':
        username = request.form.get('username')
        email = request.form.get('email')
        bio = request.form.get('bio')
        location = request.form.get('location')

        if username:
            current_user.username = username
        if email:
            current_user.email = email
        if bio:
            current_user.bio = bio
        if location:
            current_user.location = location

        if 'profile_picture' in request.files:
            file = request.files['profile_picture']
            if file and allowed_file(file.filename):
                filename = secure_filename(file.filename)
                file.save(os.path.join('static/uploads', filename))
                current_user.profile_picture = filename

        db.session.commit()
        flash('Your profile has been updated.', 'success')
        return redirect(url_for('prof.user_profile', username=current_user.username))

    return render_template('edit_profile.html', user=current_user)

@profile.route('/follow/<username>', methods=['POST'])
@login_required
def follow(username):
    user = User.query.filter_by(username=username).first_or_404()
    current_user.follow(user)
    db.session.commit()
    return redirect(url_for('prof.user_profile', username=username))

@profile.route('/unfollow/<username>', methods=['POST'])
@login_required
def unfollow(username):
    user = User.query.filter_by(username=username).first_or_404()
    current_user.unfollow(user)
    db.session.commit()
    return redirect(url_for('prof.user_profile', username=username))

@profile.route('/followers/<username>')
@login_required
@cache.cached(timeout=300, key_prefix='followers_%s')
def followers(username):
    user = User.query.filter_by(username=username).first_or_404()
    followers = user.followers.all()
    return render_template('followers.html', user=user, followers=followers)

@profile.route('/following/<username>')
@login_required
@cache.cached(timeout=300, key_prefix='following_%s')
def following(username):
    user = User.query.filter_by(username=username).first_or_404()
    following = user.following.all()
    return render_template('following.html', user=user, following=following)



@profile.route('/post/<int:post_id>/like', methods=['POST'])
@login_required
def like_post(post_id):
    post = Post.query.get_or_404(post_id)
    if current_user not in post.likes:
        post.likes.append(current_user)
        db.session.commit()
    return redirect(request.referrer)

@profile.route('/post/<int:post_id>/unlike', methods=['POST'])
@login_required
def unlike_post(post_id):
    post = Post.query.get_or_404(post_id)
    if current_user in post.likes:
        post.likes.remove(current_user)
        db.session.commit()
    return redirect(request.referrer)