From c8698cb943839fef702feebf143eb96e4cb3472a Mon Sep 17 00:00:00 2001
From: admin <l.barontini@open-squared.ch>
Date: Mon, 12 Jan 2026 14:17:00 +0000
Subject: [PATCH] Add extract.py

---
 extract.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)
 create mode 100644 extract.py

diff --git a/extract.py b/extract.py
new file mode 100644
index 0000000..1c1e5e9
--- /dev/null
+++ b/extract.py
@@ -0,0 +1,22 @@
+from openai import OpenAI
+import json, re
+
+client = OpenAI()
+
+def extract_weight_report(text, lab):
+    prompt = f"""
+    Return ONLY valid JSON matching this structure:
+    ...
+    Document:
+    {text}
+    """
+
+    response = client.responses.create(
+        model="gpt-4.1-mini",
+        input=[{"role":"user","content":[{"type":"input_text","text":prompt}]}]
+    )
+
+    raw = response.output[0].content[0].text
+    raw = re.search(r"\{.*\}", raw, re.S).group()
+
+    return json.loads(raw)