Add file processing logic to prevent duplicate file handling in BotController

src/controllers/_bot_controller.py

@@ -8,6 +8,8 @@ from src.models import ContentDeliveryFrequency
 from src.services import FileService, SlackBotService, SlackTeamService, UserService
 from src.utils import logger
 
+file_set = set()
+
 
 class BotController:
     def __init__(self):
@@ -23,11 +25,25 @@ class BotController:
         return f"""Hi {firstname} 👋\n\nWelcome to the Mandala Slackbot! Every week, you'll receive a learning tailored from the Resilient Leadership program to support you on your leadership journey. If you'd like to make this content even more personalized, share your Resilient Leadership guidebook with us either as an attachment or a link to the slide deck.\n\nOtherwise, we'll plan on sharing our curated weekly insights to keep you growing. 🙂\n\nYou can refresh the content at any time by simply typing */refresh* and sending that guidebook back our way. Looking forward to getting started! Your first learning will arrive next week.\n\nWarmly,\nTeam Mandala"""
 
     async def process_event_background(self, event: dict):
+        file_ids = []
+        global file_set
+        send_message = True
         try:
-
             actual_event = event["event"]
-            slack_user_id = actual_event.get("user") or actual_event.get("user_id")
+            if "files" in actual_event:
+                file_ids = [file.get("id") for file in actual_event.get("files", [])]
+                if all(file_id in file_set for file_id in file_ids):
+                    return
+                if any(file_id in file_set for file_id in file_ids):
+                    send_message = False
+                file_set.update(file_ids)
+            if "file" in actual_event:
+                file_ids = [actual_event.get("file").get("id")]
+                if any(file_id in file_set for file_id in file_ids):
+                    return
+                file_set.update(file_ids)
 
+            slack_user_id = actual_event.get("user") or actual_event.get("user_id")
             async with UserService() as user_service:
                 user = await user_service.get_user_by_slack_id(slack_user_id)
                 if (
@@ -44,14 +60,17 @@ class BotController:
                     raise Exception("User not found")
 
                 if "files" in actual_event or "file" in actual_event:
-                    await user_service.send_message(
-                        user.id,
-                        "You're very welcome! 🎉 We'll be sharing these insights with you gradually over the next few weeks. 😊",
-                    )
+                    if send_message:
+                        await user_service.send_message(
+                            user.id,
+                            "You're very welcome! 🎉 We'll be sharing these insights with you gradually over the next few weeks. 😊",
+                        )
                     await self.handle_files(event, actual_event, user, user_service)
-
         except Exception as e:
             logger.error(f"Background task error: {e}")
+            logger.exception(e)
+        finally:
+            file_set.difference_update(file_ids)
 
     async def handle_new_user(self, event, slack_user_id, user_service):
         async with SlackTeamService() as slack_team_service:
@@ -112,15 +131,13 @@ class BotController:
 
             uploaded_file = await slack_bot_service.save_file(team, file)
             if uploaded_file:
-                await file_service.process_file(
+                return await file_service.process_file(
                     file=uploaded_file,
                     user=user,
                     slack_file_id=file_id,
                     slack_token=team.token,
                 )
-            else:
-                logger.error(f"Failed to save file {file_id}")
-
+            raise Exception(f"Failed to save file")
         except Exception as e:
             logger.error(f"Error processing file {file_id}: {e}")
             await user_service.send_message(