Update main.py

main.py

@@ -243,7 +243,7 @@ def process_pdf_pages(pdf_file):
         if isinstance(txns, list):
             all_txns.extend(txns)
 
-    return all_txns
+    return all_txns, txt
 
 # --------- Chat Endpoint ---------
 @app.route("/chat", methods=["POST"])
@@ -292,7 +292,7 @@ def upload_statements():
 
         
         # extract + store transactions
-        txns = process_pdf_pages(f)
+        txns, txt = process_pdf_pages(f)
         for txn in txns:
             try:
                 dt = datetime.strptime(txn["Date"], "%d/%m/%Y")
@@ -312,7 +312,7 @@ def upload_statements():
             fs.collection("transactions").add(record)
             stored_count += 1
 
-    return jsonify({"message": f"Stored {stored_count} transactions", str(f)}), 200
+    return jsonify({"message": f"Stored {stored_count} transactions", "file": str(txt)}), 200
 
 # --------- Endpoint: Retrieve or Generate Financial Statement ---------