changes In Upload

FileStream/bot/plugins/FileHandlers/callback.py

@@ -2,7 +2,7 @@ import math
 import asyncio
 import datetime
 
-from pyrogram import filters, Client
+from pyrogram import filters, Client,raw, types
 from pyrogram.errors import FloodWait
 from pyrogram.enums.parse_mode import ParseMode
 from pyrogram.file_id import FileId, FileType, PHOTO_TYPES
@@ -102,7 +102,7 @@ async def cb_data(bot: Client, update: CallbackQuery):
         disable_web_page_preview=True,
         reply_markup=reply_markup,
     )
-  elif usr_cmd[0] == "pubup":
+  elif usr_cmd[0] == "initilize":
     try:
       user_id = str(usr_cmd[1])
       message_id = int(usr_cmd[2])
@@ -115,7 +115,41 @@ async def cb_data(bot: Client, update: CallbackQuery):
       file_caption = getattr(message, "caption",  f"{get_name(message)}" ) or "None/Unknown"
       file_name = get_name(message)
       print("Daetail",file_caption,file_name)
-      polls=await FileStream.send_poll(user_id, "Is this a poll question?", ["Yes", "No", "Maybe"])
+      await FileStream.send_poll(user_id, "Is this a poll question?", ["Yes", "No", "Maybe"])
+      #print(polls)
+
+    except Exception as e:
+      print(f"An error occurred: {str(e)}")
+      await FileStream.send_message(
+          chat_id=Telegram.ULOG_GROUP,
+          text=f"An error occurred: {str(e)}"
+      )
+
+  elif usr_cmd[0] == "pubup":
+    try:
+      user_id = str(usr_cmd[1])
+      message_id = int(usr_cmd[2])
+      message = await FileStream.get_messages(user_id, message_id)
+      instruction = {
+          "privacy_type": "PUBLIC",
+          "user_id": user_id,
+          "user_type": "TELEGRAM"
+      }
+      file_caption = getattr(message, "caption",  f"{get_name(message)}" ) or "None/Unknown"
+      file_name = get_name(message)
+      print("Daetail",file_caption,"\nFile:",file_name)
+      new_question = file_caption
+      new_options = ["Yes", "No", "Maybe"]
+      poll = raw.types.InputMediaPoll(
+        poll=raw.types.Poll(
+            id=client.rnd_id(),
+            question=new_question,
+            answers=[raw.types.PollAnswer(text=text, option=bytes([i])) for i, text in enumerate(new_options)]
+        )
+      )
+      await FileStream.edit_message_media(chat_id, message_id, media=poll)
+
+      """
       print(polls)
       file_info = get_file_info(message, instruction)
       # Here we are Adding the File Into the database First
@@ -129,7 +163,7 @@ async def cb_data(bot: Client, update: CallbackQuery):
           parse_mode=ParseMode.HTML,
           disable_web_page_preview=True,
           reply_markup=reply_markup,
-      )
+      )"""
     except FloodWait as e:
       print(f"Sleeping for {str(e.value)}s")
       await asyncio.sleep(e.value)

FileStream/bot/plugins/FileHandlers/polls.py

@@ -0,0 +1,10 @@
+from FileStream.bot import FileStream
+
+
+@FileStream.on_poll()
+def handle_poll(FileStream, poll):
+    print(f"Poll ID: {poll.id}")
+    print(f"Question: {poll.question}")
+    print(f"Options: {poll.options}")
+    print(f"Total Voters: {poll.total_voter_count}")
+    #app.loop.create_task(poll_handler(client, poll))

FileStream/bot/plugins/FileHandlers/stream.py

@@ -124,7 +124,6 @@ async def reply_handler(
     message: Message,
     thumb: Union[str, BinaryIO] = None,
 ):
-
     replied_message = message.reply_to_message
     resp = ["n", "no", "not"]
     if not replied_message:
@@ -172,11 +171,9 @@ async def reply_handler(
             file_name=file_info['file']['file_name'],
             file_size=file_info['file']['file_size'],
             progress=progress,
-            progress_args=(replied_message, file_info['file']['file_name'],
-                           file_info['file']['file_size'], start_time))
+            progress_args=(replied_message, file_info['file']['file_name'],file_info['file']['file_size'], start_time))
 
-        success_text = LANG.SUCCESS_PRIV_FILE.format(
-            file_info['file']['file_name'], humanbytes(file_size))
+        success_text = LANG.SUCCESS_PRIV_FILE.format(file_info['file']['file_name'], humanbytes(file_size))
         await replied_message.edit_text(
             text=success_text,
             parse_mode=ParseMode.HTML,