Update main.py

main.py

@@ -1,7 +1,8 @@
 import os
 import requests
 import logging
-from telethon import TelegramClient, events
+from hydrogram import Client, filters
+from hydrogram.types import Message # Import Message type hint
 
 # --- Configuration ---
 # Set up basic logging
@@ -51,23 +52,32 @@ if template_content is None:
     # You might want to exit(1) here or use a default fallback template
     # template_content = "Default template: An error occurred fetching the primary one."
 
-# Initialize the Telethon client
-# 'bot_session' is the name of the session file that will be created.
-logger.info("Initializing Telegram Client...")
+# Initialize the Hydrogram (Pyrogram) client
+# "my_bot" is the session name. A file named my_bot.session will be created.
+logger.info("Initializing Hydrogram Client...")
 try:
-    client = TelegramClient('bot_session', API_ID, API_HASH)
-    logger.info("Telegram Client initialized.")
+    # When using bot_token, api_id and api_hash are used to authorize the bot owner's actions (if needed)
+    # or simply required by the library structure.
+    app = Client(
+        name="my_bot",
+        api_id=API_ID,
+        api_hash=API_HASH,
+        bot_token=BOT_TOKEN
+        # You might need 'plugins' argument if structuring with plugins
+        # plugins=dict(root="plugins") # Example if you have a 'plugins' folder
+    )
+    logger.info("Hydrogram Client initialized.")
 except Exception as e:
-    logger.error(f"Failed to initialize Telegram Client: {e}")
+    logger.error(f"Failed to initialize Hydrogram Client: {e}")
     exit(1)
 
 # --- Bot Event Handlers ---
-@client.on(events.NewMessage(pattern='/start'))
-async def start_handler(event):
+@app.on_message(filters.command("start"))
+async def start_handler(client: Client, message: Message):
     """Handler for the /start command."""
-    sender = await event.get_sender()
-    sender_name = getattr(sender, 'first_name', 'User') # Get sender's first name
-    logger.info(f"Received /start command from {sender_name} (ID: {event.sender_id})")
+    sender_name = message.from_user.first_name if message.from_user else "User"
+    sender_id = message.from_user.id if message.from_user else "Unknown ID"
+    logger.info(f"Received /start command from {sender_name} (ID: {sender_id})")
 
     # Example of using the fetched template content
     start_message = f"Hello {sender_name}!\n\n"
@@ -76,40 +86,37 @@ async def start_handler(event):
     else:
         start_message += "Could not load the template content."
 
-    await event.reply(start_message)
+    await message.reply_text(start_message) # Use reply_text for text messages
 
-@client.on(events.NewMessage(pattern='/help'))
-async def help_handler(event):
+@app.on_message(filters.command("help"))
+async def help_handler(client: Client, message: Message):
     """Handler for the /help command."""
-    logger.info(f"Received /help command from {event.sender_id}")
-    await event.reply("This is a sample bot.\nCommands:\n/start - Show welcome message and template\n/help - Show this help message")
+    sender_id = message.from_user.id if message.from_user else "Unknown ID"
+    logger.info(f"Received /help command from {sender_id}")
+    await message.reply_text(
+        "This is a sample bot (Hydrogram).\nCommands:\n/start - Show welcome message and template\n/help - Show this help message"
+    )
 
 # Add more handlers as needed
-# @client.on(events.NewMessage)
-# async def message_handler(event):
+# @app.on_message(filters.private & ~filters.command(["start", "help"])) # Example: handle other private messages
+# async def message_handler(client: Client, message: Message):
 #     """Handles any other message."""
-#     logger.info(f"Received message from {event.sender_id}: {event.text}")
+#     sender_id = message.from_user.id if message.from_user else "Unknown ID"
+#     logger.info(f"Received message from {sender_id}: {message.text}")
 #     # Process the message...
-#     # await event.reply(f"You said: {event.text}")
+#     await message.reply_text(f"You said: {message.text}")
 
 
 # --- Main Execution ---
-async def main():
-    """Main function to start the bot."""
+if __name__ == '__main__':
     if template_content is not None:
         logger.info("Template Content Loaded:\n" + template_content[:200] + "..." if len(template_content) > 200 else template_content) # Log first 200 chars
     else:
         logger.warning("Template content is not available.")
 
     logger.info("Starting bot...")
-    # Start the client using the bot token
-    await client.start(bot_token=BOT_TOKEN)
-    logger.info("Bot started successfully and listening for messages.")
-    # Keep the client running until disconnected
-    await client.run_until_disconnected()
+    # Run the client
+    app.run()
+    # No code will execute after app.run() until the bot stops
     logger.info("Bot stopped.")
 
-if __name__ == '__main__':
-    # Run the main function in the asyncio event loop
-    client.loop.run_until_complete(main())
-