Update app.py

app.py

@@ -1,14 +1,9 @@
 import gradio as gr
 import sqlite3
-import json
 import pandas as pd
-from datetime import datetime, timedelta
+from datetime import datetime
 import folium
 from folium import plugins
-import requests
-import threading
-import time
-import os
 
 # Configuração do banco de dados
 def init_db():
@@ -67,7 +62,6 @@ def get_updated_map():
     try:
         conn = sqlite3.connect('locations.db', check_same_thread=False)
         
-        # Buscar localizações mais recentes de cada dispositivo
         query = '''
             SELECT device_id, device_name, latitude, longitude, 
                    timestamp, accuracy, battery_level,
@@ -77,12 +71,11 @@ def get_updated_map():
         '''
         
         df = pd.read_sql_query(query, conn)
-        df_latest = df[df['rn'] == 1].copy()  # Apenas a localização mais recente de cada dispositivo
+        df_latest = df[df['rn'] == 1].copy()
         
         conn.close()
         
         if df_latest.empty:
-            # Mapa padrão centrado no Brasil
             m = folium.Map(location=[-14.235, -51.9253], zoom_start=4)
             folium.Marker(
                 [-14.235, -51.9253],
@@ -91,25 +84,20 @@ def get_updated_map():
             ).add_to(m)
             return m._repr_html_()
         
-        # Criar mapa centrado na média das localizações
         center_lat = df_latest['latitude'].mean()
         center_lng = df_latest['longitude'].mean()
         m = folium.Map(location=[center_lat, center_lng], zoom_start=12)
         
-        # Adicionar marcadores para cada dispositivo
         colors = ['red', 'blue', 'green', 'purple', 'orange', 'darkred', 'lightred', 
                  'beige', 'darkblue', 'darkgreen', 'cadetblue', 'darkpurple', 'white', 'pink', 'lightblue', 'lightgreen', 'gray', 'black', 'lightgray']
         
         for idx, row in df_latest.iterrows():
-            # Determinar status (online se última atualização foi há menos de 5 min)
             last_update = datetime.strptime(row['timestamp'], '%Y-%m-%d %H:%M:%S')
-            is_online = (datetime.now() - last_update).total_seconds() < 300  # 5 minutos
+            is_online = (datetime.now() - last_update).total_seconds() < 300
             
-            # Escolher cor do marcador
             color = colors[idx % len(colors)]
             icon_color = 'green' if is_online else 'red'
             
-            # Informações do popup
             popup_html = f"""
             <div style="font-family: Arial; width: 200px;">
                 <h4>📱 {row['device_name']}</h4>
@@ -130,7 +118,6 @@ def get_updated_map():
                 icon=folium.Icon(color=icon_color, icon='phone', prefix='fa')
             ).add_to(m)
             
-            # Adicionar círculo de precisão
             folium.Circle(
                 [row['latitude'], row['longitude']],
                 radius=row['accuracy'],
@@ -141,7 +128,6 @@ def get_updated_map():
                 weight=1
             ).add_to(m)
         
-        # Adicionar histórico de movimento (últimas 24h)
         for device_id in df_latest['device_id'].unique():
             device_history = df[df['device_id'] == device_id].sort_values('timestamp')
             if len(device_history) > 1:
@@ -154,10 +140,7 @@ def get_updated_map():
                     popup=f"Trajeto: {device_history.iloc[0]['device_name']}"
                 ).add_to(m)
         
-        # Adicionar plugin de tela cheia
         plugins.Fullscreen().add_to(m)
-        
-        # Adicionar mini mapa
         minimap = plugins.MiniMap()
         m.add_child(minimap)
         
@@ -170,24 +153,20 @@ def get_updated_map():
 def get_statistics():
     try:
         conn = sqlite3.connect('locations.db', check_same_thread=False)
-        
-        # Contar dispositivos únicos
         cursor = conn.cursor()
+        
         cursor.execute("SELECT COUNT(DISTINCT device_id) FROM locations")
         total_devices = cursor.fetchone()[0]
         
-        # Contar localizações nas últimas 24h
         cursor.execute("SELECT COUNT(*) FROM locations WHERE timestamp > datetime('now', '-1 day')")
         locations_24h = cursor.fetchone()[0]
         
-        # Dispositivos online (última atualização < 5 min)
         cursor.execute("""
             SELECT COUNT(DISTINCT device_id) FROM locations 
             WHERE timestamp > datetime('now', '-5 minutes')
         """)
         online_devices = cursor.fetchone()[0]
         
-        # Última atualização geral
         cursor.execute("SELECT MAX(timestamp) FROM locations")
         last_update = cursor.fetchone()[0]
         
@@ -212,7 +191,7 @@ def get_statistics():
     except Exception as e:
         return f"❌ Erro ao carregar estatísticas: {str(e)}"
 
-# Função para obter lista de dispositivos
+# Função para obter lista de dispositivos (modificada para sempre retornar DataFrame)
 def get_device_list():
     try:
         conn = sqlite3.connect('locations.db', check_same_thread=False)
@@ -235,29 +214,35 @@ def get_device_list():
         conn.close()
         
         if df.empty:
-            return "Nenhum dispositivo encontrado nos últimos 7 dias."
+            return pd.DataFrame([{
+                "Nome": "Nenhum dispositivo encontrado",
+                "ID": "",
+                "Status": "",
+                "Coordenadas": "",
+                "Última Atualização": "",
+                "Bateria (%)": "",
+                "Precisão (m)": ""
+            }])
         
-        # Filtrar apenas a localização mais recente de cada dispositivo
         df_latest = df[df['rn'] == 1].copy()
         
-        # Adicionar coluna de status
         df_latest['status'] = df_latest['timestamp'].apply(
             lambda x: '🟢 Online' if (datetime.now() - datetime.strptime(x, '%Y-%m-%d %H:%M:%S')).total_seconds() < 300 else '🔴 Offline'
         )
         
-        # Formatar coordenadas
         df_latest['coordenadas'] = df_latest.apply(
             lambda row: f"{row['latitude']:.6f}, {row['longitude']:.6f}", axis=1
         )
         
-        # Selecionar colunas para exibição
         display_df = df_latest[['device_name', 'device_id', 'status', 'coordenadas', 'timestamp', 'battery_level', 'accuracy']].copy()
         display_df.columns = ['Nome', 'ID', 'Status', 'Coordenadas', 'Última Atualização', 'Bateria (%)', 'Precisão (m)']
         
         return display_df
         
     except Exception as e:
-        return f"Erro ao carregar lista de dispositivos: {str(e)}"
+        return pd.DataFrame([{
+            "Erro": f"Erro ao carregar lista de dispositivos: {str(e)}"
+        }])
 
 # Função para limpeza de dados antigos
 def cleanup_old_data():
@@ -265,7 +250,6 @@ def cleanup_old_data():
         conn = sqlite3.connect('locations.db', check_same_thread=False)
         cursor = conn.cursor()
         
-        # Remover dados com mais de 30 dias
         cursor.execute("DELETE FROM locations WHERE timestamp < datetime('now', '-30 days')")
         deleted = cursor.rowcount
         
@@ -287,7 +271,6 @@ def export_data():
         if df.empty:
             return None, "Nenhum dado para exportar."
         
-        # Salvar como CSV
         filename = f"localizacoes_export_{datetime.now().strftime('%Y%m%d_%H%M%S')}.csv"
         df.to_csv(filename, index=False, encoding='utf-8')
         
@@ -310,56 +293,25 @@ def create_interface():
         """)
         
         with gr.Tabs():
-            # Aba 1: Adicionar Localização
             with gr.TabItem("📱 Enviar Localização"):
                 gr.Markdown("### 📡 Enviar Nova Localização")
                 
                 with gr.Row():
-                    device_id = gr.Textbox(
-                        label="🆔 ID do Dispositivo",
-                        placeholder="Ex: celular_joao",
-                        info="Identificador único do dispositivo"
-                    )
-                    device_name = gr.Textbox(
-                        label="📱 Nome do Dispositivo",
-                        placeholder="Ex: Celular do João",
-                        info="Nome amigável para identificação"
-                    )
+                    device_id = gr.Textbox(label="🆔 ID do Dispositivo", placeholder="Ex: celular_joao")
+                    device_name = gr.Textbox(label="📱 Nome do Dispositivo", placeholder="Ex: Celular do João")
                 
                 with gr.Row():
-                    latitude = gr.Textbox(
-                        label="🌐 Latitude",
-                        placeholder="Ex: -23.550520",
-                        info="Coordenada de latitude (GPS)"
-                    )
-                    longitude = gr.Textbox(
-                        label="🌐 Longitude", 
-                        placeholder="Ex: -46.633309",
-                        info="Coordenada de longitude (GPS)"
-                    )
+                    latitude = gr.Textbox(label="🌐 Latitude", placeholder="Ex: -23.550520")
+                    longitude = gr.Textbox(label="🌐 Longitude", placeholder="Ex: -46.633309")
                 
                 with gr.Row():
-                    accuracy = gr.Number(
-                        label="🎯 Precisão (metros)",
-                        value=10,
-                        info="Precisão da localização em metros"
-                    )
-                    battery = gr.Slider(
-                        label="🔋 Nível da Bateria (%)",
-                        minimum=0,
-                        maximum=100,
-                        value=100,
-                        step=1
-                    )
+                    accuracy = gr.Number(label="🎯 Precisão (metros)", value=10)
+                    battery = gr.Slider(label="🔋 Nível da Bateria (%)", minimum=0, maximum=100, value=100, step=1)
                 
                 send_btn = gr.Button("📡 Enviar Localização", variant="primary", size="lg")
                 send_result = gr.Textbox(label="Resultado", interactive=False)
                 
-                send_btn.click(
-                    add_location,
-                    inputs=[device_id, device_name, latitude, longitude, accuracy, battery],
-                    outputs=send_result
-                )
+                send_btn.click(add_location, inputs=[device_id, device_name, latitude, longitude, accuracy, battery], outputs=send_result)
                 
                 gr.Markdown("""
                 **💡 Dica:** Para obter suas coordenadas atuais:
@@ -368,82 +320,36 @@ def create_interface():
                 3. Copie as coordenadas que aparecem
                 """)
             
-            # Aba 2: Monitoramento
             with gr.TabItem("🗺️ Mapa em Tempo Real"):
                 gr.Markdown("### 🌍 Visualização em Tempo Real")
-                
-                with gr.Row():
-                    refresh_btn = gr.Button("🔄 Atualizar Mapa", variant="secondary")
-                    auto_refresh = gr.Checkbox(label="🔄 Atualização Automática (30s)", value=True)
-                
-                map_html = gr.HTML(value=get_updated_map(), label="Mapa")
-                
-                refresh_btn.click(
-                    get_updated_map,
-                    outputs=map_html
-                )
-                
-                # Auto-refresh do mapa
+                refresh_btn = gr.Button("🔄 Atualizar Mapa", variant="secondary")
+                map_html = gr.HTML(label="Mapa")
+                refresh_btn.click(get_updated_map, outputs=map_html)
                 gr.Timer(30.0, get_updated_map, [], [map_html])
             
-            # Aba 3: Lista de Dispositivos
             with gr.TabItem("📋 Lista de Dispositivos"):
                 gr.Markdown("### 📱 Dispositivos Monitorados")
-                
                 refresh_devices_btn = gr.Button("🔄 Atualizar Lista", variant="secondary")
-                devices_table = gr.Dataframe(
-                    value=get_device_list(),
-                    label="Dispositivos Ativos",
-                    interactive=False
-                )
-                
-                refresh_devices_btn.click(
-                    get_device_list,
-                    outputs=devices_table
-                )
-                
-                # Auto-refresh da tabela
+                devices_table = gr.Dataframe(label="Dispositivos Ativos", interactive=False)
+                refresh_devices_btn.click(get_device_list, outputs=devices_table)
                 gr.Timer(30.0, get_device_list, [], [devices_table])
             
-            # Aba 4: Estatísticas
             with gr.TabItem("📊 Estatísticas"):
                 gr.Markdown("### 📈 Estatísticas do Sistema")
-                
                 refresh_stats_btn = gr.Button("🔄 Atualizar Estatísticas", variant="secondary")
-                stats_display = gr.Markdown(value=get_statistics())
-                
-                refresh_stats_btn.click(
-                    get_statistics,
-                    outputs=stats_display
-                )
-                
-                # Auto-refresh das estatísticas
+                stats_display = gr.Markdown()
+                refresh_stats_btn.click(get_statistics, outputs=stats_display)
                 gr.Timer(60.0, get_statistics, [], [stats_display])
             
-            # Aba 5: Administração
             with gr.TabItem("⚙️ Administração"):
                 gr.Markdown("### 🛠️ Ferramentas de Administração")
-                
                 with gr.Row():
                     cleanup_btn = gr.Button("🧹 Limpar Dados Antigos", variant="secondary")
                     export_btn = gr.Button("📥 Exportar Dados", variant="secondary")
-                
                 admin_result = gr.Textbox(label="Resultado da Operação", interactive=False)
                 export_file = gr.File(label="Arquivo Exportado", visible=False)
-                
-                cleanup_btn.click(
-                    cleanup_old_data,
-                    outputs=admin_result
-                )
-                
-                def handle_export():
-                    file_path, message = export_data()
-                    return message, file_path
-                
-                export_btn.click(
-                    handle_export,
-                    outputs=[admin_result, export_file]
-                )
+                cleanup_btn.click(cleanup_old_data, outputs=admin_result)
+                export_btn.click(export_data, outputs=[admin_result, export_file])
                 
                 gr.Markdown("""
                 **⚠️ Informações Importantes:**
@@ -456,7 +362,6 @@ def create_interface():
         gr.Markdown("""
         ---
         ### 📱 Como Usar no Celular:
-        
         1. **Para Enviar Localização:** Use a aba "Enviar Localização"
         2. **Para Monitorar:** Use a aba "Mapa em Tempo Real"  
         3. **Atualizações:** O sistema atualiza automaticamente