Update main.py

main.py

@@ -85,17 +85,30 @@ def reply_to_mentions():
         return
 
     for mention in reversed(mentions.data):
-        tweet_to_roast_id = mention.referenced_tweets[0].id
-        tweet_to_roast = client.get_tweet(id=tweet_to_roast_id)
-        text_to_roast = tweet_to_roast.data.text
-
-        text_out = None
-        if "roast" in mention.text.lower():
-            text_out = roast(text_to_roast)
-        elif "toast" in mention.text.lower():
-            text_out = toast(text_to_roast)
-
-        if text_out is None:
+        try:
+            if mention.author_id == 1612106815096999936:
+                continue
+            
+            if mention.referenced_tweets is None:
+                logger.info(f"Skipping {mention.id} as it is not a reply")
+            tweet_to_roast_id = mention.referenced_tweets[0].id
+            tweet_to_roast = client.get_tweet(id=tweet_to_roast_id)
+            text_to_roast = tweet_to_roast.data.text
+
+            text_out = None
+            text_in = mention.text.lower()
+            text_in.replace('@roastortoastgpt', '')
+            text_in.replace('roastortoastgpt','')
+            
+            if "roast" in text_in:
+                text_out = roast(text_to_roast)
+            elif "toast" in text_in:
+                text_out = toast(text_to_roast)
+
+            if text_out is None:
+                continue
+        except Exception as e:
+            logger.error(e)
             continue
 
         try: