import discord
from discord.ext import commands
import speech_recognition as sr
import asyncio
import os

# Konfiguracja intencji (uprawnień)
intents = discord.Intents.default()
intents.message_content = True
intents.voice_states = True

bot = commands.Bot(command_prefix="!", intents=intents)
r = sr.Recognizer()

# Funkcja callback - uruchamiana po zakończeniu nagrywania
async def finished_callback(sink, channel: discord.TextChannel, *args):
    print("Nagrywanie zakończone. Przetwarzanie audio...")
    
    # sink.audio_data.items() zwraca pary: user_id, AudioData
    for user_id, audio in sink.audio_data.items():
        filename = f"audio_{user_id}.wav"
        
        # Zapisz plik na dysku
        with open(filename, "wb") as f:
            f.write(audio.file.read())  # Używamy .read() dla pewności

        # --- Reszta kodu bez zmian (rozpoznawanie mowy) ---
        try:
            with sr.AudioFile(filename) as source:
                # Załaduj audio do pamięci
                audio_data = r.record(source)
                # Użyj Google Web Speech API (darmowe, ale ma limity)
                text = r.recognize_google(audio_data, language="pl-PL")
                
                user = await bot.fetch_user(user_id)
                output = f"Użytkownik {user.name} powiedział: {text}"
                
                # Print do konsoli (zgodnie z prośbą)
                print(output)
                # Opcjonalnie wyślij na kanał tekstowy:
                # await channel.send(output)

        except sr.UnknownValueError:
            print(f"Nie zrozumiano mowy użytkownika {user_id}.")
        except sr.RequestError as e:
            print(f"Błąd serwisu Google Speech Recognition: {e}")
        except Exception as e:
            print(f"Wystąpił błąd: {e}")
        finally:
            # Posprzątaj pliki tymczasowe
            if os.path.exists(filename):
                os.remove(filename)

@bot.event
async def on_ready():
    print(f'Zalogowano jako {bot.user}')

@bot.command()
async def join(ctx):
    """Bot dołącza do Twojego kanału głosowego"""
    if ctx.author.voice:
        channel = ctx.author.voice.channel
        await channel.connect()
        await ctx.send(f"Dołączono do {channel}")
    else:
        await ctx.send("Musisz być na kanale głosowym!")

@bot.command()
async def start(ctx):
    """Zacznij nagrywanie"""
    if ctx.voice_client:
        print("Rozpoczynam nagrywanie...")
        # WaveSink nagrywa do formatu WAV
        ctx.voice_client.start_recording(
            discord.sinks.WaveSink(),
            finished_callback,
            ctx.channel,
        )
        await ctx.send("Nasłuchuję... wpisz !stop aby transkrybować.")
    else:
        await ctx.send("Bot nie jest połączony z kanałem głosowym.")

@bot.command()
async def stop(ctx):
    """Zatrzymaj nagrywanie i wypisz tekst"""
    if ctx.voice_client:
        print("Zatrzymywanie nagrywania...")
        ctx.voice_client.stop_recording()
        await ctx.send("Zatrzymano nagrywanie. Sprawdź konsolę.")
    else:
        await ctx.send("Bot nic nie nagrywa.")

@bot.command()
async def leave(ctx):
    """Bot wychodzi z kanału"""
    if ctx.voice_client:
        await ctx.voice_client.disconnect()
        await ctx.send("Rozłączono.")
    else:
        await ctx.send("Bot nie jest połączony.")

# Uruchom bota
token = os.getenv("DISCORD_TOKEN")
if not token:
    raise SystemExit("Missing DISCORD_TOKEN environment variable. Set it and retry.")

bot.run(token)