Update app.py

app.py

@@ -1,10 +1,12 @@
 import gradio as gr
+import re
+import html
 from transformers import pipeline
 import numpy as np
 
 # Initialize pipelines
-ocr = pipeline("image-to-text", model="microsoft/trocr-base-printed")
-classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
+ocr = pipeline("image-to-text", model="microsoft/trocr-base-printed")  # Consider upgrading to trocr-large-printed
+classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")  # Consider fine-tuning for phishing
 
 # Common phishing indicators
 SUSPICIOUS_PHRASES = [
@@ -14,6 +16,9 @@ SUSPICIOUS_PHRASES = [
     "security alert", "confirm your identity", "gift card", "lottery winner"
 ]
 
+# URL detection regex
+URL_PATTERN = r'http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\\(\\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+'
+
 def extract_text_from_image(image):
     if image is None:
         return ""
@@ -67,13 +72,15 @@ def analyze_text(text):
             </div>
         """
     
-    # Find suspicious phrases
+    # Find suspicious phrases and URLs
     found_phrases = []
     text_lower = text.lower()
     for phrase in SUSPICIOUS_PHRASES:
         if phrase in text_lower:
             found_phrases.append(phrase)
     
+    urls = re.findall(URL_PATTERN, text)
+    
     # Generate detailed analysis report with modern styling
     report = [
         "<div style='background: white; padding: 24px; border-radius: 12px; box-shadow: 0 2px 8px rgba(0,0,0,0.05);'>",
@@ -88,17 +95,28 @@ def analyze_text(text):
         "</div>"
     ]
     
-    if found_phrases:
-        report.extend([
-            "<div style='margin-top: 20px;'>",
-            "<h4 style='color: #d32f2f;'>🚩 Suspicious Elements Detected:</h4>",
-            "<ul style='list-style-type: none; padding-left: 0;'>"
-        ])
-        for phrase in found_phrases:
-            report.append(f"<li style='margin-bottom: 8px; padding: 8px 12px; background: #ffebee; border-radius: 6px;'>Found: '{phrase}'</li>")
-        report.append("</ul></div>")
+    if found_phrases or urls:
+        report.append("<div style='margin-top: 20px;'>")
+        if found_phrases:
+            report.extend([
+                "<h4 style='color: #d32f2f;'>🚩 Suspicious Elements Detected:</h4>",
+                "<ul style='list-style-type: none; padding-left: 0;'>"
+            ])
+            for phrase in found_phrases:
+                report.append(f"<li style='margin-bottom: 8px; padding: 8px 12px; background: #ffebee; border-radius: 6px;'>Found: '{html.escape(phrase)}'</li>")
+            report.append("</ul>")
+        
+        if urls:
+            report.extend([
+                "<h4 style='color: #d32f2f;'>🔗 Detected URLs:</h4>",
+                "<ul style='list-style-type: none; padding-left: 0;'>"
+            ])
+            for url in urls:
+                report.append(f"<li style='margin-bottom: 8px; padding: 8px 12px; background: #e3f2fd; border-radius: 6px; word-break: break-all;'>{html.escape(url)}</li>")
+            report.append("</ul>")
+        report.append("</div>")
     else:
-        report.append("<p style='color: #2e7d32;'>✅ No common suspicious phrases detected.</p>")
+        report.append("<p style='color: #2e7d32;'>✅ No common suspicious phrases or URLs detected.</p>")
     
     report.append("<div style='margin-top: 20px; padding: 16px; background: #e3f2fd; border-radius: 8px;'>")
     if confidence > 0.9:
@@ -137,16 +155,26 @@ custom_theme = gr.themes.Soft().set(
     input_background_fill="white",
     button_primary_background_fill="#1a237e",
     button_primary_text_color="white",
+    button_secondary_background_fill="#e0e0e0",
+    button_secondary_text_color="#333",
 )
 
-# Create Gradio interface with modern design
-with gr.Blocks(theme=custom_theme, css="""
+# CSS for responsiveness and polish
+custom_css = """
     .container { max-width: 1000px; margin: auto; }
     .header { text-align: center; margin-bottom: 2rem; }
     .tool-description { max-width: 800px; margin: 0 auto 2rem auto; }
     .input-section { margin-bottom: 2rem; }
     .analysis-section { margin-top: 2rem; }
-""") as demo:
+    @media (max-width: 600px) {
+        .header h1 { font-size: 1.8rem; }
+        .input-section { padding: 0 10px; }
+        .gr-button { width: 100%; margin-bottom: 10px; }
+    }
+"""
+
+# Create Gradio interface
+with gr.Blocks(theme=custom_theme, css=custom_css) as demo:
     gr.HTML("""
         <div class="header">
             <h1 style="color: #1a237e; font-size: 2.5rem; margin-bottom: 1rem;">🛡️ AI Phishing Guard</h1>
@@ -160,17 +188,17 @@ with gr.Blocks(theme=custom_theme, css="""
                 <div class="tool-description">
                     <h3 style="color: #1a237e;">How to Use</h3>
                     <ol style="color: #555; line-height: 1.6;">
-                        <li>Either paste message text or upload a screenshot</li>
-                        <li>Click 'Analyze' to check for phishing indicators</li>
-                        <li>Review the detailed analysis results</li>
+                        <li>Paste message text or upload a screenshot</li>
+                        <li>Analyze instantly as you type or click 'Analyze' for images</li>
+                        <li>Review the detailed results and stay cautious</li>
                     </ol>
                     <div style="background: #e3f2fd; padding: 16px; border-radius: 8px; margin-top: 1rem;">
                         <h4 style="color: #1a237e; margin-top: 0;">This tool detects:</h4>
                         <ul style="color: #555; margin-bottom: 0;">
                             <li>Suspicious language patterns</li>
                             <li>Common phishing phrases</li>
+                            <li>URLs and potential threats</li>
                             <li>Urgency indicators</li>
-                            <li>Security threat language</li>
                         </ul>
                     </div>
                 </div>
@@ -192,31 +220,49 @@ with gr.Blocks(theme=custom_theme, css="""
                 elem_id="image_input"
             )
     
-    analyze_button = gr.Button("🔍 Analyze", variant="primary", size="lg")
+    with gr.Row():
+        analyze_button = gr.Button("🔍 Analyze", variant="primary", size="lg")
+        clear_button = gr.Button("🗑️ Clear", variant="secondary", size="lg")
+    
+    loading = gr.HTML("<div style='text-align: center; padding: 10px;'>Analyzing... <span>⏳</span></div>", visible=False)
     
-    with gr.Column(visible=True) as output_col:
+    with gr.Column(visible=False) as output_col:
         alert_html = gr.HTML()
         analysis = gr.HTML()
     
     # Examples
     examples = [
-        ["Subject: URGENT - Account Security Alert\n\nDear User,\n\nWe detected unusual activity in your account. Click here immediately to verify your identity and reset your password. If you don't respond within 24 hours, your account will be suspended.\n\nBank Security Team", None],
+        ["Subject: URGENT - Account Security Alert\n\nDear User,\n\nWe detected unusual activity in your account. Click here immediately to verify your identity and reset your password at https://fakebank.com/reset. If you don't respond within 24 hours, your account will be suspended.\n\nBank Security Team", None],
         ["Subject: Team Meeting Tomorrow\n\nHi everyone,\n\nJust a reminder that we have our weekly team meeting tomorrow at 10 AM in the main conference room. Please bring your project updates.\n\nBest regards,\nSarah", None],
     ]
     
     gr.Examples(
         examples=examples,
         inputs=[text_input, image_input],
-        outputs=[alert_html, analysis],
+        outputs=[alert_html, analysis, output_col],
         fn=process_input,
         cache_examples=True
     )
     
-    analyze_button.click(
+    # Event handlers
+    def with_loading(*args):
+        return (gr.update(visible=True), *process_input(*args), gr.update(visible=False))
+    
+    text_input.change(
         fn=process_input,
         inputs=[text_input, image_input],
         outputs=[alert_html, analysis, output_col]
     )
+    analyze_button.click(
+        fn=with_loading,
+        inputs=[text_input, image_input],
+        outputs=[loading, alert_html, analysis, output_col]
+    )
+    clear_button.click(
+        fn=lambda: ("", None, "", "", gr.update(visible=False)),
+        inputs=[],
+        outputs=[text_input, image_input, alert_html, analysis, output_col]
+    )
 
 # Launch the app
 if __name__ == "__main__":