SCapi / app.py
Tanxshh's picture
Rename server.py to app.py
d8bd695 verified
raw
history blame
642 Bytes
from flask import Flask, request, jsonify
from pipeline import run_pipeline
import os
app = Flask(__name__)
PDF_FOLDER = "/app/pdfs"
os.makedirs(PDF_FOLDER, exist_ok=True)
@app.route("/upload_pdf", methods=["POST"])
def upload_pdf():
if 'pdf' not in request.files:
return jsonify({"error": "No PDF uploaded"}), 400
files = request.files.getlist('pdf')
for file in files:
path = os.path.join(PDF_FOLDER, file.filename)
file.save(path)
df = run_pipeline(PDF_FOLDER)
return df.to_json(orient="records")
if __name__ == "__main__":
app.run(host="0.0.0.0", port=7860)