logica de logo

app.py

@@ -1,5 +1,7 @@
 import streamlit as st
 from utils import carga_modelo, genera
+import os
+
 
 ## Página principal
 st.title('Generador de Mariposas')
@@ -7,7 +9,13 @@ st.write('Este es un modelo light GAN entrenado para generación de mariposas')
 
 ## Barra Lateral
 st.sidebar.subheader('Esta mariposa no existe, ¿Puedes creerlo?')
-st.sidebar.image('assets/logo.png', width=200)
+logo_path = 'assets/logo.png'
+
+if os.path.exists(logo_path):
+    st.sidebar.image(logo_path, width=200)
+else:
+    st.sidebar.write("⚠️ Logo not found.")
+
 st.sidebar.caption('Demo creado en vivo')
 
 ## Cargamos el modelo