Kyan14's picture
Update app.py
5b895ed
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)