import asyncio
import numpy as np
import whisper
import tempfile
import os
import scipy.io.wavfile
import soundcard as sc
import torch

import telebot

bot = telebot.TeleBot(token=YOUR_BOT_TOKEN)

chan_dict = {
    , 'qc': TG_CHANNEL_ID}

# Initialize the Whisper model (you can use "base", "small", etc.)
device = "cuda" if torch.cuda.is_available() else "cpu"
model_size = "small"
model = whisper.load_model(model_size).to(device)

# Audio settings
SAMPLE_RATE = 16000  # Whisper expects 16kHz audio
NUM_FRAMES = SAMPLE_RATE * 30

# get the current default microphone on your system:
mics = sc.all_microphones(include_loopback=True)
default_mic = mics[0]
default_speaker = sc.default_speaker()

async def record_audio(numframes=NUM_FRAMES, samplerate=SAMPLE_RATE):
    print(f"[INFO] Recording {NUM_FRAMES/SAMPLE_RATE}s of audio...")
    with default_mic.recorder(samplerate=samplerate) as mic, default_speaker.player(samplerate=samplerate) as sp:
        print("Recording...")
        audio = mic.record(numframes=numframes)
        print("Recording complete.")
    return np.squeeze(audio)

async def transcribe(audio_data, samplerate=SAMPLE_RATE):
    with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as f:
        temp_filename = f.name
        scipy.io.wavfile.write(temp_filename, samplerate, audio_data)
    
    print(f"[INFO] Transcribing audio segment...")
    result = model.transcribe(temp_filename, language="ru")
    os.remove(temp_filename)
    return result["text"]

async def main_loop():
    print("[INFO] Starting asynchronous transcription loop. Press Ctrl+C to stop.")
    try:
        while True:
            audio_data = await record_audio()
            text = await transcribe(audio_data)
            bot.send_message(chan_dict['qc']
              , text
              , parse_mode='MARKDOWN'
              , disable_web_page_preview=True)
            print(f"[TRANSCRIPTION] {text}\n")
    except KeyboardInterrupt:
        print("\n[INFO] Stopped by user.")

if __name__ == "__main__":
    asyncio.run(main_loop())