Update app.py

app.py

@@ -103,6 +103,39 @@ def search_direct(query, data, notif_map):
     return final_results
 
 
+
+
+
+def log_query(query, result_event):
+    """Registra una consulta y su resultado en un archivo CSV."""
+    log_file = 'query_log.csv'
+    # La fecha y hora actual en formato legible
+    timestamp = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    
+    # Prepara la fila a escribir
+    # Usamos comillas dobles y comas para el formato CSV
+    log_entry = f'"{timestamp}","{query.replace("\"", "\"\"")}","{result_event.replace("\"", "\"\"")}"\n'
+    
+    try:
+        # Abre el archivo en modo 'append' (añadir al final)
+        # Si el archivo no existe, lo crea.
+        with open(log_file, 'a', newline='', encoding='utf-8') as f:
+            # Escribe la cabecera si el archivo está vacío
+            if f.tell() == 0:
+                f.write('"Timestamp","UserQuery","AI_Result"\n')
+            f.write(log_entry)
+        logger.info(f"Consulta registrada: {query}")
+    except Exception as e:
+        logger.error(f"Error al registrar la consulta en CSV: {e}")
+
+
+
+
+
+
+
+
+
 def search_with_gemini(query, event_list):
     if not model: return "Error: Modelo Gemini no disponible."
     system_prompt = f"""
@@ -230,6 +263,22 @@ if submitted:
             st.session_state.search_results = results
             st.session_state.last_query = query
 
+
+
+            
+            # Si se encontró al menos un resultado, lo registramos.
+            if results:
+                # Tomamos el nombre del evento del primer resultado encontrado.
+                evento_encontrado = results[0].get("info_notificacion", {}).get("Evento", "Búsqueda Directa Exitosa")
+                log_query(query, evento_encontrado)
+            else:
+                # Opcional: registrar también las búsquedas sin resultados.
+                log_query(query, "NO_ENCONTRADO")
+
+
+
+            
+
 if st.session_state.search_results is not None: st.button("Limpiar Búsqueda", on_click=clear_search_state)
 st.markdown("---")