107 lines
3.4 KiB
Python
107 lines
3.4 KiB
Python
import logging
|
|
|
|
from telegram import ForceReply, Update, InputFile
|
|
from telegram.ext import Application, CommandHandler, ContextTypes, MessageHandler, filters
|
|
|
|
from pydub import AudioSegment
|
|
import io
|
|
from io import BytesIO
|
|
|
|
from voice_manager import VoiceManager
|
|
from photo_manager import PhotoManager
|
|
|
|
|
|
# Enable logging
|
|
logging.basicConfig(
|
|
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s", level=logging.INFO
|
|
)
|
|
# set higher logging level for httpx to avoid all GET and POST requests being logged
|
|
logging.getLogger("httpx").setLevel(logging.WARNING)
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
# Define a few command handlers. These usually take the two arguments update and
|
|
# context.
|
|
async def start(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
|
"""Send a message when the command /start is issued."""
|
|
user = update.effective_user
|
|
await update.message.reply_html(
|
|
rf"Hi {user.mention_html()}!",
|
|
reply_markup=ForceReply(selective=True),
|
|
)
|
|
|
|
|
|
async def help_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
|
"""Send a message when the command /help is issued."""
|
|
await update.message.reply_text("Help!")
|
|
|
|
|
|
async def echo(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
|
print("Ricevuto messaggio testuale")
|
|
"""Echo the user message."""
|
|
await update.message.reply_text(update.message.text)
|
|
|
|
|
|
async def voice_oracle(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
|
audio_file = await update.message.voice.get_file()
|
|
audio_bytes = await audio_file.download_as_bytearray()
|
|
|
|
await context.bot.send_voice(update.message.chat_id, BytesIO(voice_oracle.vocal_manager.oracle(audio_bytes)))
|
|
|
|
voice_oracle.vocal_manager = VoiceManager()
|
|
|
|
async def get_photo(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
|
print(get_photo.photo_mgr.get_random_photo())
|
|
await update.message.reply_photo(open(get_photo.photo_mgr.get_random_photo(), 'rb'))
|
|
|
|
get_photo.photo_mgr = PhotoManager()
|
|
|
|
|
|
|
|
async def replace_faces(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
|
|
|
|
# user = update.message.from_user
|
|
|
|
photo_file = await update.message.photo[-1].get_file()
|
|
|
|
photo_bytes = await photo_file.download_as_bytearray()
|
|
|
|
out_img_bytes = replace_faces.photo_mgr.replace_faces(photo_bytes)
|
|
await update.message.reply_photo(InputFile(io.BytesIO(out_img_bytes)))
|
|
|
|
# await photo_file.download_to_drive("user_photo.jpg")
|
|
|
|
# logger.info("Photo of %s: %s", user.first_name, "user_photo.jpg")
|
|
|
|
replace_faces.photo_mgr = PhotoManager()
|
|
|
|
|
|
|
|
def main() -> None:
|
|
|
|
"""Start the bot."""
|
|
# Create the Application and pass it your bot's token.
|
|
application = Application.builder().token("6583917096:AAG2SifgcHoQEqzprVp57ZVbxVgwc8Mygpo").build()
|
|
|
|
# on different commands - answer in Telegram
|
|
application.add_handler(CommandHandler("start", start))
|
|
application.add_handler(CommandHandler("help", help_command))
|
|
application.add_handler(CommandHandler("photo", get_photo))
|
|
|
|
|
|
# on non command i.e message - echo the message on Telegram
|
|
application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, echo))
|
|
|
|
application.add_handler(MessageHandler(filters.VOICE, voice_oracle))
|
|
|
|
application.add_handler(MessageHandler(filters.PHOTO, replace_faces))
|
|
|
|
|
|
|
|
# Run the bot until the user presses Ctrl-C
|
|
application.run_polling(allowed_updates=Update.ALL_TYPES)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main() |