Callback Update

FileStream/Database/database.py

@@ -20,6 +20,7 @@ class Database:
     self.files = self.db.Public_Files
     self.pfile = self.db.Private_Files
     self.web_upload = self.db.Web_Files
+    self.web_upload = self.db.Temp_Files
 
 #---------------------[ SCHEMAS ]------------------------------#
 #---------------------[ NEW USER ]---------------------#

FileStream/bot/plugins/FileHandlers/callback.py

@@ -106,6 +106,40 @@ async def cb_data(bot: Client, update: CallbackQuery):
         disable_web_page_preview=True,
         reply_markup=reply_markup,
     )
+  elif usr_cmd[0] == "initialize":
+    try:
+      upload_type=str(usr_cmd[1])
+      user_id = str(usr_cmd[2])
+      message_id = int(usr_cmd[3])
+
+      await update.message.edit_text(
+          text=f"Please Choose The File Type",
+          parse_mode=ParseMode.HTML,
+          disable_web_page_preview=True,
+          reply_markup=InlineKeyboardMarkup([
+                [InlineKeyboardButton("Movie", callback_data=f"{'privup' if upload_type =='PRIVATE' else 'pubup'}_{instruction['user_id']}_{message_id}_Movie")],
+                [InlineKeyboardButton("WebSeries", callback_data=f"{'privup' if upload_type =='PRIVATE' else 'pubup'}_{instruction['user_id']}_{message_id}_WebSeries")],
+                [InlineKeyboardButton("Anime", callback_data=f"{'privup' if upload_type =='PRIVATE' else 'pubup'}_{instruction['user_id']}_{message_id}_Anime")],
+                [InlineKeyboardButton("Documentary", callback_data=f"{'privup' if upload_type =='PRIVATE' else 'pubup'}_{instruction['user_id']}_{message_id}_Documentary")],
+                [InlineKeyboardButton("Regular", callback_data=f"{'privup' if upload_type =='PRIVATE' else 'pubup'}_{instruction['user_id']}_{message_id}_Regular")],
+                [ InlineKeyboardButton("ᴄʟᴏsᴇ", callback_data="close")]
+            ]),
+        )   
+    except FloodWait as e:
+      print(f"Sleeping for {str(e.value)}s")
+      await asyncio.sleep(e.value)
+      await FileStream.send_message(
+          chat_id=Telegram.ULOG_GROUP,
+          text=
+          f"Gᴏᴛ FʟᴏᴏᴅWᴀɪᴛ ᴏғ {str(e.value)}s ғʀᴏᴍ [{message.from_user.first_name}](tg://user?id={message.from_user.id})\n\n**ᴜsᴇʀ ɪᴅ :** `{str(message.from_user.id)}`",
+          disable_web_page_preview=True,
+          parse_mode=ParseMode.MARKDOWN)
+    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])

FileStream/bot/plugins/FileHandlers/stream.py

@@ -65,19 +65,12 @@ async def private_receive_handler(bot: Client, message: Message):
         
         reply = await message.reply_text(LANG.PROCESSING_TEXT)
         file_info=get_file_info(message, instruction)
-        await reply.edit_text(
-          text=f"{file_info['file']['caption'] if file_info['file']['caption'] else file_info['file']['file_name']}",
-          parse_mode=ParseMode.HTML,
-          disable_web_page_preview=True,
-          reply_markup=InlineKeyboardMarkup([
-                [InlineKeyboardButton("Movie", callback_data=f"pubup_{instruction['user_id']}_{message.id}_Movie")],
-                [InlineKeyboardButton("WebSeries", callback_data=f"pubup_{instruction['user_id']}_{message.id}_WebSeries")],
-                [InlineKeyboardButton("Anime", callback_data=f"pubup_{instruction['user_id']}_{message.id}_Anime")],
-                [InlineKeyboardButton("Documentary", callback_data=f"pubup_{instruction['user_id']}_{message.id}_Documentary")],
-                [InlineKeyboardButton("Regular", callback_data=f"pubup_{instruction['user_id']}_{message.id}_Regular")],
-                [ InlineKeyboardButton("ᴄʟᴏsᴇ", callback_data="close")]
-            ]),
-        )   
+        reply_markup, stream_text = await upload_type_func(get_file_info(message, instruction))
+        await reply.edit_text(text=stream_text,
+                                 parse_mode=ParseMode.HTML,
+                                 disable_web_page_preview=True,
+                                 reply_markup=reply_markup,
+                                 quote=True)
     except FloodWait as e:
         print(f"Sleeping for {str(e.value)}s")
         await asyncio.sleep(e.value)

FileStream/utils/FileProcessors/bot_utils.py

@@ -109,11 +109,11 @@ async def upload_type_func(file_info,replied_message):
     reply_markup = InlineKeyboardMarkup([[
         InlineKeyboardButton(
             "PUBLIC UPLOAD",
-            callback_data=f"pubup_{file_info['user_id']}_{file_info['message_id']}_{replied_message}"),
+            callback_data=f"initialize_PUBLIC_{file_info['user_id']}_{file_info['message_id']}_{replied_message}"),
         InlineKeyboardButton(
             "PRIVATE UPLOAD",
             callback_data=
-            f"privup_{file_info['user_id']}_{file_info['message_id']}")
+            f"initialize_PRIVATE_{file_info['user_id']}_{file_info['message_id']}")
     ], [InlineKeyboardButton("ᴄʟᴏsᴇ", callback_data="close")]])
 
     return reply_markup, stream_text