feat: Add SMS and message analysis to scam detection AI

google-labs-jules[bot] · google-labs-jules[bot] · commit 6d886728d545 · 2025-07-19T15:55:22.000Z
diff --git a/collections/scam-detection-ai/app.py b/collections/scam-detection-ai/app.py
@@ -1,6 +1,7 @@
 from flask import Flask, render_template, request
 from email_analyzer import analyze_email
 from email_fetcher import fetch_emails
+from message_analyzer import analyze_message
 
 app = Flask(__name__)
 
@@ -15,7 +16,14 @@ def analyze():
         analysis = analyze_email(email_text)
         total_score = sum(analysis.values())
         is_scam = total_score >= 3
-        return render_template('result.html', analysis=analysis, is_scam=is_scam)
+        return render_template('result.html', analysis=analysis, is_scam=is_scam, analysis_type='Email')
+
+    elif 'message_text' in request.form:
+        message_text = request.form['message_text']
+        analysis = analyze_message(message_text)
+        total_score = sum(analysis.values())
+        is_scam = total_score >= 2
+        return render_template('result.html', analysis=analysis, is_scam=is_scam, analysis_type='Message')
 
     elif 'fetch_emails' in request.form:
         username = request.form['username']
diff --git a/collections/scam-detection-ai/message_analyzer.py b/collections/scam-detection-ai/message_analyzer.py
@@ -0,0 +1,63 @@
+import re
+
+def analyze_message(message_text):
+    """
+    Analyzes an SMS or other short message for potential scam indicators.
+
+    Args:
+        message_text: The full text of the message.
+
+    Returns:
+        A dictionary containing the analysis results.
+    """
+    scam_indicators = {
+        "shortened_urls": 0,
+        "unusual_sender": 0,
+        "urgent_requests": 0,
+        "prize_offers": 0
+    }
+
+    # Check for shortened URLs
+    shortened_url_patterns = [r'bit\.ly', r't\.co', r'goo\.gl', r'tinyurl\.com']
+    for pattern in shortened_url_patterns:
+        if re.search(pattern, message_text):
+            scam_indicators["shortened_urls"] = 1
+            break
+
+    # Check for unusual sender (e.g., non-numeric sender ID)
+    # This is a very basic check and would need to be improved.
+    sender_match = re.search(r'From:\s*(\S+)', message_text, re.IGNORECASE)
+    if sender_match:
+        sender = sender_match.group(1)
+        if not sender.isdigit() and len(sender) > 10:
+            scam_indicators["unusual_sender"] = 1
+
+    # Check for urgent requests
+    urgent_keywords = ["urgent", "verify", "confirm", "login", "account"]
+    for keyword in urgent_keywords:
+        if re.search(r'\b' + keyword + r'\b', message_text, re.IGNORECASE):
+            scam_indicators["urgent_requests"] = 1
+            break
+
+    # Check for prize offers
+    prize_keywords = ["congratulations", "winner", "prize", "claim", "won"]
+    for keyword in prize_keywords:
+        if re.search(r'\b' + keyword + r'\b', message_text, re.IGNORECASE):
+            scam_indicators["prize_offers"] = 1
+            break
+
+    return scam_indicators
+
+if __name__ == '__main__':
+    sample_message = """
+    From: 555-123-4567
+    Congratulations! You've won a $1000 gift card. Click here to claim: http://bit.ly/fake-prize
+    """
+    analysis = analyze_message(sample_message)
+    print(analysis)
+
+    total_score = sum(analysis.values())
+    if total_score >= 2:
+        print("\nThis message is likely a scam!")
+    else:
+        print("\nThis message seems legitimate.")
diff --git a/collections/scam-detection-ai/templates/index.html b/collections/scam-detection-ai/templates/index.html
@@ -22,6 +22,15 @@ <h2>Analyze Email Text</h2>
             <input type="submit" value="Analyze">
         </form>
 
+        <hr>
+
+        <h2>Analyze SMS/Message Text</h2>
+        <form action="/analyze" method="post">
+            <textarea name="message_text" placeholder="Paste message text here..."></textarea>
+            <br><br>
+            <input type="submit" value="Analyze">
+        </form>
+
         <div class="credentials">
             <h2>Fetch and Analyze Emails from Your Inbox</h2>
             <form action="/analyze" method="post">
diff --git a/collections/scam-detection-ai/templates/result.html b/collections/scam-detection-ai/templates/result.html
@@ -12,9 +12,9 @@
     </style>
 </head>
 <body>
-    <h1>Analysis Result</h1>
+    <h1>{{ analysis_type }} Analysis Result</h1>
     <div class="result {{ 'scam' if is_scam else 'legit' }}">
-        <h2>{{ 'This email is likely a scam!' if is_scam else 'This email seems legitimate.' }}</h2>
+        <h2>{{ 'This ' + analysis_type.lower() + ' is likely a scam!' if is_scam else 'This ' + analysis_type.lower() + ' seems legitimate.' }}</h2>
         <ul>
             {% for indicator, value in analysis.items() %}
                 {% if indicator != 'domain_analysis' %}
@@ -32,6 +32,6 @@ <h3>Domain Analysis</h3>
         {% endif %}
     </div>
     <br>
-    <a href="/">Analyze another email</a>
+    <a href="/">Analyze another message or email</a>
 </body>
 </html>
diff --git a/collections/scam-detection-ai/test_message_analyzer.py b/collections/scam-detection-ai/test_message_analyzer.py
@@ -0,0 +1,20 @@
+import unittest
+from message_analyzer import analyze_message
+
+class TestMessageAnalyzer(unittest.TestCase):
+
+    def test_scam_message(self):
+        scam_message = "Congratulations! You've won a prize. Click http://bit.ly/scam to claim."
+        analysis = analyze_message(scam_message)
+        self.assertEqual(analysis['shortened_urls'], 1)
+        self.assertEqual(analysis['prize_offers'], 1)
+
+    def test_legitimate_message(self):
+        legitimate_message = "Your appointment is confirmed for 3 PM tomorrow."
+        analysis = analyze_message(legitimate_message)
+        self.assertEqual(analysis['shortened_urls'], 0)
+        self.assertEqual(analysis['prize_offers'], 0)
+        self.assertEqual(analysis['urgent_requests'], 0)
+
+if __name__ == '__main__':
+    unittest.main()
