Update src/streamlit_app.py

src/streamlit_app.py

@@ -10,6 +10,7 @@ import pandas as pd
 import random
 import os
 import io
+import tempfile
 
 # Configuración de la página
 st.set_page_config(page_title="🌍 Analizador de tiempos y movimientos Avanzado", layout="wide")
@@ -29,12 +30,17 @@ app_image = load_app_image()
 if app_image:
     st.image(app_image, use_container_width=True)
 
-# Función para procesar GPX
-def process_gpx_data(gpx_content):
-    """Procesa contenido GPX y extrae datos para mapas"""
+# Función para procesar GPX sin caché primero para debugging
+def process_gpx_simple(file_content_or_path, is_file_path=False):
+    """Procesa archivo GPX de manera simple y directa"""
     try:
-        # Parsear GPX
-        gpx = gpxpy.parse(gpx_content)
+        if is_file_path:
+            # Es una ruta de archivo (ejemplo)
+            with open(file_content_or_path, 'r', encoding='utf-8') as f:
+                gpx = gpxpy.parse(f)
+        else:
+            # Es contenido de archivo (cargado)
+            gpx = gpxpy.parse(file_content_or_path)
         
         # Extraer datos
         heatmap_data = []
@@ -155,191 +161,101 @@ example_files = []
 if os.path.exists(examples_folder):
     example_files = [f for f in os.listdir(examples_folder) if f.endswith(".gpx")]
 
-# ========== INICIO LÓGICA EXACTA DE APPTEST.PY CON ADAPTACIONES PARA HF SPACES ==========
-
-# Opciones de selección de datos de ejemplo
-st.subheader("Datos de Ejemplo")
-usar_datos_ejemplo = False
-archivo_ejemplo_seleccionado = None
-
-if example_files:
-    # Crear checkbox para cada archivo de ejemplo - IGUAL QUE EN APPTEST.PY
-    for example_file in example_files:
-        if st.checkbox(f"Usar datos de ejemplo: {example_file}"):
-            if not usar_datos_ejemplo:  # Solo permitir uno a la vez
-                usar_datos_ejemplo = True
-                archivo_ejemplo_seleccionado = example_file
+# Interface de carga
+uploaded_file = st.file_uploader("📁 Cargar archivo GPX", type="gpx")
+if uploaded_file is not None:
+    selected_example = "Ninguno"  # Forzar que no se use el ejemplo si se carga archivo
+else:
+    selected_example = st.selectbox("📋 O selecciona un ejemplo:", ["Ninguno"] + example_files)
 
-# Inicializar variables de datos - IGUAL QUE EN APPTEST.PY
-data = None
-gpx_content = None
+gpx_data = None
+file_name = None
 
-# Cargar archivo de ejemplo si está seleccionado - IGUAL QUE EN APPTEST.PY
-if usar_datos_ejemplo and archivo_ejemplo_seleccionado:
+if uploaded_file is not None:
+    selected_example = "Ninguno"  # Ignorar ejemplos si se carga archivo
+    file_name = uploaded_file.name
+    st.info(f"📄 Archivo seleccionado: {file_name}")
     try:
-        example_path = os.path.join(examples_folder, archivo_ejemplo_seleccionado)
-        # USANDO PANDAS COMO EN APPTEST.PY para mantener consistencia
-        with open(example_path, 'r', encoding='utf-8') as f:
-            gpx_content = f.read()
-        
-        st.success(f"Datos de ejemplo ({archivo_ejemplo_seleccionado}) cargados exitosamente.")
-        
-        # Crear un DataFrame simulado para mantener la lógica de apptest.py
-        data = pd.DataFrame({'status': ['loaded'], 'source': ['example'], 'filename': [archivo_ejemplo_seleccionado]})
-        st.dataframe(data, use_container_width=True)
-        
+        # Leer contenido como texto directamente
+        file_bytes = uploaded_file.read()
+        file_content = file_bytes.decode("utf-8", errors="ignore")
+
+        if file_content and "<?xml" in file_content:
+            st.success("✅ Archivo GPX leído correctamente")
+            with st.spinner("🔄 Procesando archivo GPX..."):
+                gpx_data = process_gpx_simple(file_content)
+        else:
+            st.error("❌ El archivo no parece ser un GPX válido")
     except Exception as e:
-        st.error(f"Error cargando archivo de ejemplo: {e}")
+        st.error(f"❌ Error procesando archivo: {e}")
 
-# Opción para cargar datos personalizados - EXACTO COMO APPTEST.PY
-st.subheader("Carga tus Datos")
-uploaded_file = st.file_uploader("Carga un archivo GPX con los datos experimentales:", type="gpx")
+elif selected_example != "Ninguno":
+    file_name = selected_example
+    file_path = os.path.join(examples_folder, selected_example)
+    st.info(f"📄 Ejemplo seleccionado: {file_name}")
+    with st.spinner("🔄 Procesando archivo de ejemplo..."):
+        gpx_data = process_gpx_simple(file_path, is_file_path=True)
 
-if uploaded_file is not None:
-    try:
-        # MÉTODO EXACTO DE APPTEST.PY PARA HUGGING FACE SPACES
-        # Primero intentar como lo hace apptest.py con pandas (aunque no sea CSV)
-        try:
-            # Leer como bytes primero
-            file_bytes = uploaded_file.getvalue()
-            
-            # Intentar decodificar con diferentes encodings como hace apptest.py internamente
-            gpx_content = None
-            for encoding in ['utf-8', 'latin-1', 'cp1252', 'iso-8859-1']:
-                try:
-                    gpx_content = file_bytes.decode(encoding)
-                    break
-                except UnicodeDecodeError:
-                    continue
-            
-            if gpx_content is None:
-                # Fallback: usar el método original de streamlit
-                uploaded_file.seek(0)  # Reset file pointer
-                gpx_content = uploaded_file.read().decode('utf-8')
+# MOSTRAR RESULTADOS
+if gpx_data and gpx_data['success']:
+    st.success("🎉 ¡Archivo procesado exitosamente!")
+    
+    # Métricas
+    heatmap_data = gpx_data['heatmap_data']
+    track_segments = gpx_data['track_segments']
+    total_distance = gpx_data['total_distance']
+    total_time = gpx_data['total_time']
+    center = gpx_data['center']
+    
+    # Dashboard
+    st.markdown("### 📊 Resumen de la Ruta")
+    col1, col2, col3 = st.columns(3)
+    
+    with col1:
+        st.metric("📏 Distancia", f"{total_distance/1000:.2f} km" if total_distance > 0 else "N/A")
+    with col2:
+        hours = int(total_time // 3600)
+        minutes = int((total_time % 3600) // 60)
+        st.metric("⏱️ Tiempo", f"{hours}h {minutes}m" if total_time > 0 else "N/A")
+    with col3:
+        st.metric("📍 Puntos GPS", f"{len(heatmap_data):,}")
+    
+    # Crear y mostrar mapas
+    st.markdown("### 🗺️ Visualización de Movimientos")
+    
+    with st.spinner("🗺️ Generando mapas..."):
+        heat_map, spaghetti_map = create_maps(heatmap_data, track_segments, center)
+    
+    if heat_map and spaghetti_map:
+        col1, col2 = st.columns(2)
         
-        except Exception as e:
-            # Último recurso: método directo
-            uploaded_file.seek(0)
-            gpx_content = uploaded_file.read().decode('utf-8')
+        with col1:
+            st.markdown("#### 🔥 Mapa de Calor")
+            folium_static(heat_map, width=600, height=500)
         
-        if gpx_content:
-            st.success("Datos cargados exitosamente desde el archivo proporcionado.")
-            
-            # Crear DataFrame simulado para mantener consistencia con apptest.py
-            data = pd.DataFrame({
-                'status': ['loaded'], 
-                'source': ['uploaded'], 
-                'filename': [uploaded_file.name],
-                'size': [len(gpx_content)]
-            })
-            st.dataframe(data, use_container_width=True)
-        else:
-            st.error("No se pudo leer el contenido del archivo")
+        with col2:
+            st.markdown("#### 🍝 Diagrama de Spaghetti")
+            folium_static(spaghetti_map, width=600, height=500)
             
-    except Exception as e:
-        st.error(f"Error leyendo archivo: {e}")
-        # Debug information
-        st.write("Información de debug:")
-        st.write(f"Tipo de archivo: {type(uploaded_file)}")
-        st.write(f"Nombre de archivo: {uploaded_file.name if uploaded_file else 'None'}")
+        # Info adicional
+        with st.expander("ℹ️ Información técnica"):
+            st.write(f"**Tracks:** {len(set(seg['track_idx'] for seg in track_segments))}")
+            st.write(f"**Segmentos:** {len(track_segments)}")
+            st.write(f"**Centro:** {center[0]:.6f}, {center[1]:.6f}")
+    else:
+        st.error("❌ Error generando los mapas")
+
+elif gpx_data and not gpx_data['success']:
+    st.error(f"❌ Error: {gpx_data['error']}")
 
-# Validación para asegurarse de que se cargaron datos - EXACTO COMO APPTEST.PY
-if data is None:
-    st.warning("No se han cargado datos. Por favor, selecciona un archivo o usa datos de ejemplo.")
 else:
-    st.write("### Datos listos para su análisis.")
+    st.info("📤 Por favor, carga un archivo GPX o selecciona un ejemplo para comenzar.")
     
-    # VERIFICAR QUE TENEMOS CONTENIDO GPX VÁLIDO
-    if gpx_content is None:
-        st.error("Error: No se pudo extraer el contenido GPX del archivo")
-    else:
-        # Validar que es contenido GPX válido
-        if not ("<?xml" in gpx_content and ("gpx" in gpx_content.lower() or "trk" in gpx_content.lower())):
-            st.error("El archivo no parece contener datos GPX válidos")
-            with st.expander("Ver contenido del archivo para debug"):
-                st.text(gpx_content[:500])
-        else:
-            # PROCESAR DATOS GPX
-            with st.spinner("🔄 Procesando datos GPX..."):
-                gpx_data = process_gpx_data(gpx_content)
-            
-            if gpx_data and gpx_data['success']:
-                st.success("🎉 ¡Archivo GPX procesado exitosamente!")
-                
-                # Extraer datos procesados
-                heatmap_data = gpx_data['heatmap_data']
-                track_segments = gpx_data['track_segments']
-                total_distance = gpx_data['total_distance']
-                total_time = gpx_data['total_time']
-                center = gpx_data['center']
-                
-                # Dashboard de métricas
-                st.markdown("### 📊 Resumen de la Ruta")
-                col1, col2, col3 = st.columns(3)
-                
-                with col1:
-                    st.metric("📏 Distancia", f"{total_distance/1000:.2f} km" if total_distance > 0 else "N/A")
-                with col2:
-                    hours = int(total_time // 3600) if total_time > 0 else 0
-                    minutes = int((total_time % 3600) // 60) if total_time > 0 else 0
-                    st.metric("⏱️ Tiempo", f"{hours}h {minutes}m" if total_time > 0 else "N/A")
-                with col3:
-                    st.metric("📍 Puntos GPS", f"{len(heatmap_data):,}")
-                
-                # Crear y mostrar mapas
-                st.markdown("### 🗺️ Visualización de Movimientos")
-                
-                with st.spinner("🗺️ Generando mapas..."):
-                    heat_map, spaghetti_map = create_maps(heatmap_data, track_segments, center)
-                
-                if heat_map and spaghetti_map:
-                    col1, col2 = st.columns(2)
-                    
-                    with col1:
-                        st.markdown("#### 🔥 Mapa de Calor")
-                        folium_static(heat_map, width=600, height=500)
-                    
-                    with col2:
-                        st.markdown("#### 🍝 Diagrama de Spaghetti")
-                        folium_static(spaghetti_map, width=600, height=500)
-                        
-                    # Información técnica adicional
-                    with st.expander("ℹ️ Información técnica"):
-                        st.write(f"**Archivo:** {uploaded_file.name if uploaded_file else archivo_ejemplo_seleccionado}")
-                        st.write(f"**Fuente:** {'Archivo cargado' if uploaded_file else 'Archivo de ejemplo'}")
-                        st.write(f"**Tracks:** {len(set(seg['track_idx'] for seg in track_segments))}")
-                        st.write(f"**Segmentos:** {len(track_segments)}")
-                        st.write(f"**Centro:** {center[0]:.6f}, {center[1]:.6f}")
-                        st.write(f"**Puntos totales:** {len(heatmap_data)}")
-                        st.write(f"**Tamaño del archivo:** {len(gpx_content)} caracteres")
-                        
-                else:
-                    st.error("❌ Error generando los mapas")
-                    
-            elif gpx_data and not gpx_data['success']:
-                st.error(f"❌ Error procesando GPX: {gpx_data['error']}")
-                
-                # Debug completo
-                with st.expander("🔧 Información completa de debugging"):
-                    st.write("**Información del archivo:**")
-                    st.write(f"- Nombre: {uploaded_file.name if uploaded_file else archivo_ejemplo_seleccionado}")
-                    st.write(f"- Tamaño: {len(gpx_content)} caracteres")
-                    st.write(f"- Tipo de fuente: {'Cargado' if uploaded_file else 'Ejemplo'}")
-                    
-                    st.write("**Primeros 1000 caracteres del archivo:**")
-                    st.text(gpx_content[:1000] if gpx_content else "Sin contenido")
-                    
-                    st.write("**Últimos 500 caracteres del archivo:**")
-                    st.text(gpx_content[-500:] if gpx_content and len(gpx_content) > 500 else "Archivo muy corto")
-            else:
-                st.error("❌ Error desconocido procesando el archivo GPX")
-
-# Mostrar ejemplos disponibles si no hay datos cargados
-if data is None and example_files:
-    st.markdown("### 📋 Ejemplos disponibles:")
-    for i, file in enumerate(example_files, 1):
-        st.write(f"{i}. **{file}**")
+    if example_files:
+        st.markdown("### 📋 Ejemplos disponibles:")
+        for i, file in enumerate(example_files, 1):
+            st.write(f"{i}. **{file}**")
 
 # Footer
 st.markdown("---")
-st.markdown("🚀 **Optimizado para Hugging Face Spaces** | 📊 **Compatible con apptest.py**", unsafe_allow_html=True)
+st.markdown("🚀 **Optimizado para Hugging Face Spaces**", unsafe_allow_html=True)