Add application file

main.py

@@ -1,3 +1,4 @@
+import csv
 import logging
 import os
 import tempfile
@@ -9,6 +10,20 @@ from fastapi.responses import FileResponse
 from model import extract_contact
 from vcf import generate_vcf, VCF_DIR
 
+CSV_FIELDS = ["name", "company", "designation", "phone", "email", "website", "address"]
+CSV_PATH = os.path.join(VCF_DIR, "contacts.csv")
+
+
+def append_csv(data):
+    # ponytail: append-only log of every extraction; write header once on first row
+    new_file = not os.path.exists(CSV_PATH)
+    with open(CSV_PATH, "a", newline="") as f:
+        w = csv.DictWriter(f, fieldnames=CSV_FIELDS, extrasaction="ignore")
+        if new_file:
+            w.writeheader()
+        w.writerow({k: data.get(k, "") for k in CSV_FIELDS})
+    return os.path.basename(CSV_PATH)
+
 logging.basicConfig(
     level=logging.INFO,
     format="%(asctime)s %(levelname)s: %(message)s",
@@ -47,13 +62,16 @@ async def extract(file: UploadFile = File(...)):
 
         vcf_path = generate_vcf(data)
         filename = os.path.basename(vcf_path)
-        log.info("wrote vcf: %s", filename)
+        csv_name = append_csv(data)
+        log.info("wrote vcf: %s, appended csv: %s", filename, csv_name)
 
         return {
             "success": True,
             "contact": data,
             "vcf": filename,
+            "csv": csv_name,
             "download": f"/download/{filename}",
+            "csv_download": f"/download/{csv_name}",
         }
     except Exception as e:
         # ponytail: exc_info dumps the full traceback to the container logs
@@ -69,5 +87,6 @@ def download(filename: str):
     # ponytail: basename strips any ../ path traversal from the URL
     path = os.path.join(VCF_DIR, os.path.basename(filename))
     if not os.path.exists(path):
-        raise HTTPException(404, "VCF not found")
-    return FileResponse(path, filename=filename, media_type="text/vcard")
+        raise HTTPException(404, "file not found")
+    media = "text/csv" if path.endswith(".csv") else "text/vcard"
+    return FileResponse(path, filename=filename, media_type=media)