Update app.py

app.py

@@ -1,25 +1,31 @@
-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)
+from fastapi import FastAPI, UploadFile, File
+from fastapi.middleware.cors import CORSMiddleware
+import shutil
+import os
+import tempfile
+from pipeline import run_pipeline  # Make sure pipeline.py is in the same folder
+
+app = FastAPI(title="SC API")
+
+# Allow CORS so your website or n8n can POST
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # You can restrict to your website
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+@app.post("/upload_pdf")
+async def upload_pdf(files: list[UploadFile] = File(...)):
+    with tempfile.TemporaryDirectory() as tmpdirname:
+        pdf_paths = []
+        for file in files:
+            dest = os.path.join(tmpdirname, file.filename)
+            with open(dest, "wb") as f:
+                shutil.copyfileobj(file.file, f)
+            pdf_paths.append(dest)
+        
+        # Run the SC pipeline
+        results = run_pipeline(tmpdirname)
+        return results.to_dict(orient="records")