Update app.py

app.py

@@ -7,7 +7,7 @@ def extract_study_timeline(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)
+    treatment = re.search(r'(?:Treatment(?:\s*period)?|Intervention|Therapy|Dosing phase)[^,]*?\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 = {