Extend trial details

app.py

@@ -12,7 +12,8 @@ from utils import (
     get_similarities_among_diseases_uris,
     augment_the_set_of_diseaces,
     get_clinical_trials_related_to_diseases,
-    get_clinical_records_by_ids
+    get_clinical_records_by_ids,
+    render_trial_details
 )
 from llm_res import get_short_summary_out_of_json_files
 import json
@@ -165,31 +166,8 @@ with st.container():
 
         tab_titles = [f"{trial['protocolSection']['identificationModule']['nctId']}" for trial in trials]
 
-        def render_tab(trial: dict) -> None:
-            official_title = trial["protocolSection"]["identificationModule"]["officialTitle"]
-            st.write(f"##### {official_title}")
-
-            brief_summary = trial["protocolSection"]["descriptionModule"]["briefSummary"]
-            st.write(brief_summary)
-
-            status_module = {
-                "Status": trial["protocolSection"]["statusModule"]["overallStatus"],
-                "Status Date": trial["protocolSection"]["statusModule"]["statusVerifiedDate"],
-            }
-            st.write("###### Status")
-            st.table(status_module)
-
-            design_module = {
-                "Study Type": trial["protocolSection"]["designModule"]["studyType"],
-                # "Phases": trial["protocolSection"]["designModule"]["phases"], # breaks formatting because it is an array
-                "Allocation": trial["protocolSection"]["designModule"]["designInfo"]["allocation"],
-                "Participants": trial["protocolSection"]["designModule"]["enrollmentInfo"]["count"],
-            }
-            st.write("###### Design")
-            st.table(design_module)
-
         tabs = st.tabs(tab_titles)
 
         for i in range(0, len(tabs)):
             with tabs[i]:
-                render_tab(trials[i])
+                render_trial_details(trials[i])

utils.py

@@ -4,6 +4,7 @@ import os
 from sqlalchemy import create_engine, text
 import requests
 from sentence_transformers import SentenceTransformer
+import streamlit as st
 
 username = "demo"
 password = "demo"
@@ -190,6 +191,51 @@ def get_clinical_trials_related_to_diseases(
 
     return [{"nct_id": row[0], "distance": row[1]} for row in data]
 
+def to_capitalized_case(string: str) -> str:
+    string = string.replace("_", " ")
+    if string.isupper():
+        return string[0] + string[1:].lower()
+    
+def list_to_capitalized_case(strings: List[str]) -> str:
+    strings = [to_capitalized_case(s) for s in strings]
+    return ", ".join(strings)
+
+def render_trial_details(trial: dict) -> None:
+            # TODO: handle key errors for all cases (→ do not render)
+
+            official_title = trial["protocolSection"]["identificationModule"]["officialTitle"]
+            st.write(f"##### {official_title}")
+
+            brief_summary = trial["protocolSection"]["descriptionModule"]["briefSummary"]
+            st.write(brief_summary)
+
+            status_module = {
+                "Status": to_capitalized_case(trial["protocolSection"]["statusModule"]["overallStatus"]),
+                "Status Date": trial["protocolSection"]["statusModule"]["statusVerifiedDate"],
+                "Has Results": trial["hasResults"]
+            }
+            st.write("###### Status")
+            st.table(status_module)
+
+            design_module = {
+                "Study Type": to_capitalized_case(trial["protocolSection"]["designModule"]["studyType"]),
+                "Phases": list_to_capitalized_case(trial["protocolSection"]["designModule"]["phases"]),
+                "Allocation": to_capitalized_case(trial["protocolSection"]["designModule"]["designInfo"]["allocation"]),
+                "Primary Purpose": to_capitalized_case(trial["protocolSection"]["designModule"]["designInfo"]["primaryPurpose"]),
+                "Participants": trial["protocolSection"]["designModule"]["enrollmentInfo"]["count"],
+                "Masking": to_capitalized_case(trial["protocolSection"]["designModule"]["designInfo"]["maskingInfo"]["masking"]),
+                "Who Masked": list_to_capitalized_case(trial["protocolSection"]["designModule"]["designInfo"]["maskingInfo"]["whoMasked"])
+            }
+            st.write("###### Design")
+            st.table(design_module)
+
+            interventions_module = {}
+            for intervention in trial["protocolSection"]["armsInterventionsModule"]["interventions"]:
+                name = intervention["name"]
+                desc = intervention["description"]
+                interventions_module[name] = desc
+            st.write("###### Interventions")
+            st.table(interventions_module)
 
 if __name__ == "__main__":
     username = "demo"