Update main.py

main.py

@@ -4,12 +4,10 @@ from PIL import Image
 import sys
 import os
 
-# Obtenemos la ruta absoluta de la carpeta actual donde se ejecuta el script
+# --- CORRECCIÓN DE RUTAS ---
 current_dir = os.path.dirname(os.path.abspath(__file__))
-# Definimos la ruta de la carpeta 'src' relativa a este archivo
-src_path = os.path.join(current_dir, 'src')
-# Agregamos 'src' al path de Python para que pueda encontrar los módulos
-sys.path.append(src_path)
+src_path = os.path.join(current_dir, 'src') # Quitamos la barra inicial '/' que causaba error en Linux
+sys.path.append(os.path.abspath(src_path))
 
 try:
     from inference import MemePredictor
@@ -27,31 +25,31 @@ st.set_page_config(
 # --- 2. ESTILOS CSS "APPLE STYLE" ---
 st.markdown("""
     <style>
-        /* Tipografía limpia (System Fonts) */
+        /* Tipografía limpia */
         html, body, [class*="css"] {
             font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, Helvetica, Arial, sans-serif;
             color: #1d1d1f;
         }
         
-        /* Ocultar elementos nativos de Streamlit */
+        /* Ocultar elementos nativos */
         #MainMenu {visibility: hidden;}
         footer {visibility: hidden;}
         header {visibility: hidden;}
         
-        /* Contenedor principal con más aire */
         .block-container {
             padding-top: 3rem;
             padding-bottom: 3rem;
             max-width: 700px;
         }
-/* --- BOTÓN ANALIZAR --- */
+
+        /* Botón Principal */
         div.stButton > button:first-child {
-            background-color: #0071e3; /* Azul Apple */
+            background-color: #0071e3;
             color: white;
             border: none;
             border-radius: 12px;
-            padding: 15px 30px; /* Más grande */
-            font-size: 18px;    /* Texto más grande */
+            padding: 15px 30px;
+            font-size: 18px;
             font-weight: 600;
             width: 100%;
             transition: all 0.2s ease;
@@ -63,12 +61,13 @@ st.markdown("""
             transform: scale(1.01);
         }
 
-        /* --- RADIO BUTTON (La bolita azul) --- */
+        /* Radio Button */
         div[role="radiogroup"] label > div:first-child {
             background-color: #0071e3 !important;
             border-color: #0071e3 !important;
         }
-        /* Estilo de las Métricas (Tarjetas grises tipo iOS) */
+
+        /* Tarjetas de Métricas */
         div[data-testid="stMetric"] {
             background-color: #f5f5f7;
             border-radius: 18px;
@@ -88,19 +87,17 @@ st.markdown("""
             color: #1d1d1f;
         }
 
-        /* Imágenes con bordes redondeados */
+        /* Imágenes */
         img {
             border-radius: 18px;
             box-shadow: 0 4px 20px rgba(0,0,0,0.08);
         }
         
-        /* File Uploader más limpio */
         div[data-testid="stFileUploader"] {
             margin-top: 20px;
             margin-bottom: 20px;
         }
         
-        /* Títulos */
         h1, h2, h3 {
             font-weight: 600;
             letter-spacing: -0.02em;
@@ -116,36 +113,33 @@ def get_engine():
 try:
     predictor = get_engine()
 except Exception as e:
-    st.error("El servicio no está disponible en este momento.")
+    st.error(f"Error iniciando el motor: {e}")
     st.stop()
 
 # --- 4. INTERFAZ DE USUARIO ---
 
-# Encabezado Minimalista
 st.markdown("<h1 style='text-align: center; margin-bottom: 10px;'>DIME-MEX</h1>", unsafe_allow_html=True)
 st.markdown("<p style='text-align: center; color: #86868b; font-size: 18px;'>Análisis de contenido mediante IA.</p>", unsafe_allow_html=True)
 
-st.write("") # Espaciador
+st.write("") 
 
-# Configuración (Acordeón limpio)
+# Configuración
 with st.expander("Elegir modelo"):
     task_mode = st.radio(
         "Sensibilidad del modelo",
         options=["simple", "complex"],
-        format_func=lambda x: "Estándar ( none, inappropriate, hate-speech )" if x == "simple" else "Detallado ( none, inappropriate, sexism, racism, classicis, hate-speech )",
+        format_func=lambda x: "Estándar ( none, inappropriate, hate-speech )" if x == "simple" else "Detallado ( 6 clases )",
         label_visibility="collapsed"
     )
 
-# Área de Carga
+# Carga
 uploaded_file = st.file_uploader("Subir imagen", type=["jpg", "png", "jpeg"], label_visibility="collapsed")
     
 if uploaded_file is not None:
-    # Mostrar imagen
     image = Image.open(uploaded_file)
-    # CORRECCIÓN: Usamos use_container_width en lugar de use_column_width
     st.image(image, use_container_width=True)
     
-    st.write("") # Espaciador vertical
+    st.write("") 
     
     # Botón de Acción
     if st.button("Analizar"):
@@ -154,11 +148,12 @@ if uploaded_file is not None:
                 result = predictor.predict(uploaded_file, task=task_mode)
                 
                 if "error" in result:
-                    st.error("No se pudo procesar la imagen.")
+                    # Mostrar error lógico (del backend)
+                    st.error(f"Error de procesamiento: {result['error']}")
                 else:
-                    st.write("") # Espaciador
+                    st.write("") 
                     
-                    # --- RESULTADOS (Tarjetas) ---
+                    # --- RESULTADOS ---
                     col1, col2 = st.columns(2)
                     
                     with col1:
@@ -167,20 +162,37 @@ if uploaded_file is not None:
                     with col2:
                         st.metric(label="Certeza", value=f"{result['confidence']:.1%}")
                     
-                    # --- DETALLES (Acordeón inferior) ---
+                    st.progress(result['confidence'])
+                    
+                    # --- DETALLES ---
                     st.write("")
                     with st.expander("Detalles del análisis"):
                         st.caption("TEXTO DETECTADO")
-                        st.markdown(f"_{result['ocr_text']}_")
+                        # Usamos .get() para evitar errores si la clave no existe
+                        st.markdown(f"_{result.get('ocr_text', 'No disponible')}_")
                         
                         st.write("")
                         st.caption("PROBABILIDADES")
-                        # Crear dataframe limpio para la gráfica
-                        chart_data = pd.DataFrame({
-                            "Categoría": result['all_labels'],
-                            "Probabilidad": result['probabilities']
-                        })
-                        st.bar_chart(chart_data.set_index("Categoría"), color="#86868b")
-
-            except Exception:
-                st.error("Ocurrió un error inesperado durante el análisis.")
+                        
+                        if 'all_labels' in result and 'probabilities' in result:
+                            chart_data = pd.DataFrame({
+                                "Categoría": result['all_labels'],
+                                "Probabilidad": result['probabilities']
+                            })
+                            st.bar_chart(chart_data.set_index("Categoría"), color="#86868b")
+
+            # --- CAPTURA DE ERROR TÉCNICO (AQUÍ VERÁS QUÉ PASA) ---
+            except Exception as e:
+                st.error(f"Error técnico crítico: {str(e)}")
+                # Pista probable
+                if "libGL" in str(e):
+                    st.warning("Falta el archivo 'packages.txt' con libgl1 en el repositorio.")
+else:
+    st.markdown(
+        """
+        <div style='text-align: center; color: #86868b; padding: 40px; border: 2px dashed #e5e5e5; border-radius: 18px; margin-top: 20px;'>
+            Arrastra una imagen aquí o haz clic para explorar
+        </div>
+        """, 
+        unsafe_allow_html=True
+    )