Update app.py

app.py

@@ -16,22 +16,39 @@ def image_to_base64(image_path):
         return base64.b64encode(img_file.read()).decode()
 
 
+st.markdown("""
+    <style>
+    @font-face {
+        font-family: 'Palatino';
+        font-style: normal;
+        font-weight: 400;
+        src: local('Palatino Linotype'), local('Book Antiqua'),
+             url(https://fonts.gstatic.com/s/cardo/v22/wlp_gwIBjINFON1YNFX6WcrrlQ.ttf) format('truetype');
+    }
+    /* Applica il font a tutto */
+    html, body, [class*="st"] {
+        font-family: 'Palatino', serif;
+    }
+    </style>
+    """, unsafe_allow_html=True)
+
+
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 # Dati di esempio predefiniti
 esempi = {
-    "Frontal -> Lateral": {'Frontal': 'FtoL.png', 'Lateral': 'LfromF.png'},
-    "Frontal -> Report": {'Frontal': '31d9847f-987fcf63-704f7496-d2b21eb8-63cd973e.tiff', 'Report': 'Small bilateral pleural effusions, left greater than right.'},
-    "Frontal -> Lateral + Report": {'Frontal': '81bca127-0c416084-67f8033c-ecb26476-6d1ecf60.tiff', 'Lateral': 'd52a0c5c-bb7104b0-b1d821a5-959984c3-33c04ccb.tiff', 'Report': 'No acute intrathoracic process. Heart Size is normal. Lungs are clear. No pneumothorax'},
-    "Lateral -> Frontal": {'Lateral': 'LtoF.png', 'Frontal': 'FfromL.png'},
-    "Lateral -> Report": {'Lateral': 'd52a0c5c-bb7104b0-b1d821a5-959984c3-33c04ccb.tiff', 'Report': 'no acute cardiopulmonary process. if concern for injury persists, a dedicated rib series with markers would be necessary to ensure no rib fractures.'},
-    "Lateral -> Frontal + Report": {'Lateral': 'reald52a0c5c-bb7104b0-b1d821a5-959984c3-33c04ccb.tiff', 'Frontal': 'ab37274f-b4c1fc04-e2ff24b4-4a130ba3-cd167968.tiff', 'Report': 'No acute intrathoracic process. If there is strong concern for rib fracture, a dedicated rib series may be performed.'},
-    "Report -> Frontal": {'Report': 'Left lung opacification which may reflect pneumonia superimposed on metastatic disease.', 'Frontal': '02aa804e-bde0afdd-112c0b34-7bc16630-4e384014.tiff'},
-    "Report -> Lateral": {'Report': 'Bilateral pleural effusions, cardiomegaly and mild edema suggest fluid overload.', 'Lateral': '489faba7-a9dc5f1d-fd7241d6-9638d855-eaa952b1.tiff'},
-    "Report -> Frontal + Lateral": {'Report': 'No acute intrathoracic process. The lungs are clean and heart is normal size.', 'Frontal': 'f27ba7cd-44486c2e-29f3e890-f2b9f94e-84110448.tiff', 'Lateral': 'b20c9570-de77944a-b8604ba0-73305a7b-d608a72b.tiff'},
-    "Frontal + Lateral -> Report": {'Frontal': '95856dd1-5878b5b1-9c104817-760c0122-6187946f.tiff', 'Lateral': '3723d912-71940d69-4fef2dd2-27af5a7b-127ba20c.tiff', 'Report': 'Opacities in the right upper or middle lobe, maybe early pneumonia.'},
-    "Frontal + Report -> Lateral": {'Frontal': 'e7f21453-7956d79a-44e44614-fae8ff16-d174d1a0.tiff', 'Report': 'No focal consolidation.', 'Lateral': '8037e6b9-06367464-a4ccd63a-5c5c5a81-ce3e7ffc.tiff'},
-    "Lateral + Report -> Frontal": {'Lateral': '02c66644-b1883a91-54aed0e7-62d25460-398f9865.tiff', 'Report': 'No evidence of acute cardiopulmonary process.', 'Frontal': 'b1f169f1-12177dd5-2fa1c4b1-7b816311-85d769e9.tiff'}
+    "Frontal ➔ Lateral": {'Frontal': 'FtoL.png', 'Lateral': 'LfromF.png'},
+    "Frontal ➔ Report": {'Frontal': '31d9847f-987fcf63-704f7496-d2b21eb8-63cd973e.tiff', 'Report': 'Small bilateral pleural effusions, left greater than right.'},
+    "Frontal ➔ Lateral + Report": {'Frontal': '81bca127-0c416084-67f8033c-ecb26476-6d1ecf60.tiff', 'Lateral': 'd52a0c5c-bb7104b0-b1d821a5-959984c3-33c04ccb.tiff', 'Report': 'No acute intrathoracic process. Heart Size is normal. Lungs are clear. No pneumothorax'},
+    "Lateral ➔ Frontal": {'Lateral': 'LtoF.png', 'Frontal': 'FfromL.png'},
+    "Lateral ➔ Report": {'Lateral': 'd52a0c5c-bb7104b0-b1d821a5-959984c3-33c04ccb.tiff', 'Report': 'no acute cardiopulmonary process. if concern for injury persists, a dedicated rib series with markers would be necessary to ensure no rib fractures.'},
+    "Lateral ➔ Frontal + Report": {'Lateral': 'reald52a0c5c-bb7104b0-b1d821a5-959984c3-33c04ccb.tiff', 'Frontal': 'ab37274f-b4c1fc04-e2ff24b4-4a130ba3-cd167968.tiff', 'Report': 'No acute intrathoracic process. If there is strong concern for rib fracture, a dedicated rib series may be performed.'},
+    "Report ➔ Frontal": {'Report': 'Left lung opacification which may reflect pneumonia superimposed on metastatic disease.', 'Frontal': '02aa804e-bde0afdd-112c0b34-7bc16630-4e384014.tiff'},
+    "Report ➔ Lateral": {'Report': 'Bilateral pleural effusions, cardiomegaly and mild edema suggest fluid overload.', 'Lateral': '489faba7-a9dc5f1d-fd7241d6-9638d855-eaa952b1.tiff'},
+    "Report ➔ Frontal + Lateral": {'Report': 'No acute intrathoracic process. The lungs are clean and heart is normal size.', 'Frontal': 'f27ba7cd-44486c2e-29f3e890-f2b9f94e-84110448.tiff', 'Lateral': 'b20c9570-de77944a-b8604ba0-73305a7b-d608a72b.tiff'},
+    "Frontal + Lateral ➔ Report": {'Frontal': '95856dd1-5878b5b1-9c104817-760c0122-6187946f.tiff', 'Lateral': '3723d912-71940d69-4fef2dd2-27af5a7b-127ba20c.tiff', 'Report': 'Opacities in the right upper or middle lobe, maybe early pneumonia.'},
+    "Frontal + Report ➔ Lateral": {'Frontal': 'e7f21453-7956d79a-44e44614-fae8ff16-d174d1a0.tiff', 'Report': 'No focal consolidation.', 'Lateral': '8037e6b9-06367464-a4ccd63a-5c5c5a81-ce3e7ffc.tiff'},
+    "Lateral + Report ➔ Frontal": {'Lateral': '02c66644-b1883a91-54aed0e7-62d25460-398f9865.tiff', 'Report': 'No evidence of acute cardiopulmonary process.', 'Frontal': 'b1f169f1-12177dd5-2fa1c4b1-7b816311-85d769e9.tiff'}
 }
 
 
@@ -141,7 +158,7 @@ if st.session_state['step'] == 1:
 
     # Immagine con didascalia migliorata e con dimensione della caption aumentata
     image_path = "./DEMO/Loghi/model_final.png"  # Sostituisci con il percorso della tua immagine
-    st.image(image_path, caption='', use_container_width=True)
+    st.image(image_path, caption='', use_column_width=True)
 
     # Caption con dimensione del testo migliorata
     st.markdown("""
@@ -163,10 +180,10 @@ if st.session_state['step'] == 1:
 if st.session_state['step'] == 2:
     # Opzioni disponibili
     options = [
-        "Frontal -> Lateral", "Frontal -> Report", "Frontal -> Lateral + Report",
-        "Lateral -> Frontal", "Lateral -> Report", "Lateral -> Frontal + Report",
-        "Report -> Frontal", "Report -> Lateral", "Report -> Frontal + Lateral",
-        "Frontal + Lateral -> Report", "Frontal + Report -> Lateral", "Lateral + Report -> Frontal"
+        "Frontal ➔ Lateral", "Frontal ➔ Report", "Frontal ➔ Lateral + Report",
+        "Lateral ➔ Frontal", "Lateral ➔ Report", "Lateral ➔ Frontal + Report",
+        "Report ➔ Frontal", "Report ➔ Lateral", "Report ➔ Frontal + Lateral",
+        "Frontal + Lateral ➔ Report", "Frontal + Report ➔ Lateral", "Lateral + Report ➔ Frontal"
     ]
 
     # Messaggio di selezione con dimensione aumentata
@@ -220,17 +237,17 @@ if st.session_state['step'] == 3:
         unsafe_allow_html=True)
 
     # Carica l'immagine frontale
-    if "Frontal" in st.session_state['selected_option'].split(" ->")[0]:
+    if "Frontal" in st.session_state['selected_option'].split(" ➔")[0]:
         st.markdown("<h5 style='font-size: 18px;'>Load the Frontal X-ray in DICOM format</h5>", unsafe_allow_html=True)
         st.session_state['frontal_file'] = st.file_uploader("", type=["dcm"])
 
     # Carica l'immagine laterale
-    if "Lateral" in st.session_state['selected_option'].split(" ->")[0]:
+    if "Lateral" in st.session_state['selected_option'].split(" ➔")[0]:
         st.markdown("<h5 style='font-size: 18px;'>Load the Lateral X-ray in DICOM format</h5>", unsafe_allow_html=True)
         st.session_state['lateral_file'] = st.file_uploader("", type=["dcm"])
 
     # Inserisci il report clinico
-    if "Report" in st.session_state['selected_option'].split(" ->")[0]:
+    if "Report" in st.session_state['selected_option'].split(" ➔")[0]:
         st.markdown("<h5 style='font-size: 18px;'>Type the clinical report</h5>", unsafe_allow_html=True)
         st.session_state['report'] = st.text_area("", value=st.session_state['report'])
 
@@ -249,11 +266,11 @@ if st.session_state['step'] == 3:
             with st.spinner("Preprocessing the data..."):
                 time.sleep(3)
             # Controllo che i file necessari siano stati caricati
-            if "Frontal" in st.session_state['selected_option'].split(" ->")[0] and not st.session_state['frontal_file']:
+            if "Frontal" in st.session_state['selected_option'].split(" ➔")[0] and not st.session_state['frontal_file']:
                 st.error("Load the Frontal image.")
-            elif "Lateral" in st.session_state['selected_option'].split(" ->")[0] and not st.session_state['lateral_file']:
+            elif "Lateral" in st.session_state['selected_option'].split(" ➔")[0] and not st.session_state['lateral_file']:
                 st.error("Load the Lateral image.")
-            elif "Report" in st.session_state['selected_option'].split(" ->")[0] and not st.session_state['report']:
+            elif "Report" in st.session_state['selected_option'].split(" ➔")[0] and not st.session_state['report']:
                 st.error("Type the clinical report.")
             else:
                 st.write(f"Execution of: {st.session_state['selected_option']}")
@@ -319,20 +336,20 @@ if st.session_state['step'] == 3:
                     st.write(f"Loaded Report: {report}")
 
                 inputs = []
-                if "Frontal" in st.session_state['selected_option'].split(" ->")[0]:
+                if "Frontal" in st.session_state['selected_option'].split(" ➔")[0]:
                     inputs.append('frontal')
-                if "Lateral" in st.session_state['selected_option'].split(" ->")[0]:
+                if "Lateral" in st.session_state['selected_option'].split(" ➔")[0]:
                     inputs.append('lateral')
-                if "Report" in st.session_state['selected_option'].split(" ->")[0]:
+                if "Report" in st.session_state['selected_option'].split(" ➔")[0]:
                     inputs.append('text')
 
                 # Ora vediamo cosa c'è dopo la freccia
                 outputs = []
-                if "Frontal" in st.session_state['selected_option'].split(" ->")[1]:
+                if "Frontal" in st.session_state['selected_option'].split(" ➔")[1]:
                     outputs.append('frontal')
-                if "Lateral" in st.session_state['selected_option'].split(" ->")[1]:
+                if "Lateral" in st.session_state['selected_option'].split(" ➔")[1]:
                     outputs.append('lateral')
-                if "Report" in st.session_state['selected_option'].split(" ->")[1]:
+                if "Report" in st.session_state['selected_option'].split(" ➔")[1]:
                     outputs.append('text')
 
                 # Ultima cosa che va fatta è passare allo step 4, prima di farlo però, tutte le variabili che ci servono
@@ -383,26 +400,26 @@ if st.session_state['step'] == 5:
         unsafe_allow_html=True)
 
     inputs = []
-    if "Frontal" in st.session_state['selected_option'].split(" ->")[0]:
+    if "Frontal" in st.session_state['selected_option'].split(" ➔")[0]:
         inputs.append('Frontal')
-    if "Lateral" in st.session_state['selected_option'].split(" ->")[0]:
+    if "Lateral" in st.session_state['selected_option'].split(" ➔")[0]:
         inputs.append('Lateral')
-    if "Report" in st.session_state['selected_option'].split(" ->")[0]:
+    if "Report" in st.session_state['selected_option'].split(" ➔")[0]:
         inputs.append('Report')
 
     outputs = []
-    if "Frontal" in st.session_state['selected_option'].split(" ->")[1]:
+    if "Frontal" in st.session_state['selected_option'].split(" ➔")[1]:
         outputs.append('Frontal')
-    if "Lateral" in st.session_state['selected_option'].split(" ->")[1]:
+    if "Lateral" in st.session_state['selected_option'].split(" ➔")[1]:
         outputs.append('Lateral')
-    if "Report" in st.session_state['selected_option'].split(" ->")[1]:
+    if "Report" in st.session_state['selected_option'].split(" ➔")[1]:
         outputs.append('Report')
 
     esempio = esempi[st.session_state['selected_option']]
 
     # Mostra i file associati all'esempio
     st.markdown(
-        "<h3 style='text-align: justify'><strong>INPUTS</strong></h3>",
+        "<h3 style='text-align: center'><strong>INPUT:</strong></h3>",
         unsafe_allow_html=True)
 
     # Colonne per gli INPUTS
@@ -431,7 +448,7 @@ if st.session_state['step'] == 5:
                 st.write(f"Report: {esempio['Report']}")
 
     st.markdown(
-        "<h3 style='text-align: justify'><strong>OUTPUTS</strong></h3>",
+        "<h3 style='text-align: center'><strong>OUTPUT:</strong></h3>",
         unsafe_allow_html=True)
 
     # Colonne per gli OUTPUTS