Update pages/telegram.py

pages/telegram.py

@@ -1,120 +1,99 @@
 # import telebot
 from telebot import types
 import asyncio
-from pages.bot import get_llm_answer
-
 from telebot.async_telebot import AsyncTeleBot
+from Warning import *
+from pages.bot import get_llm_answer
 
-# from Warning import *
-
-
-#### Julian neu #####
-
-with open("token.txt") as file:
-    token = file.read()
-bot = AsyncTeleBot(token)
-
-@bot.message_handler(commands=['start'])
-async def handle_start(message):
-    #print(message)
-    button_akt_Meldung = types.InlineKeyboardButton('Aktuelle Meldungen zum Wetter', callback_data='akt_Meldung')
-    button_allg = types.InlineKeyboardButton('Allgemeine Informationen zum Katastrophenschutz', callback_data='allg')
-
-    keyboard = types.InlineKeyboardMarkup()
-    keyboard.add(button_allg)
-
-    await bot.reply_to(message, text='Keyboard example', reply_markup=keyboard)
-
-@bot.callback_query_handler(func=lambda call: True)
-async def handle_button_click(call):
-    match call.data:
-        case "allg":
-            # Hier wird die Funktion aufgerufen, die du mit Button 1 verknüpfen möchtest
-            await function_allg(call.message)
-        case "akt_Meldung":
-            # Hier wird die Funktion aufgerufen, die du mit Button 1 verknüpfen möchtest
-            await function_akt_meldungen(call.message)
-
-        case _:
-            # Hier wird die Funktion aufgerufen, die du mit Button 2 verknüpfen möchtest
-            await bot.send_message(call.message.chat.id, "Der Button ist nicht angebunden.")
-
-
-async def function_allg(message):
-    # hier bitte den Zaubereipart einsenden
-    antwort = get_llm_answer(message.text)
-    await bot.send_message(message, antwort)
-
-async def function_akt_meldungen(message):
-    # hier bitte Pauls abfrage einfügen
-    antwort = "Du möchtest Wetter Informationen"
-    await bot.send_message(message, antwort)
-
-# Handle '/start' and '/help'
-# @bot.message_handler(commands=['help', 'start'])
-# async def send_welcome(message):
-#     await bot.reply_to(message, """hhjvkhv\
-#             Hi there, I am EchoBot.
-#             I am here to echo your kind words back to you. Just say anything nice and I'll say the exact same thing to you!\
-#             """)
+class Telegram: 
+    def __init__(self):
+        self.last_message = ""
+        self.gesuchte_zeile = Warning()    
 
-@bot.message_handler(func=lambda message: True)
-async def get_Message(message):
-    frage = message.text
+        with open("token.txt") as file:
+            token = file.read()
     
-    await bot.reply_to(message, frage)
-
-
-
-
-asyncio.run(bot.polling())
-
-
-
-
-
-
-# ### Paul alt ####
-# with open("token.txt") as file:
-#     token = file.read()
-
-# bot = telebot.TeleBot(token)
-# frage = ""
-
-# # Startnachricht 
-# startnachricht = "Disclaimer zu unserem Chatbot: ..."
-# markup = types.InlineKeyboardMarkup()
-# button = types.InlineKeyboardButton('Aktuelle Informationen zu Warnungen', callback_data='api_warnung')
-# button = types.InlineKeyboardButton('Allgemeine Informationen zu Katastrophen', callback_data='allg_infos')
-# markup.add(button)
-
-# bot.send_message(chat_id='6475480143',text=startnachricht, reply_markup=markup)
-
-# # Command Handler
-# @bot.message_handler(commands=['start'])
-# def start(message):
-#     markup = types.InlineKeyboardMarkup()
-#     button = types.InlineKeyboardButton('Aktuelle Informationen zu Warnungen', callback_data='api_warnung')
-
-#     markup.add(button)
-
-#     bot.send_message(message.chat.id, 'Hallo! Klicke auf den Button:', reply_markup=markup)
-
-# # Nachricht erkennen
-# @bot.callback_query_handler(func=lambda call: True)
-# def callback_handler(call):
-#     # Antworten wenn API Infos gefordert
-#     if call.data == 'api_warnung':
-#         bot.send_message(call.message.chat.id, 'Antworten zu aktuellen Informationen:')
-#     if call.data == 'allg_infos': 
-#         bot.send_message(call.message.chat.id, 'Antworten zu allgemeinen Informationen:')
-
-# # Get_Message
-# @bot.message_handler(func=lambda message: True)
-# def get_Message(message):
-# 	#bot.reply_to(message, message.text)
-# 	frage = message.text
-# 	print(frage)
-
-# # Bot starten 
-# bot.infinity_polling()
+        self.bot = AsyncTeleBot(token)
+    
+    async def send_status_message(self, message):
+        status_message = (
+        "Hallo! Willkommen beim KatHelferPro Chatbot! Ich bin Rene. Hier gibt es Informationen zum Zivil- und Katastrophenschutz in Deutschland. Beachte bitte, dass ich Fehler machen kann. Prüfe daher Wichtiges nochmal nach."
+            " Frag mich nach allgemeinen Informationen zum Katastrophenschutz, Tipps zur Vorsorge, Verhalten bei verschiedenen Katastrophen und wie du dich (auch) langfristig bei verschiedenen Organisationen engagieren kannst. Aktuelle Pegelstände der Gewässer stehen ebenfalls zur Verfügung."
+            " Sicherheit geht vor! 🚨"
+
+            "                                                                        Impressum Platzhalter"
+        )
+        button_akt_Meldung = types.InlineKeyboardButton('Aktuelle Warnungen', callback_data='akt_Meldung')
+        button_allg = types.InlineKeyboardButton('Allgemeine Informationen zum Katastrophenschutz', callback_data='allg')
+
+        keyboard = types.InlineKeyboardMarkup()
+        keyboard.add(button_allg)
+        keyboard.add(button_akt_Meldung)
+
+        last_message = status_message
+        await self.bot.reply_to(message, text=status_message, reply_markup=keyboard)
+        
+    def start_polling(self):
+        @self.bot.message_handler(commands=['start'])
+        async def send_welcome(message):
+            await self.send_status_message(message)
+
+        @self.bot.message_handler(content_types=['new_chat_members'])
+        async def send_welcome_new_members(message):
+            for member in message.new_chat_members:
+                if member.id == self.bot.get_me().id:
+                    await self.send_status_message(message.chat.id)
+        
+
+        @self.bot.callback_query_handler(func=lambda call: True)
+        async def handle_button_click(call):
+            match call.data:
+                case "allg":
+                    await self.function_allg(call.message)
+                case "akt_Meldung":
+                    # Hier wird die Funktion aufgerufen, die du mit Button 1 verknüpfen möchtest
+                    await self.function_akt_meldungen(call.message)
+
+                case _:
+                    # Hier wird die Funktion aufgerufen, die du mit Button 2 verknüpfen möchtest
+                    last_message = "Der Button ist nicht angebunden."
+                    await self.send_message(call.message.chat.id, last_message)
+                    
+        @self.bot.message_handler(func=lambda message: True)
+        async def get_Message(message):
+            match self.last_message:
+                case "Okay, um welche Region handelt es sich?":
+                # Pauls APi abruf
+                    ort = message.text
+                    antwort = self.gesuchte_zeile.getWarningOrt(ort)
+                    
+                    if antwort == "Keine Warnung gefunden":
+                        erw_ant = "Zu diesem Ort haben wir keine Meldungen. Probiere es mit einem anderen. Oder tippe \start um etwas anderes zu fragen."
+                        self.last_message = "Okay, um welche Region handelt es sich?"
+                        await self.bot.send_message(message.chat.id, erw_ant)
+                    else:
+                        erw_ant = "Wir haben zu diesem Ort folgende Meldungen: \n" + antwort +  "\nSuche nach einem neuen ort oder tippe \start um etwas anderes zu fragen."
+                        self.last_message = "Okay, um welche Region handelt es sich?"
+                        await self.bot.send_message(message.chat.id, erw_ant)
+                case "Okay, es geht um allgemeine Informationen zum Katastrophenschutz. Stelle mir einfach eine Frage und ich gebe mein Bestes, um dir weiterzuhelfen!":
+                    ######zauberei einfügen#####
+                    self.last_message = "Okay, es geht um allgemeine Informationen zum Katastrophenschutz. Stelle mir einfach eine Frage und ich gebe mein Bestes, um dir weiterzuhelfen!"
+                    erw_ant = get_llm_answer(message)
+                    await self.bot.send_message(message.chat.id, erw_ant)        
+                case _:
+                    self.last_message = "Bitte starte den Bot mit /start neu."
+                    await self.bot.send_message(message.chat.id, self.last_message)
+        asyncio.run(self.bot.polling())
+
+    async def function_allg(self, message):
+        antwort = "Okay, es geht um allgemeine Informationen zum Katastrophenschutz. Stelle mir einfach eine Frage und ich gebe mein Bestes, um dir weiterzuhelfen!"
+        self.last_message = antwort
+        await self.bot.send_message(message.chat.id, antwort)
+
+    async def function_akt_meldungen(self, message):
+        antwort = "Okay, um welche Region handelt es sich?"
+        self.last_message = antwort
+        await self.bot.send_message(message.chat.id, antwort)
+
+telegram = Telegram()
+telegram.start_polling()