Spaces:
Sleeping
Sleeping
File size: 1,052 Bytes
dbe2c62 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
from flask import Flask, request, jsonify
from flask_cors import CORS
import App_Caller
app = Flask(__name__)
CORS(app)
@app.route("/process_pdf", methods=["POST"])
def process_pdf():
"""API nhận file PDF và trả về summary + category."""
if "file" not in request.files:
return jsonify({"error": "Thiếu file PDF"}), 400
pdf_file = request.files["file"]
if not pdf_file.filename.endswith(".pdf"):
return jsonify({"error": "File không hợp lệ"}), 400
try:
pdf_bytes = pdf_file.read()
result = App_Caller.fileProcess(pdf_bytes)
return jsonify({
"status": "success",
"checkstatus": result["checkstatus"],
"metrics": result["metrics"],
"summary": result["summary"],
"category": result["category"],
"top_candidates": result["reranked"]
})
except Exception as e:
return jsonify({"status": "error", "message": str(e)}), 500
if __name__ == "__main__":
app.run(host="0.0.0.0", port=8000)
|