Update app.py

app.py

@@ -173,6 +173,9 @@ async def handler_doc(event):
         os.remove(filename)
     except Exception as e:
         await event.reply(f"❌ Error: {e}")
+user_context = {}  # make sure this is defined once globally
+
+
 @bot.on(events.NewMessage(pattern="/cused"))
 async def correction_used_handler(event):
     if not is_owner(event): return
@@ -186,8 +189,11 @@ async def collect_links(event):
     ctx = user_context.get(event.sender_id)
     if ctx and ctx.get("mode") == "correct_used":
         links = [l.strip() for l in event.text.split() if l.startswith("http")]
-        ctx["links"].extend(links)
-        await event.reply(f"🔗 Collected {len(links)} link(s). Send more or use /doneused.")
+        if links:
+            ctx["links"].extend(links)
+            await event.reply(f"🔗 Collected {len(links)} link(s). Send more or use /doneused.")
+        else:
+            await event.reply("⚠️ No valid links detected. Please send links starting with http.")
 
 
 @bot.on(events.NewMessage(pattern="/doneused"))
@@ -224,7 +230,7 @@ async def process_correction(event):
         if not found:
             not_found += 1
 
-    del user_context[event.sender_id]  # clear context
+    del user_context[event.sender_id]  # Clear mode after done
 
     await event.reply(
         f"✅ Done processing {len(links)} link(s).\n\n"