from flask import Flask, render_template, request, redirect, url_for
from generate import mood_art_generator
from PIL import Image
from io import BytesIO

app = Flask(__name__)

@app.route('/')
def index():
    return render_template('index.html')

@app.route('/generate', methods=['POST'])
def generate():
    image = request.files['image']
    image = Image.open(BytesIO(image.read()))
    art, narrative = mood_art_generator(image)
    if art:
        return render_template('result.html', art=art, narrative=narrative)
    else:
        return redirect(url_for('index'))

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

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