Update app.py

app.py

@@ -1,19 +1,14 @@
 import telebot
 import os
-import requests  # For Telegram API requests
+import requests
 import google.generativeai as genai
 import tempfile
 import logging
 import re
-from telegram import Bot
-from telegram.constants import ParseMode
-from telegram.error import TelegramError
+from flask import Flask, request
 from requests.exceptions import RequestException, ConnectionError
 from telebot.apihelper import ApiTelegramException
 import time
-from flask import Flask, request
-from telegram import Update
-from telegram.ext import Dispatcher, CommandHandler, MessageHandler, Filters
 
 # Replace with your actual API keys and bot token
 GOOGLE_API_KEY = 'AIzaSyAYXUMnwmR4nUGDCs97FJJsafcQAPAAuzE'
@@ -50,62 +45,60 @@ model = genai.GenerativeModel(
 chat_session = model.start_chat(
     history=[
         {"role": "user", "parts": ["hi\n"]},
-        {"role": "model", "parts": ["Hello! 👋  How can I help you today? 😊 \n"]},
-        {"role": "user", "parts": ["i am looking for photo booth service?"]},
+        {"role": "model", "parts": ["Hello! 👋 How can I help you today? 😊 \n"]},
+        {"role": "user", "parts": ["I am looking for photo booth service?"]},
         {"role": "model", "parts": ["That's great! 🎉 I can definitely help you with information about Aforative Media's photo booth services. \n\nTo give you the most relevant information, could you tell me a little more about what you're looking for? ..."]},
-        {"role": "user", "parts": ["how much for photo booth services?"]},
+        {"role": "user", "parts": ["How much for photo booth services?"]},
         {"role": "model", "parts": ["You're smart to ask about pricing upfront! 😉 \n\nAforative Media's Mr. & Ms. Booth photo booth services start at **USD 390 for a minimum of 2 hours**. ..."]},
         {"role": "user", "parts": ["How about videography service?"]},
-        {"role": "model", "parts": ["You're thinking about capturing the memories on film too? Excellent choice! Videography adds a whole other dimension to remembering special events.  \n\nAforative Media offers excellent videography services, and just like their photo booths, their videography packages are competitively priced and flexible. ..."]},
+        {"role": "model", "parts": ["You're thinking about capturing the memories on film too? Excellent choice! Videography adds a whole other dimension to remembering special events. \n\nAforative Media offers excellent videography services, and just like their photo booths, their videography packages are competitively priced and flexible. ..."]},
     ]
 )
 
 # Initialize Flask app
 app = Flask(__name__)
 
-# Define your command and message handlers
-def start(update, context):
-    context.bot.send_message(chat_id=update.effective_chat.id, text="Hello, I am your bot!")
+# Define the webhook route
+@app.route('/' + BOT_TOKEN, methods=['POST'])
+def webhook():
+    try:
+        json_str = request.get_data(as_text=True)
+        update = telebot.types.Update.de_json(json_str)
+        handle_message(update)
+    except ConnectionError as e:
+        logger.error(f"Network-related error occurred: {e}")
+    except ApiTelegramException as e:
+        logger.error(f"Telegram API error occurred: {e}")
+    except Exception as e:
+        logger.error(f"Unexpected error occurred: {e}")
+    return 'ok', 200
 
-def handle_message(update, context):
+# Function to handle incoming messages
+def handle_message(update):
+    message = update.message
     try:
-        chat_session = model.start_chat()  # Start a new chat session for context
-        prompt = f"Respond to the user: {update.message.text}"
+        prompt = f"Respond to the user: {message.text}"
         response = chat_session.send_message(prompt)  # Generate response using text and prompt
         response_text = response.text
-        
+
         # Determine whether to reply in private or group chat with Markdown formatting
-        if update.message.chat.type == 'private':
-            context.bot.send_message(chat_id=update.effective_chat.id, text=response_text, parse_mode=ParseMode.MARKDOWN)
+        if message.chat.type == 'private':
+            bot.send_message(message.chat.id, response_text, parse_mode='Markdown')
         else:
-            context.bot.send_message(chat_id=update.effective_chat.id, text=response_text, parse_mode=ParseMode.MARKDOWN)
-            
-        logger.info(f"Response sent to chat_id {update.effective_chat.id}")
+            bot.send_message(message.chat.id, response_text, parse_mode='Markdown')
         
+        logger.info(f"Response sent to chat_id {message.chat.id}")
+
     except Exception as e:
         logger.error(f"Error during GenAI processing: {e}")
         error_message = "Sorry, I can't answer this query right now but I will improve from time to time."
         
-        if update.message.chat.type == 'private':
-            context.bot.send_message(chat_id=update.effective_chat.id, text=error_message, parse_mode=ParseMode.MARKDOWN)
+        if message.chat.type == 'private':
+            bot.send_message(message.chat.id, error_message, parse_mode='Markdown')
         else:
-            context.bot.send_message(chat_id=update.effective_chat.id, text=error_message, parse_mode=ParseMode.MARKDOWN)
-            
-        logger.error(f"Error message sent to chat_id {update.effective_chat.id}")
-
-# Define webhook endpoint
-@app.route('/' + BOT_TOKEN, methods=['POST'])
-def webhook():
-    try:
-        update = Update.de_json(request.get_json(force=True), bot)
-        dispatcher.process_update(update)
-    except ConnectionError as e:
-        logger.error(f"Network-related error occurred: {e}")
-    except TelegramError as e:
-        logger.error(f"Telegram API error occurred: {e}")
-    except Exception as e:
-        logger.error(f"Unexpected error occurred: {e}")
-    return 'ok', 200
+            bot.send_message(message.chat.id, error_message, parse_mode='Markdown')
+        
+        logger.error(f"Error message sent to chat_id {message.chat.id}")
 
 # Function to set the webhook with Telegram
 def set_webhook():