Fix deprecated forward_from_chat and add missing get_client import

utils/bot_mode.py

@@ -11,6 +11,7 @@ from utils.mongo_indexer import (
     index_channel_messages_with_iter,
     MongoNotConfiguredError,
 )
+from utils.clients import get_client
 
 logger = Logger(__name__)
 
@@ -135,13 +136,21 @@ async def index_handler(client: Client, message: Message):
                     except Exception:
                         await message.reply_text("❌ Invalid message link!")
                         return
-                # Check if it's a forwarded message
-                elif msg_input.forward_from_chat:
-                    forward_chat_id = msg_input.forward_from_chat.username or msg_input.forward_from_chat.id
-                    if forward_chat_id != chat_id:
-                        await message.reply_text("❌ Forwarded message is from a different channel!")
+                # Check if it's a forwarded message (using new forward_origin API)
+                elif msg_input.forward_origin:
+                    from pyrogram.types import MessageOriginChannel
+                    if isinstance(msg_input.forward_origin, MessageOriginChannel):
+                        forward_chat = msg_input.forward_origin.chat
+                        forward_chat_id = forward_chat.username if forward_chat else None
+                        if not forward_chat_id:
+                            forward_chat_id = msg_input.forward_origin.chat.id if msg_input.forward_origin.chat else None
+                        if forward_chat_id != chat_id:
+                            await message.reply_text("❌ Forwarded message is from a different channel!")
+                            return
+                        last_msg_id = msg_input.forward_origin.message_id
+                    else:
+                        await message.reply_text("❌ Please forward a message from a channel.")
                         return
-                    last_msg_id = msg_input.forward_from_message_id
                 # Check if it's just a number (skip count)
                 elif msg_input.text and msg_input.text.strip().isdigit():
                     skip = int(msg_input.text.strip())
@@ -209,15 +218,24 @@ async def index_handler(client: Client, message: Message):
             except Exception:
                 await message.reply_text("❌ Invalid message link!")
                 return
-        # Check if it's a forwarded message
-        elif msg.forward_from_chat:
-            chat_type = getattr(msg.forward_from_chat, 'type', None)
-            type_str = str(chat_type).lower()
-            if "channel" not in type_str and not getattr(msg.forward_from_chat, 'is_channel', False):
+        # Check if it's a forwarded message (using new forward_origin API)
+        elif msg.forward_origin:
+            from pyrogram.types import MessageOriginChannel
+            if isinstance(msg.forward_origin, MessageOriginChannel):
+                forward_chat = msg.forward_origin.chat
+                chat_type = getattr(forward_chat, 'type', None) if forward_chat else None
+                type_str = str(chat_type).lower() if chat_type else ""
+                if "channel" not in type_str and not getattr(forward_chat, 'is_channel', False) if forward_chat else False:
+                    await message.reply_text("❌ Please forward a message from a channel.")
+                    return
+                last_msg_id = msg.forward_origin.message_id
+                chat_id = forward_chat.username if forward_chat and forward_chat.username else (forward_chat.id if forward_chat else None)
+                if chat_id is None:
+                    await message.reply_text("❌ Could not determine channel from forwarded message.")
+                    return
+            else:
                 await message.reply_text("❌ Please forward a message from a channel.")
                 return
-            last_msg_id = msg.forward_from_message_id
-            chat_id = msg.forward_from_chat.username or msg.forward_from_chat.id
         else:
             await message.reply_text("❌ Please forward a message from a channel or send a message link.")
             return