🎤 Medindo o barulho da galera

# -*- coding: utf-8 -*-
# ============================================================
# ROBÔ FOFOQUEIRO - AULA 2: SENSOR DE SOM IRREVERENTE
# "Medindo o barulho da galera: o primeiro olho do robô"
# ============================================================
# Alunos: [Seu nome aqui]
# Tutora: Gisele Nunes
# Data: 2026
# ============================================================

# ===== BIBLIOTECAS NECESSÁRIAS =====
import pyaudio
import numpy as np
import time
import threading
import csv
from datetime import datetime
import matplotlib.pyplot as plt
from IPython.display import clear_output

# ===== CONFIGURAÇÃO INICIAL =====
# No Colab: !pip install pyaudio keyboard matplotlib numpy

# ===== FUNÇÃO PRINCIPAL =====
def medidor_fofoqueiro():
    """
    Função que simula o sensor KY-038 usando o microfone real.
    Retorna:
    tuple: (decibeis_simulados, alerta, mensagem_meme)
    """
    # PARÂMETROS DE ÁUDIO
    DURACAO = 3
    TAXA_AMOSTRAGEM = 44100
    TAMANHO_BUFFER = 1024
    
    audio = pyaudio.PyAudio()
    stream = audio.open(
        format=pyaudio.paInt16,
        channels=1,
        rate=TAXA_AMOSTRAGEM,
        input=True,
        frames_per_buffer=TAMANHO_BUFFER
    )
    
    print("🎤 Gravando 3 segundos de áudio...")
    
    frames = []
    for _ in range(0, int(TAXA_AMOSTRAGEM / TAMANHO_BUFFER * DURACAO)):
        data = stream.read(TAMANHO_BUFFER)
        dados_audio = np.frombuffer(data, dtype=np.int16)
        frames.append(dados_audio)
    
    stream.stop_stream()
    stream.close()
    audio.terminate()
    
    audio_data = np.concatenate(frames)
    
    # CÁLCULO DA INTENSIDADE SONORA
    rms = np.sqrt(np.mean(audio_data**2))
    max_rms = 32768
    decibeis = (rms / max_rms) * 120
    decibeis = min(120, max(0, decibeis))
    
    # EMOJI CONFORME VOLUME
    if decibeis < 20:
        emoji = "🔇🤫"
    elif decibeis < 40:
        emoji = "🤫"
    elif decibeis < 60:
        emoji = "🗣️"
    elif decibeis < 80:
        emoji = "📢"
    elif decibeis < 100:
        emoji = "🔊"
    else:
        emoji = "💀📢"
    
    # MENSAGEM MEME POR NÍVEL
    if decibeis < 40:
        mensagem = "🤫 silêncio absoluto... até assusta"
        alerta = False
    elif decibeis < 60:
        mensagem = "🗣️ papinho normal, nada de mais"
        alerta = False
    elif decibeis < 80:
        mensagem = "📢 TÁ FICANDO BOM! logo logo viro fofoqueiro"
        alerta = True
    elif decibeis < 100:
        mensagem = "🔊 CALA A BOCA! vou piscar o LED vermelho"
        alerta = True
    else:
        mensagem = "💀 INFERNO ACÚSTICO! vou chamar a diretora robô"
        alerta = True
    
    alerta = decibeis > 75
    
    print("\n" + "="*50)
    print(f"Nível atual: {emoji}")
    print(f"Decibéis simulados: {decibeis:.1f} dB")
    print(f"📢 {mensagem}")
    print("="*50)
    
    return (decibeis, alerta, mensagem)

# ===== FUNÇÃO PARA SALVAR LOG EM CSV =====
def salvar_log(decibeis, mensagem, arquivo="log_fofoqueiro.csv"):
    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
    with open(arquivo, mode='a', newline='', encoding='utf-8') as file:
        writer = csv.writer(file)
        file.seek(0, 2)
        if file.tell() == 0:
            writer.writerow(["Timestamp", "Decibeis (dB)", "Mensagem"])
        writer.writerow([timestamp, f"{decibeis:.1f}", mensagem])
    print(f"💾 Log salvo: {timestamp} - {decibeis:.1f} dB")

# ===== FUNÇÃO PARA MOSTRAR GRÁFICO =====
def mostrar_grafico(arquivo="log_fofoqueiro.csv"):
    try:
        with open(arquivo, 'r', encoding='utf-8') as file:
            leitor = csv.reader(file)
            cabecalho = next(leitor)
            timestamps = []
            decibeis = []
            contador = 0
            for linha in leitor:
                timestamps.append(linha[0])
                decibeis.append(float(linha[1]))
                contador += 1
            
            ultimos = min(10, contador)
            timestamps = timestamps[-ultimos:]
            decibeis = decibeis[-ultimos:]
            
            plt.figure(figsize=(12, 6))
            cores = ['green' if db <= 75 else 'red' for db in decibeis]
            barras = plt.bar(range(len(decibeis)), decibeis, color=cores, alpha=0.7)
            plt.axhline(y=75, color='orange', linestyle='--', linewidth=2, label='Limite de Alerta (75 dB)')
            plt.xlabel('Medições (mais recente na direita)', fontsize=12)
            plt.ylabel('Decibéis Simulados (dB)', fontsize=12)
            plt.title('📊 Histórico do Robô Fofoqueiro - Últimas Medições', fontsize=14, fontweight='bold')
            plt.xticks(range(len(timestamps)), timestamps, rotation=45, ha='right')
            plt.ylim(0, 120)
            plt.grid(True, alpha=0.3)
            plt.legend()
            
            for i, (barra, db) in enumerate(zip(barras, decibeis)):
                altura = barra.get_height()
                plt.text(barra.get_x() + barra.get_width()/2., altura + 2, f'{db:.0f}dB', ha='center', va='bottom', fontsize=9)
            
            plt.tight_layout()
            plt.show()
            print(f"📈 Gráfico mostrado com os últimos {ultimos} registros!")
    except FileNotFoundError:
        print("⚠️ Ainda não há dados para mostrar. Faça algumas medições primeiro!")
    except Exception as e:
        print(f"❌ Erro ao gerar gráfico: {e}")

# ===== LOOP PRINCIPAL =====
def main():
    print("""
╔══════════════════════════════════════════════════════════╗
║                                                          ║
║ 🤖 ROBÔ FOFOQUEIRO - VERSÃO PARANÁENSE 🤖                ║
║                                                          ║
║ "Bora medir o barulho da galera!"                        ║
║ "Égua, que gritaria danada!"                             ║
║                                                          ║
║ Pressione 'q' a qualquer momento para sair              ║
║                                                          ║
╚══════════════════════════════════════════════════════════╝
    """)
    
    contador_medicoes = 0
    historico_dB = []
    
    try:
        while True:
            contador_medicoes += 1
            print(f"\n🔁 Medição #{contador_medicoes}")
            
            dB, alerta, mensagem = medidor_fofoqueiro()
            salvar_log(dB, mensagem)
            historico_dB.append(dB)
            
            if alerta:
                print("🚨 ALERTA! 🚨")
                print("🔴🔴🔴 PASSOU DOS 75 dB! TÁ MUITO ALTO! 🔴🔴🔴")
            
            comando = input("\nPressione ENTER para nova medição ou 'q' para sair: ")
            if comando.lower() == 'q':
                break
    except KeyboardInterrupt:
        print("\n\n🛑 Programa interrompido pelo usuário!")
    
    print("\n" + "="*50)
    print("📊 RESUMO DO ROBÔ FOFOQUEIRO")
    print("="*50)
    print(f"Total de medições realizadas: {contador_medicoes}")
    if historico_dB:
        print(f"Maior nível registrado: {max(historico_dB):.1f} dB")
        print(f"Menor nível registrado: {min(historico_dB):.1f} dB")
        print(f"Média dos níveis: {sum(historico_dB)/len(historico_dB):.1f} dB")
    
    print("\n📊 Gerando gráfico final com todo o histórico...")
    mostrar_grafico()
    print("\n✅ Programa encerrado com sucesso!")
    print("🙏 Valeu por usar o Robô Fofoqueiro - Tchê!")

if __name__ == "__main__":
    print("⚠️ IMPORTANTE: Execute no terminal do Colab:")
    print("!pip install pyaudio keyboard matplotlib numpy")
    print("!apt-get install portaudio19-dev")
    input("\nPressione ENTER após instalar as bibliotecas...")
    main()