From a93c61045b0523120482e2520a77c88d8f73409e Mon Sep 17 00:00:00 2001
From: yasir <yasiramunandar@gmail.com>
Date: Thu, 9 Feb 2023 15:20:20 +0700
Subject: [PATCH] gennss link

---
 misskaty/helper/ssgen_helper.py | 22 +++++++++++++---------
 1 file changed, 13 insertions(+), 9 deletions(-)

diff --git a/misskaty/helper/ssgen_helper.py b/misskaty/helper/ssgen_helper.py
index 6b244bb9..a8d5d0bf 100644
--- a/misskaty/helper/ssgen_helper.py
+++ b/misskaty/helper/ssgen_helper.py
@@ -1,14 +1,18 @@
-from misskaty.core.message_utils import *
-import uuid
-import os
-import time
-import shlex
-from pathlib import Path
 import datetime
-from pyrogram import enums
+import os
 import random
+import shlex
+import time
 import traceback
-from pyrogram.types import InputMediaPhoto, InlineKeyboardButton, InlineKeyboardMarkup
+import uuid
+from pathlib import Path
+
+from pyrogram import enums
+from pyrogram.types import (InlineKeyboardButton, InlineKeyboardMarkup,
+                            InputMediaPhoto)
+
+from misskaty.core.message_utils import *
+
 
 async def run_subprocess(cmd):
     process = await asyncio.create_subprocess_shell(
@@ -75,7 +79,7 @@ async def screenshot_flink(c, m):
         await editPesan(m, 'Give me some time bruh!! 😴')
         
         await editPesan(m, '😀 Taking Snaps!')
-        file_link = m.command[1]
+        file_link = m.reply_to_message.command[1]
         duration = await get_duration(file_link)
         if isinstance(duration, str):
             await editPesan(m, "Oops, What's that? Couldn't Open the file😟.")