Update app logic
Browse files
app.py
CHANGED
|
@@ -1,4 +1,23 @@
|
|
| 1 |
# app.py (HF Space)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
}
|
| 3 |
|
| 4 |
|
|
|
|
| 1 |
# app.py (HF Space)
|
| 2 |
+
|
| 3 |
+
|
| 4 |
+
# dose_per_intake: e.g., "1회 1정", "1정", "5 mL"
|
| 5 |
+
m_dose = re.search(r"(1회\s*)?(\d+)\s*([가-힣a-zA-Z]+|mL|ml|mg)", t)
|
| 6 |
+
dose_per_intake = None
|
| 7 |
+
if m_dose:
|
| 8 |
+
dose_per_intake = f"{m_dose.group(2)} {m_dose.group(3)}"
|
| 9 |
+
|
| 10 |
+
|
| 11 |
+
# drug name (heuristic): token before mg/mL or first uppercase-like word
|
| 12 |
+
m_drug = re.search(r"([가-힣A-Za-z]+)\s*(\d+\s*(mg|mL|ml))", t)
|
| 13 |
+
drug_name = m_drug.group(1) if m_drug else None
|
| 14 |
+
|
| 15 |
+
|
| 16 |
+
return {
|
| 17 |
+
"drug_name": drug_name,
|
| 18 |
+
"dose_per_intake": dose_per_intake,
|
| 19 |
+
"times_per_day": times_per_day,
|
| 20 |
+
"time_slots": time_slots or None,
|
| 21 |
}
|
| 22 |
|
| 23 |
|