telegram_aibot.py

import logging
from telegram import InlineKeyboardButton, InlineKeyboardMarkup
from telegram.ext import Updater, CommandHandler, MessageHandler, Filters, CallbackQueryHandler
from aibot import BOT
#import ftransc.core as ft
import time

type_dict = {"-1": "پرسش خارج از توان",
             "1": "آب و هوا",
             "2": "اوقات شرعی",
             "3": "ساعت",
             "4": "تقویم"}

# build the bot
bot = BOT()

# Enable logging
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
                    level=logging.INFO)

logger = logging.getLogger(__name__)


# Define a few command handlers. These usually take the two arguments update and
# context. Error handlers also receive the raised TelegramError object in error.


def start(update, context):
    """Send a message when the command /start is issued."""
    update.message.reply_text(""" سلام. ممنون از شما که وقت میزارید و در تست این ربات به ما کمک می‌کنید!
     سوال های شما باید از چهار نوع
     آب و هوا
     اوقات شرعی
     ساعت
     تقویم
     باشند. اگر سوال مربوط به این ۴ نوع نبود ربات عبارت <<خارج از توان>> را برمیگردونه. لطفا در صورت هرگونه مشکل به آیدی زیر پیام دهید.
     @dpooria75
     یا از طریق <<گزارش خطا>> گزارش دهید.
        """)


def report(update, context):
    """Send a message when the command /report is issued."""
    with open("report/reports.txt", "a") as freport:
        print(update.message.text, file=freport)
    update.message.reply_text('ممنون. گزارش شما ثبت شد')


st_res_show = "نوع سوال: {} \n نام شهرها: {} \n تاریخ: {} \n زمان: {} \n اوقات شرعی: {} \n نوع تقویم: {} \n مناسبت‌ها: {} \n جواب شما: {} \n لطفا در زیر درست یا غلط بودن پاسخ را مشخص کنید"


def echo(update, context):
    """on text message"""
    res, generated_sentence = bot.aibot(update.message.text)
    res_str = st_res_show.format(type_dict[res["type"][0]], res["city"], res["date"],
                                 res["time"], res["religious_time"], res["calendar_type"], res["event"], res["result"])

    keyboard = [[
        InlineKeyboardButton("👍", callback_data='positive'),
        InlineKeyboardButton("👎", callback_data='negative'),
    ]]
    reply_markup = InlineKeyboardMarkup(keyboard)

    update.message.reply_text(res_str, reply_markup=reply_markup)
    update.message.reply_text(generated_sentence)
    file_name = "userID{}messageID{}".format(
        update.message.chat.username, update.message.message_id)
    with open("collect/{}.txt".format(file_name), "w") as f_res:
        print(update.message.text, file=f_res)
        print("\n", file=f_res)
        print(res_str, file=f_res)
        print("\n", file=f_res)
        print(res, file=f_res)
        print("\n\n\n", file=f_res)
    # question = cleaning(update.message.text)
    # t, l = nerQuestion(bot.ner_model, bot.ner_tokenizer,
    #                    bot.ner_config, question)
    # pd.DataFrame({"tokens": t, "labels": l}).to_csv(
    #     "ner_labelling/" + file_name + ".csv")


# def transcribe_voice(update, context):
#     voice = context.bot.getFile(update.message.voice.file_id)

#     file_name = "userID{}messageID{}".format(
#         update.message.chat.username, update.message.message_id)
#     voice.download('voice/temp.ogg')
#     time.sleep(1)
#     ft.transcode("voice/temp.ogg", 'wav', output_folder="voice")  # file.wav

#     res, response, question, generated_sentence = bot.aibot_voice(
#         "voice/temp.wav")
#     res_str = st_res_show.format(type_dict[res["type"][0]], res["city"], res["date"],
#                                  res["time"], res["religious_time"], res["calendar_type"], res["event"], res["result"])

#     keyboard = [[
#         InlineKeyboardButton("👍", callback_data='positive'),
#         InlineKeyboardButton("👎", callback_data='negative'),
#     ]]
#     reply_markup = InlineKeyboardMarkup(keyboard)

#     update.message.reply_text(res_str, reply_markup=reply_markup)
#     update.message.reply_text(
#         "پرسش شما:\n{}\nپاسخ تولید شده:\n{}".format(question, generated_sentence))

#     with open("voice/temp.wav".format(file_name), mode='wb') as f:
#         f.write(response.content)

#     # ft.transcode("voice/{}res.wav".format(file_name),
#     #              "ogg", output_folder="voice")

#     ft.transcode("voice/temp.wav".format(file_name),
#                  "ogg", output_folder="voice")

#     # context.bot.send_audio(update.message.chat_id, audio=open(
#     #     "voice/{}res.ogg".format(file_name), "rb"))

#     context.bot.send_audio(update.message.chat_id, audio=open(
#         "voice/temp.ogg", "rb"))

#     with open("collect/{}.txt".format(file_name), "w") as f_res:
#         print(question, file=f_res)
#         print("\n", file=f_res)
#         print(res_str, file=f_res)
#         print("\n", file=f_res)
#         print(res, file=f_res)
#         print("\n\n\n", file=f_res)


def help(update, context):
    """on help"""
    helpText = """سوال‌های شما باید به صورت جمله کامل و در مورد یکی از موضوع‌های آب و هوا، تقویم، ساعت و اوقات شرعی باشند؛
        نمونه‌ای از سوالات:
        *آب و هوا
        -هوای تهران در اول بهمن ماه ساعت ۸ شب چند درجه است
        -دمای هوای اصفهان الان چقدر است؟
        -وضعیت هوای مشهد فردا ساعت ۱۱ چطور است؟
        -امروز تبریز در چه ساعتی سردتر است
        -اختلاف دمای تهران و اصفهان در موقع اذان ظهر چه قدر است؟
        *تقویم
        -امروز چه مناسبتی وجود دارد
        -چهارشنبه هفته بعد چندم است
        -روز ۱۲-۱۰-۱۳۹۹ چند شنبه بود
        -عاشورای حسینی سال بعد چندم است
        *ساعت
        -الان ساعت در برلین چند است
        -تا ساعت ۵ بعد از ظهر فردا چند ساعت مانده است
        -اختلاف ساعت مسکو و تهران چند ساعت است
        *اوقات شرعی
        -اذان صبح به افق تهران چه موقع است
        -زمان طلوع آفتاب در اصفهان را بگو
        -نیمه شب شرعی پس فردا شب در مشهد چه زمانی است
        -فاصله بین اذان مغرب و غروب آفتاب امروز در تبریز چقدر است
        ***همچنین سوالات شما می‌تواند به صورت صوتی باشد***
        """
    update.message.reply_text(helpText)


def button(update, context):
    query = update.callback_query

    # CallbackQueries need to be answered, even if no notification to the user is needed
    # Some clients may have trouble otherwise. See https://core.telegram.org/bots/api#callbackquery
    query.answer()

    if not query.data == "Null":
        with open("collect2/userID{}FeedBack{}.txt".format(query.id, query.data), "a") as ffeed:
            print(query.message, end="\n\n", file=ffeed)
        keyboard = [[InlineKeyboardButton(
            'ممنون از شما!', callback_data='Null')]]
        reply_markup = InlineKeyboardMarkup(keyboard)
        query.message.edit_reply_markup(reply_markup=reply_markup)


def error(update, context):
    """Log Errors caused by Updates."""
    with open("errorlog.txt", "a") as ferr:
        logger.warning('Update "%s" caused error "%s"', update, context.error)
        print('Update "%s" caused error "%s"' %
              (update, context.error), file=ferr)


def main():
    """Start the bot."""
    
    REQUEST_KWARGS = {'proxy_url': 'http://127.0.0.1:8118'}
    updater = Updater(
        "****", use_context=True, request_kwargs=REQUEST_KWARGS)

    # Get the dispatcher to register handlers
    dp = updater.dispatcher
    # on different commands - answer in Telegram
    dp.add_handler(CommandHandler("start", start))
    dp.add_handler(CommandHandler("report", report))
    dp.add_handler(CommandHandler("help", help))
    dp.add_handler(CommandHandler("question", echo))
    dp.add_handler(CallbackQueryHandler(button))

    # on noncommand i.e message - echo the message on Telegram
    dp.add_handler(MessageHandler(Filters.text, echo))
#     dp.add_handler(MessageHandler(Filters.voice, transcribe_voice))

    # log all errors
    dp.add_error_handler(error)

    # Start the Bot
    updater.start_polling()
    updater.idle()


if __name__ == '__main__':
    main()