import streamlit as st
import tifffile
import pydicom
from scipy.ndimage import zoom
import torch
from core.models.dani_model import dani_model
import numpy as np
from PIL import Image
import base64
import time
# Funzione per convertire un'immagine in base64
def image_to_base64(image_path):
with open(image_path, "rb") as img_file:
return base64.b64encode(img_file.read()).decode()
st.markdown("""
<style>
@import url('https://fonts.googleapis.com/css2?family=Roboto:wght@400;700&display=swap');
/* Apply the font to everything */
html, body, [class*="st"] {
font-family: 'Roboto', sans-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'}
}
# CSS per personalizzare il tema
st.markdown("""
<style>
/* Sfondo scuro */
body {
background-color: #121212;
color: white;
}
/* Personalizzazione del titolo */
.title {
font-size: 35px !important;
font-weight: bold;
color: #f63366;
}
/* Personalizzazione dei sottotitoli e testi principali */
.stText, .stButton, .stMarkdown {
font-size: 18px !important;
}
</style>
""", unsafe_allow_html=True)
# Sostituisci questo con il link dell'immagine online
logo_1_path = "./DEMO/Loghi/Logo_UCBM.png" # Sostituisci con il percorso del primo logo
logo_2_path = "./DEMO/Loghi/Logo UmU.png" # Sostituisci con il percorso del secondo logo
logo_3_path = "./DEMO/Loghi/Logo COSBI.png" # Sostituisci con il percorso del terzo logo
logo_4_path = "./DEMO/Loghi/logo trasparent.png" # Sostituisci con il percorso del quarto logo
# Converti le immagini in base64
logo_1_base64 = image_to_base64(logo_1_path)
logo_2_base64 = image_to_base64(logo_2_path)
logo_3_base64 = image_to_base64(logo_3_path)
logo_4_base64 = image_to_base64(logo_4_path)
# CSS per posizionare i loghi in basso a destra e renderli piccoli
st.markdown(f"""
<style>
.footer {{
position: fixed;
bottom: 20px;
right: 20px;
z-index: 100;
display: flex;
gap: 10px; /* Spazio tra i loghi */
}}
.footer img {{
height: 60px; /* Altezza dei loghi */
width: auto; /* Mantiene il rapporto di aspetto originale */
}}
</style>
<div class="footer">
<img src="data:image/png;base64,{logo_1_base64}" alt="Logo 1">
<img src="data:image/png;base64,{logo_2_base64}" alt="Logo 2">
<img src="data:image/png;base64,{logo_3_base64}" alt="Logo 3">
<img src="data:image/png;base64,{logo_4_base64}" alt="Logo 4">
</div>
""", unsafe_allow_html=True)
# Inizializzazione dello stato della sessione
if 'step' not in st.session_state:
st.session_state['step'] = 1
if 'selected_option' not in st.session_state:
st.session_state['selected_option'] = None
if 'frontal_file' not in st.session_state:
st.session_state['frontal_file'] = None
if 'lateral_file' not in st.session_state:
st.session_state['lateral_file'] = None
if 'report' not in st.session_state:
st.session_state['report'] = ""
if 'inputs' not in st.session_state:
st.session_state['inputs'] = None
if 'outputs' not in st.session_state:
st.session_state['outputs'] = None
if 'frontal' not in st.session_state:
st.session_state['frontal'] = None
if 'lateral' not in st.session_state:
st.session_state['lateral'] = None
if 'report' not in st.session_state:
st.session_state['report'] = ""
if 'generate' not in st.session_state:
st.session_state['generate'] = False
# Inizializza inference_tester solo una volta
if 'inference_tester' not in st.session_state:
model_load_paths = ['CoDi_encoders.pth', 'CoDi_text_diffuser.pth', 'CoDi_video_diffuser_8frames.pth']
st.session_state['inference_tester'] = dani_model(model='thesis_model',
data_dir='/mimer/NOBACKUP/groups/snic2022-5-277/dmolino/checkpoints/',
pth=model_load_paths, load_weights=False)
inference_tester = st.session_state['inference_tester']
# Caricamento dei pesi Clip, Optimus, Frontal, Lateral e Text una sola volta
if 'weights_loaded' not in st.session_state:
st.session_state['weights_loaded'] = True # Indica che i pesi sono stati caricati
# Usa inference_tester dalla sessione
inference_tester = st.session_state['inference_tester']
st.markdown('<h1 style="text-align: center" class="title">MedCoDi-M</h1>', unsafe_allow_html=True)
if st.session_state['step'] == 1:
# Breve descrizione del lavoro
st.markdown("""
<div style='text-align: justify; font-size: 18px; line-height: 1.6;'>
This work introduces MedCoDi-M, a novel multi-prompt foundation model for multi-modal medical data generation.
In this demo, you will be able to perform various generation tasks including frontal and lateral chest X-rays and clinical report generation.
MedCoDi-M enables flexible, any-to-any generation across different medical data modalities, utilizing contrastive learning and a modular approach for enhanced performance.
</div>
""", unsafe_allow_html=True)
# lasciamo un po' di spazio
st.markdown('<br>', unsafe_allow_html=True)
# 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)
# Caption con dimensione del testo migliorata
st.markdown("""
<div style='text-align: center; font-size: 16px; font-style: italic; margin-top: 10px;'>
Framework of MedCoDi-M: This demo allows you to generate frontal and lateral chest X-rays, as well as medical reports, through the MedCoDi-M model.
</div>
""", unsafe_allow_html=True)
# lasciamo un po' di spazio
st.markdown('<br>', unsafe_allow_html=True)
# Bottone con testo "Try it out"
if st.button("Try it out!"):
st.session_state['step'] = 2
st.rerun()
# Fase 1: Selezione dell'opzione
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"
]
# Messaggio di selezione con dimensione aumentata
st.markdown(
"<h4 style='text-align: justify'><strong>Select the type of generation you want to perform:</strong></h4>",
unsafe_allow_html=True)
# Aumentare la dimensione di "Please select an option:"
st.markdown(
"<h4 style='text-align: justify'><strong>Please select an option:</strong></h4>",
unsafe_allow_html=True)
# Reset esplicito del valore di `selectbox` in caso di reset
st.session_state['selected_option'] = st.selectbox(
"", options, key='selectbox_option', index=0) # Rimuoviamo il testo dal selectbox
st.markdown('<br>', unsafe_allow_html=True)
# Creiamo colonne per i pulsanti
col1, col2, col3 = st.columns(3)
# Pulsante per provare un esempio
with col1:
if st.button("Inference"):
st.session_state['step'] = 3 # Passa al passo 3
st.rerun()
# Pulsante per provare un esempio
with col2:
if st.button("Try an example"):
st.session_state['step'] = 5 # Passa al passo 5
st.rerun()
# Pulsante per tornare all'inizio
with col3:
if st.button("Return to the beginning"):
# Ripristina lo stato della sessione
st.session_state['step'] = 1
st.session_state['selected_option'] = None
st.session_state['selected_option2'] = None
st.session_state['frontal_file'] = None
st.session_state['lateral_file'] = None
st.session_state['report'] = ""
st.rerun()
# Fase 2: Caricamento file
if st.session_state['step'] == 3:
st.markdown(
f"<h4 style='text-align: justify'><strong>You selected: {st.session_state['selected_option']}. Now, please upload the required files below:</strong></h4>",
unsafe_allow_html=True)
# Carica l'immagine frontale
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]:
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]:
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'])
# lasciamo un po' di spazio
st.markdown('<br>', unsafe_allow_html=True)
# Creare colonne per allineare i pulsanti in orizzontale
col1, col2 = st.columns(2)
with col1:
if st.button("Start Generation"):
frontal = None
lateral = None
report = None
# Dato che questo step è velocissimo, prima di procedere mettiamo una finta barra di caricamento di 3 secondi
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']:
st.error("Load the Frontal image.")
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']:
st.error("Type the clinical report.")
else:
st.write(f"Execution of: {st.session_state['selected_option']}")
# Carica l'immagine e avvia l'inferenza
if st.session_state['frontal_file']:
dicom = pydicom.dcmread(st.session_state['frontal_file'])
image = dicom.pixel_array
if dicom.PhotometricInterpretation == 'MONOCHROME1':
image = (2 ** dicom.BitsStored - 1) - image
if dicom.ImagerPixelSpacing != [0.139, 0.139]:
zoom_factor = [0.139 / dicom.ImagerPixelSpacing[0], 0.139 / dicom.ImagerPixelSpacing[1]]
image = zoom(image, zoom_factor)
image = image / (2 ** dicom.BitsStored - 1)
# Se l'immagine non è quadrata, facciamo padding
if image.shape[0] != image.shape[1]:
diff = abs(image.shape[0] - image.shape[1])
pad_size = diff // 2
if image.shape[0] > image.shape[1]:
padded_image = np.pad(image, ((0, 0), (pad_size, pad_size)))
else:
padded_image = np.pad(image, ((pad_size, pad_size), (0, 0)))
# Resizing a 256x256 e a 512x512
zoom_factor = [256 / padded_image.shape[0], 256 / padded_image.shape[1]]
image_256 = zoom(padded_image, zoom_factor)
frontal = image_256
if frontal.dtype != np.uint8:
frontal2 = (255 * (frontal - frontal.min()) / (frontal.max() - frontal.min())).astype(np.uint8)
frontal = torch.tensor(frontal, dtype=torch.float32).unsqueeze(0).unsqueeze(0)
frontal2 = Image.fromarray(frontal2)
st.write("Frontal Image loaded successfully!")
# Mostra l'immagine caricata
st.image(frontal2, caption="Frontal Image Loaded", use_container_width=True)
if st.session_state['lateral_file']:
dicom = pydicom.dcmread(st.session_state['lateral_file'])
image = dicom.pixel_array
if dicom.PhotometricInterpretation == 'MONOCHROME1':
image = (2 ** dicom.BitsStored - 1) - image
if dicom.ImagerPixelSpacing != [0.139, 0.139]:
zoom_factor = [0.139 / dicom.ImagerPixelSpacing[0], 0.139 / dicom.ImagerPixelSpacing[1]]
image = zoom(image, zoom_factor)
image = image / (2 ** dicom.BitsStored - 1)
# Se l'immagine non è quadrata, facciamo padding
if image.shape[0] != image.shape[1]:
diff = abs(image.shape[0] - image.shape[1])
pad_size = diff // 2
if image.shape[0] > image.shape[1]:
padded_image = np.pad(image, ((0, 0), (pad_size, pad_size)))
else:
padded_image = np.pad(image, ((pad_size, pad_size), (0, 0)))
# Resizing a 256x256 e a 512x512
zoom_factor = [256 / padded_image.shape[0], 256 / padded_image.shape[1]]
image_256 = zoom(padded_image, zoom_factor)
lateral = image_256
if lateral.dtype != np.uint8:
lateral2 = (255 * (lateral - lateral.min()) / (lateral.max() - lateral.min())).astype(np.uint8)
lateral = torch.tensor(lateral, dtype=torch.float32).unsqueeze(0).unsqueeze(0)
lateral2 = Image.Frontalmarray(lateral2)
st.write("Lateral Image loaded successfully!")
st.image(lateral2, caption="Lateral Image Loaded", use_container_width=True)
if st.session_state['report']:
report = st.session_state['report']
st.write(f"Loaded Report: {report}")
inputs = []
if "Frontal" in st.session_state['selected_option'].split(" ➔")[0]:
inputs.append('frontal')
if "Lateral" in st.session_state['selected_option'].split(" ➔")[0]:
inputs.append('lateral')
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]:
outputs.append('frontal')
if "Lateral" in st.session_state['selected_option'].split(" ➔")[1]:
outputs.append('lateral')
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
# devono essere salvate nello stato della sessione
st.session_state['inputs'] = inputs
st.session_state['outputs'] = outputs
st.session_state['frontal'] = frontal
st.session_state['lateral'] = lateral
st.session_state['report'] = report
st.session_state['generate'] = True
st.session_state['step'] = 4
st.rerun()
with col2:
if st.button("Return to the beginning"):
# Ripristina lo stato della sessione
st.session_state['step'] = 1
st.session_state['selected_option'] = None
st.session_state['selected_option2'] = None
st.session_state['frontal_file'] = None
st.session_state['lateral_file'] = None
st.session_state['report'] = ""
st.rerun()
if st.session_state['step'] == 4:
# Costruzione del prompt
if st.session_state['generate'] is True:
conditioning = []
for inp in st.session_state['inputs']:
if inp == 'frontal':
cim = inference_tester.net.clip_encode_vision(st.session_state['frontal'], encode_type='encode_vision').to(device)
uim = inference_tester.net.clip_encode_vision(torch.zeros_like(st.session_state['frontal']).to(device),
encode_type='encode_vision').to(device)
conditioning.append(torch.cat([uim, cim]))
elif inp == 'lateral':
cim = inference_tester.net.clip_encode_vision(st.session_state['lateral'], encode_type='encode_vision').to(device)
uim = inference_tester.net.clip_encode_vision(torch.zeros_like(st.session_state['lateral']).to(device),
encode_type='encode_vision').to(device)
conditioning.append(torch.cat([uim, cim]))
elif inp == 'text':
ctx = inference_tester.net.clip_encode_text(1 * [st.session_state['report']], encode_type='encode_text').to(device)
utx = inference_tester.net.clip_encode_text(1 * [""], encode_type='encode_text').to(device)
conditioning.append(torch.cat([utx, ctx]))
# Costruzione delle shapes
shapes = []
for out in st.session_state['outputs']:
if out == 'frontal' or out == 'lateral':
shape = [1, 4, 256 // 8, 256 // 8]
shapes.append(shape)
elif out == 'text':
shape = [1, 768]
shapes.append(shape)
progress_bar = st.progress(0)
# Inferenza
z, _ = inference_tester.sampler.sample(
steps=50,
shape=shapes,
condition=conditioning,
unconditional_guidance_scale=7.5,
xtype=st.session_state['outputs'],
condition_types=st.session_state['inputs'],
eta=1,
verbose=False,
mix_weight={'lateral': 1, 'text': 1, 'frontal': 1},
progress_bar=progress_bar)
# Decoder e visualizzazione dei risultati
output_cols = st.columns(len(st.session_state['outputs']))
# Definire due colonne per le immagini
col1, col2 = st.columns(2)
# Iterare sugli output e assegnare le immagini alle colonne corrispondenti
for i, out in enumerate(st.session_state['outputs']):
if out == 'frontal':
x = inference_tester.net.autokl_decode(z[i])
x = torch.clamp((x[0] + 1.0) / 2.0, min=0.0, max=1.0)
im = x[0].cpu().numpy()
with col1: # Mostrare la frontal image nella prima colonna
st.image(im, caption="Generated Frontal Image")
elif out == 'lateral':
x = inference_tester.net.autokl_decode(z[i])
x = torch.clamp((x[0] + 1.0) / 2.0, min=0.0, max=1.0)
im = x[0].cpu().numpy()
with col2: # Mostrare la lateral image nella seconda colonna
st.image(im, caption="Generated Lateral Image")
elif out == 'text':
x = inference_tester.net.optimus_decode(z[i], max_length=100)
x = [a.tolist() for a in x]
rec_text = [inference_tester.net.optimus.tokenizer_decoder.decode(a) for a in x]
rec_text = rec_text[0].replace('<BOS>', '').replace('<EOS>', '')
st.write(f"Generated Report: {rec_text}")
st.write("Generation completed successfully!")
st.session_state['generate'] = False
if st.button("Return to the beginning"):
# Ripristina lo stato della sessione
st.session_state['generate'] = False
st.session_state['step'] = 1
st.session_state['selected_option'] = None
st.session_state['frontal_file'] = None
st.session_state['lateral_file'] = None
st.session_state['report'] = ""
st.session_state['inputs'] = None
st.session_state['outputs'] = None
st.session_state['frontal'] = None
st.session_state['lateral'] = None
st.session_state['report'] = ""
st.rerun()
if st.session_state['step'] == 5:
st.markdown(
f"<h4 style='text-align: justify'><strong>You selected: {st.session_state['selected_option']}</strong></h4>",
unsafe_allow_html=True)
inputs = []
if "Frontal" in st.session_state['selected_option'].split(" ➔")[0]:
inputs.append('Frontal')
if "Lateral" in st.session_state['selected_option'].split(" ➔")[0]:
inputs.append('Lateral')
if "Report" in st.session_state['selected_option'].split(" ➔")[0]:
inputs.append('Report')
outputs = []
if "Frontal" in st.session_state['selected_option'].split(" ➔")[1]:
outputs.append('Frontal')
if "Lateral" in st.session_state['selected_option'].split(" ➔")[1]:
outputs.append('Lateral')
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: center'><strong>INPUT:</strong></h3>",
unsafe_allow_html=True)
# Colonne per gli INPUTS
input_cols = st.columns(len(inputs))
for idx, inp in enumerate(inputs):
with input_cols[idx]:
if inp == 'Frontal':
path = "./DEMO/ESEMPI/" + esempio['Frontal']
print(path)
if path.endswith(".tiff"):
im = tifffile.imread(path)
im = np.clip(im, 0, 1)
elif path.endswith(".png"):
im = Image.open(path)
st.image(im, caption="Frontal Image")
elif inp == 'Lateral':
path = "./DEMO/ESEMPI/" + esempio['Lateral']
if path.endswith(".tiff"):
im = tifffile.imread(path)
im = np.clip(im, 0, 1)
elif path.endswith(".png"):
im = Image.open(path)
st.image(im, caption="Lateral Image")
elif inp == 'Report':
st.markdown(
f"<p style='font-size:20px;'><strong>Report:</strong> {esempio['Report']}</p>",
unsafe_allow_html=True
)
st.markdown(
"<h3 style='text-align: center'><strong>OUTPUT:</strong></h3>",
unsafe_allow_html=True)
# Colonne per gli OUTPUTS
output_cols = st.columns(len(outputs))
for idx, out in enumerate(outputs):
with output_cols[idx]:
if out == 'Frontal':
path = "./DEMO/ESEMPI/" + esempio['Frontal']
if path.endswith(".tiff"):
im = tifffile.imread(path)
# facciamo clamp tra 0 e 1
im = np.clip(im, 0, 1)
elif path.endswith(".png"):
im = Image.open(path)
st.image(im, caption="Frontal Image")
elif out == 'Lateral':
path = "./DEMO/ESEMPI/" + esempio['Lateral']
if path.endswith(".tiff"):
im = tifffile.imread(path)
# facciamo clamp tra 0 e 1
im = np.clip(im, 0, 1)
elif path.endswith(".png"):
im = Image.open(path)
st.image(im, caption="Lateral Image")
elif out == 'Report':
st.markdown(
f"<p style='font-size:20px;'><strong>Report:</strong> {esempio['Report']}</p>",
unsafe_allow_html=True
)
# Pulsante per tornare all'inizio
if st.button("Return to the beginning"):
# Ripristina lo stato della sessione
st.session_state['step'] = 1
st.session_state['selected_option'] = None
st.session_state['selected_option2'] = None
st.session_state['frontal_file'] = None
st.session_state['lateral_file'] = None
st.session_state['report'] = ""
st.rerun()