app.py

import os
from functools import wraps
from flask import Flask, request, jsonify, render_template, session, redirect, url_for, flash
import smtplib
from src.agent import generate_blog_draft
from src.db import get_db_connection, setup as db_setup
from src.logg import logger
from dotenv import load_dotenv

load_dotenv()

app = Flask(__name__)
app.secret_key = os.getenv("FLASK_SECRET_KEY")
GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
ADMIN_PASSWORD = os.getenv("ADMIN_PASSWORD")
COMPAMY_EMAIL = os.getenv("COMPANY_EMAIL")
EMAIL_PASSWORD = os.getenv("EMAIL_PASSWORD")

# Ensuring DB is ready
db_setup()


def login_required(f):
    @wraps(f)
    def decorated_function(*args, **kwargs):
        if not session.get('logged_in'):
            return redirect(url_for('login'))
        return f(*args, **kwargs)
    return decorated_function

# routes
@app.route('/')
def home():
    conn = get_db_connection()

    blogs = conn.execute("SELECT * FROM blogs ORDER BY created_at DESC").fetchall()
    projects = conn.execute("SELECT * FROM projects ORDER BY created_at DESC").fetchall()
    profile = conn.execute("SELECT * FROM profile ORDER BY id DESC LIMIT 1").fetchone()
    conn.close()
    
    # if profile is empty
    if not profile:
        profile = {"bio": "Welcome to my portfolio.", "skills": "AI, Python"}

    return render_template('index.html', blogs=blogs, projects=projects, profile=profile)


@app.route('/login', methods=['GET', 'POST'])
def login():
    if request.method == 'POST':
        if request.form.get('password') == ADMIN_PASSWORD:
            session['logged_in'] = True
            return redirect(url_for('admin'))
        flash('Invalid password')
    return render_template('login.html')


@app.route('/logout')
def logout():
    session.clear()
    return redirect(url_for('home'))



@app.route('/admin')
@login_required
def admin():
    conn = get_db_connection()
    blogs = conn.execute("SELECT * FROM blogs ORDER BY created_at DESC").fetchall()
    projects = conn.execute("SELECT * FROM projects ORDER BY created_at DESC").fetchall()
    # Fetch messages
    messages = conn.execute("SELECT * FROM messages ORDER BY created_at DESC").fetchall()
    conn.close()
    return render_template('admin.html', blogs=blogs, projects=projects, messages=messages)

# API
@app.route('/api/generate', methods=['POST'])
@login_required
def generate_api():
    data = request.json
    draft = generate_blog_draft(data.get('prompt'))
    if draft:
        return jsonify(draft)
    return jsonify({"error": "Agent failed"}), 500

# contact form api
@app.route('/api/contact', methods=['POST'])
def contact_form():
    data = request.json
    name = data.get('name')
    email = data.get('email')
    message = data.get('message')

    if not name or not email or not message:
        return jsonify({"error": "All fields are required"}), 400
    else:
        msg = f"New contact form submission:\n\nName: {name}\nEmail: {email}\nMessage:\n{message}"

    conn = get_db_connection()
    try:
        conn.execute(
            "INSERT INTO messages (name, email, message) VALUES (?, ?, ?)",
            (name, email, message)
        )
        conn.commit()
        
        try:
            with smtplib.SMTP("smtp.gmail.com", 587) as connection:
                connection.starttls()
                connection.login(COMPAMY_EMAIL, EMAIL_PASSWORD)
                connection.sendmail(
                    from_addr=COMPAMY_EMAIL,
                    to_addrs=COMPAMY_EMAIL,
                    msg=msg.encode('utf-8')
                )
            logger.info("Contact form email sent successfully.")
        except Exception as e:
            logger.error(f"Failed to send email notification: {e}")

        return jsonify({"status": "success"})
    except Exception as e:
        return jsonify({"error": str(e)}), 500
    
    finally:
        conn.close()


@app.route('/api/save', methods=['POST'])
@login_required
def save_content():
    data = request.json
    content_type = data.get('type', 'blog') # Default to blog

    try:
        with open('data/history.txt', 'a', encoding='utf-8') as file:
            file.write(f"--- NEW ENTRY ({content_type}) ---\n")
            file.write(f"Title: {data.get('title', 'No Title')}\n")
            file.write(f"Tags: {data.get('tags', '')}\n")
            file.write(f"Content:\n{data.get('content', '')}\n")
            file.write("-" * 30 + "\n\n")
    except Exception as e:
        logger.error(f"Warning: Could not save to history.txt: {e}")

    
    conn = get_db_connection()
    try:
        cur = conn.cursor()
        
        if content_type == 'project':
            cur.execute(
                "INSERT INTO projects (title, description, tech_stack) VALUES (?, ?, ?)",
                (data['title'], data['content'], data['tags'])
            )
        elif content_type == 'about':
            # A new profile (always fetch the latest one)
            cur.execute(
                "INSERT INTO profile (bio, skills) VALUES (?, ?)",
                (data['content'], data['tags'])
            )
        else:
            # Blog
            cur.execute(
                "INSERT INTO blogs (title, content, tags) VALUES (?, ?, ?)",
                (data['title'], data['content'], data['tags'])
            )
            
        conn.commit()
        conn.close()
        return jsonify({"status": "success", "type": content_type})
    except Exception as e:
        conn.close()
        return jsonify({"error": str(e)}), 500

# Delete Route
@app.route('/api/delete/<content_type>/<int:item_id>', methods=['POST'])
@login_required
def delete_item(content_type, item_id):
    conn = get_db_connection()
    try:
        if content_type == 'blog':
            conn.execute("DELETE FROM blogs WHERE id = ?", (item_id,))
        elif content_type == 'project':
            conn.execute("DELETE FROM projects WHERE id = ?", (item_id,))
        
        conn.commit()
        return jsonify({"status": "success"})
    except Exception as e:
        return jsonify({"error": str(e)}), 500
    finally:
        conn.close()


if __name__ == '__main__':
    app.run(debug=True)