from flask import Flask, request, jsonify
import requests
from pipeline import pdf2markdown_pipeline


app = Flask(__name__)


@app.route('/pdf-qa-server/pdf-to-md', methods=['POST'])
def pdf2markdown():
    data = request.json
    pdf_paths = data['pathList']
    callback_url = data['webhookUrl']
    for pdf_path in pdf_paths:
        process_status, pdf_id = pdf2markdown_pipeline(pdf_path)
        requests.post(callback_url, json={'pdfId': pdf_id, 'processStatus': process_status})
    return jsonify({})


if __name__ == '__main__':
    app.run(host='0.0.0.0', port=8000)