Update app.py

app.py

@@ -5,9 +5,10 @@ def extract_study_timeline(text):
     """
     Extracts Screening, Treatment, and Follow-up durations from study timeline text.
     """
-    screening = re.search(r'(?:Screening|Pre-study observation|Initial phase)\s*[:\-]?\s*(\d+)\s*weeks?', text)
-    treatment = re.search(r'(?:Treatment|Intervention|Therapy|Dosing period)\s*[:\-]?\s*(\d+)\s*weeks?', text)
-    follow_up = re.search(r'(?:Follow[-\s]*up|Recovery|Post-study phase|Monitoring period)\s*[:\-]?\s*(\d+)\s*weeks?', text)
+
+    screening = re.search(r'(?:Screening(?:\s*phase| period)?|Pre-study observation)\s*(?:of|:|-)?\s*(\d+)\s*weeks?', text, re.IGNORECASE)
+    treatment = re.search(r'(?:Treatment(?:\s*period)?|Intervention|Therapy|Dosing phase)\s*(?:of|:|-)?\s*(\d+)\s*weeks?', text, re.IGNORECASE)
+    follow_up = re.search(r'(?:Follow[-\s]*up(?:\s*phase| period)?|Recovery|Post-study monitoring)\s*(?:of|:|-)?\s*(\d+)\s*weeks?', text, re.IGNORECASE)
 
     timeline = {
         "Screening": int(screening.group(1)) if screening else None,