50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
import openai
|
|
import ffmpeg
|
|
import os
|
|
from telegram import ForceReply, Update
|
|
from telegram.ext import Application, CommandHandler, ContextTypes, MessageHandler, filters
|
|
import asyncio
|
|
import textwrap
|
|
import configparser
|
|
from pathlib import Path
|
|
|
|
Path("temp/").mkdir(parents=True, exist_ok=True)
|
|
|
|
config = configparser.ConfigParser()
|
|
config.read('config.ini')
|
|
|
|
openai = OpenAI(api_key=config['DEFAULT']['OpenAIKey']) # initiate with your openai key
|
|
allowed_senders = config['DEFAULT']['AllowedTelegramSenders'].split(',')
|
|
|
|
|
|
async def on_msg(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
|
message = update.message
|
|
sender_id = message.from_user.id
|
|
print(sender_id)
|
|
if sender_id not in allowed_senders:
|
|
return
|
|
|
|
f = await message.effective_attachment.get_file()
|
|
await f.download_to_drive('temp/in')
|
|
|
|
input = ffmpeg.input('temp/in')
|
|
out = ffmpeg.output(input, 'temp/out.mp3')
|
|
ffmpeg.run(out, overwrite_output=True)
|
|
audio_file = open("temp/out.mp3", "rb")
|
|
|
|
transcript = openai.audio.transcriptions.create(model="whisper-1", file=audio_file)
|
|
text = transcript.text
|
|
for t in textwrap.wrap(text, 4000):
|
|
await update.message.reply_text(t)
|
|
|
|
|
|
async def main():
|
|
return
|
|
|
|
|
|
if __name__ == '__main__':
|
|
token = config['DEFAULT']['TelegramToken']
|
|
application = Application.builder().token(token).build()
|
|
|
|
application.add_handler(MessageHandler(filters=None, callback=on_msg))
|
|
application.run_polling()
|