Update main.py

main.py

@@ -1,34 +1,42 @@
-import telegram
-from telegram import Bot
-from telegram.ext import *
-import PIL.Image
+%%writefile app.py
+import logging
+import shutil
+import requests
 import os
+import PIL.Image
 import google.generativeai as genai
 import requests
 import speech_recognition as sr
 import pydub
 from pydub import AudioSegment
+from telegram import Update
+from telegram.ext import (ApplicationBuilder, CommandHandler, ContextTypes,
+                          MessageHandler, filters)
+a=[]
 api_genai = os.environ.get("gemini_api")
 api_tele = os.environ.get("tele_api")
+TELEGRAM_API_TOKEN = f"{api_tele}"
 genai.configure(api_key=f'{api_genai}')
-model = genai.GenerativeModel('gemini-pro-vision')
-bot = Bot(f"{api_tele}")
-a=[]
 
+# logging.basicConfig(
+#     format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+#     level=logging.INFO
+# )
+async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    await context.bot.send_message(chat_id=update.effective_chat.id, text="I'm a bot, please talk to me!")
 
-def start_command(update, context):
-    name = update.message.chat.first_name
-    update.message.reply_text("Hello " + name)
-    update.message.reply_text("Please share your image")
-
-def image_handler(update, context):
-    #print(update)
+async def echo(update: Update, context: ContextTypes.DEFAULT_TYPE):
     if update.message.photo:
       file = update.message.photo[-1].file_id
-      obj = context.bot.get_file(file)
-      obj.download(f'{file}.png')
-      text=None
       a.append(file)
+      obj = await context.bot.getFile(file)
+      
+      response = requests.get(obj.file_path, stream=True)
+      if response.status_code == 200:
+          with open(f'{file}.png', 'wb') as file:
+              file.write(response.content)
+      text=None
+      
     else:
       text = update.message.text
 
@@ -37,23 +45,24 @@ def image_handler(update, context):
       img = PIL.Image.open(f'{a[-1]}.png')
     except:
       img=None
-
+    
+    
     if img is not None and text is not None:
       print(text,a[-1])
       model = genai.GenerativeModel('gemini-pro-vision')
       response = model.generate_content([img,text])
-      update.message.reply_text(response.text)
+      await context.bot.send_message(chat_id=update.effective_chat.id, text=response.text)
       text=None
       img=None
       os.remove(f'{a[-1]}.png')
-    if text is not None and img is None:
-      model = genai.GenerativeModel('gemini-pro')
-      response = model.generate_content(text)
-      update.message.reply_text(response.text)
+    
     if update.message.voice:
       file_aud = update.message.voice.file_id
-      obj = context.bot.get_file(file_aud)
-      obj.download(f'{file_aud}.ogg')
+      obj = await context.bot.getFile(file_aud)
+      response = requests.get(obj.file_path, stream=True)
+      if response.status_code == 200:
+          with open(f'{file_aud}.ogg', 'wb') as file:
+              file.write(response.content)
       audio_file = f'{file_aud}.ogg'
 
       # Convert the audio file to WAV format
@@ -72,21 +81,22 @@ def image_handler(update, context):
       print(text1)
       model = genai.GenerativeModel('gemini-pro')
       response = model.generate_content(text1)
-      update.message.reply_text(response.text)
+      await context.bot.send_message(chat_id=update.effective_chat.id, text=response.text)
       os.remove(f'{file_aud}.wav')
       os.remove(f'{file_aud}.ogg')
 
+    if text is not None and img is None:
+      model = genai.GenerativeModel('gemini-pro')
+      response = model.generate_content(text)
+      await context.bot.send_message(chat_id=update.effective_chat.id, text=response.text)
 
-def main():
-    print("Started")
-    TOKEN = f"{api_tele}"
-    updater = Updater(TOKEN, use_context=True)
-    dp = updater.dispatcher
-    dp.add_handler(CommandHandler("start", start_command))
-    dp.add_handler(MessageHandler(Filters.text | Filters.photo | Filters.audio | Filters.voice, image_handler))
+if __name__ == '__main__':
+    application = ApplicationBuilder().token(TELEGRAM_API_TOKEN).build()
 
-    updater.start_polling()
-    updater.idle()
+    start_handler = CommandHandler('start', start)
+    echo_handler = MessageHandler(filters.PHOTO & (~filters.COMMAND) | filters.TEXT & (~filters.COMMAND) | 
+                                  filters.VOICE & (~filters.COMMAND), echo)
+    application.add_handler(start_handler)
+    application.add_handler(echo_handler)
 
-if __name__ == '__main__':
-    main()
+    application.run_polling()