temporary-indexing-app/app.py

import os
from pathlib import Path

from flask import Flask, redirect, render_template, url_for
from flask_wtf import FlaskForm
from flask_wtf.csrf import CSRFProtect
from flask_wtf.file import FileAllowed, FileField, FileRequired
from werkzeug.utils import secure_filename
from wtforms import RadioField, StringField
from wtforms.validators import DataRequired

app = Flask(__name__)
app.config["SECRET_KEY"] = "foo"
app.config["WTF_CSRF_SECRET_KEY"] = "bar"


CWD = Path().resolve()


class UploadForm(FlaskForm):
    pdf = FileField(validators=[FileAllowed(["pdf"], "PDFs only!"), FileRequired()])


class CardSidesForm(FlaskForm):
    title = RadioField("title", choices=[("side a", "side a"), ("side b", "side b")])


@app.route("/")
def home():
    upload_form = UploadForm()
    card_sides_form = CardSidesForm()
    return render_template(
        "index.html", upload_form=upload_form, card_sides_form=card_sides_form
    )


@app.route("/upload", methods=["POST"])
def upload():
    upload_form = UploadForm()
    card_sides_form = CardSidesForm()

    if upload_form.validate_on_submit():
        f = upload_form.pdf.data
        filename = secure_filename(f.filename)
        f.save(os.path.join(CWD, "pdfs", filename))
        return redirect(url_for("home"))
    return render_template(
        "index.html", upload_form=upload_form, card_sides_form=card_sides_form
    )


@app.route("/generate", methods=["POST"])
def generate():
    upload_form = UploadForm()
    card_sides_form = CardSidesForm()

    if card_sides_form.validate_on_submit():
        print(card_sides_form.title.data)
    else:
        print("didnt work")

    return render_template(
        "index.html", upload_form=upload_form, card_sides_form=card_sides_form
    )