Update app.py

app.py

@@ -136,19 +136,26 @@ def analyze_phishing(content):
     
     return classification, confidence, analysis, inference_time
 
-def batch_analyze(file_content):
+def batch_analyze(file_path):
     """
     Batch analysis function for file upload
     
     Args:
-        file_content (str): File content with one URL/text per line
+        file_path (str): Path to uploaded file
         
     Returns:
         str: Formatted results
     """
-    if not file_content:
+    if not file_path:
         return "Please upload a file with content to analyze (one item per line)"
     
+    try:
+        # Read the file content
+        with open(file_path, 'r', encoding='utf-8') as f:
+            file_content = f.read()
+    except Exception as e:
+        return f"Error reading file: {str(e)}"
+    
     lines = [line.strip() for line in file_content.split('\n') if line.strip()]
     
     if not lines:
@@ -262,7 +269,7 @@ examples = [
     
     # Cryptocurrency/Investment Scams
     ["Make $10,000 per day with Bitcoin! Limited time offer - invest now: bitcoin-millionaire.crypto-scam.org"],
-    ["Elon Musk is giving away FREE cryptocurrency! Claim yours now: musk-crypto-giveaway.fake-tesla.com"],
+    ["Elon Musk is giving750 giving away FREE cryptocurrency! Claim yours now: musk-crypto-giveaway.fake-tesla.com"],
     ["Join our exclusive trading group. 1000% returns guaranteed: forex-millionaire.trading-scam.net"],
     
     # Fake Government/Authority Messages
@@ -338,13 +345,33 @@ examples = [
     ["I lost 100 pounds without diet or exercise! Here's my secret: weight-loss-secret.fitness-fraud.net"]
 ]
 
-# Create Gradio interface
+# Quick test button functions
+def set_suspicious_1():
+    return "Urgent: Your account will be suspended in 24 hours! Verify now: secure-verification.fake-bank.com"
+
+def set_suspicious_2():
+    return "Congratulations! You've won $10,000! Claim immediately: lottery-winner.scam-site.org"
+
+def set_suspicious_3():
+    return "Apple ID locked due to suspicious activity. Unlock now: apple-security.phishing-domain.net"
+
+def set_legitimate_1():
+    return "Your monthly statement is ready for download on our secure banking portal."
+
+def set_legitimate_2():
+    return "Thank you for your purchase. Your order will ship within 2-3 business days."
+
+def set_legitimate_3():
+    return "Appointment reminder: Your doctor's appointment is scheduled for tomorrow at 2 PM."
+
+# Create Gradio interface with center alignment
 with gr.Blocks(
     title="🔐 PhishGuard AI - Advanced Phishing Detection",
-    theme=gr.themes.Soft(),
+    theme=gr.themes.Ocean(),
     css="""
     .gradio-container {
         max-width: 1400px !important;
+        margin: 0 auto !important;
     }
     .title {
         text-align: center;
@@ -366,183 +393,218 @@ with gr.Blocks(
         border: 2px solid #e1e5e9;
         border-radius: 10px;
         padding: 1em;
-        margin: 0.5em;
+        margin: 0.5em auto;
         background: linear-gradient(135deg, #f5f7fa 0%, #c3cfe2 100%);
+        text-align: center;
+    }
+    .container {
+        text-align: center;
+    }
+    .main-content {
+        margin: 0 auto;
+        padding: 20px;
+    }
+    .tab-nav {
+        justify-content: center;
+    }
+    .gradio-row {
+        justify-content: center;
+    }
+    .gradio-column {
+        display: flex;
+        flex-direction: column;
+        align-items: center;
     }
     """
 ) as app:
     
     gr.HTML("""
-    <div class="title">🔐 PhishGuard AI</div>
-    <div class="subtitle">
-        🚀 Advanced AI-Powered Phishing Detection System<br>
-        Analyze URLs, emails, SMS messages, and social content for sophisticated threats
+    <div class="container">
+        <div class="title">🔐 PhishGuard AI</div>
+        <div class="subtitle">
+            🚀 Advanced AI-Powered Phishing Detection System<br>
+            Analyze URLs, emails, SMS messages, and social content for sophisticated threats
+        </div>
     </div>
     """)
     
     with gr.Tabs():
         # Single Analysis Tab
         with gr.TabItem("🔍 Single Analysis", elem_id="single-analysis"):
-            gr.Markdown("### 🎯 Analyze Individual Content")
-            gr.Markdown("Paste any suspicious URL, email, SMS, or text content below for instant AI analysis")
-            
-            with gr.Row():
-                with gr.Column(scale=2):
-                    input_text = gr.Textbox(
-                        label="📝 Enter Content to Analyze",
-                        placeholder="Examples: URLs, email content, SMS messages, social media posts, or any suspicious text...",
-                        lines=4,
-                        max_lines=12
-                    )
+            with gr.Column(elem_classes=["main-content"]):
+                gr.Markdown("### 🎯 Analyze Individual Content", elem_classes=["container"])
+                gr.Markdown("Paste any suspicious URL, email, SMS, or text content below for instant AI analysis", elem_classes=["container"])
+                
+                with gr.Row():
+                    with gr.Column(scale=2):
+                        input_text = gr.Textbox(
+                            label="📝 Enter Content to Analyze",
+                            placeholder="Examples: URLs, email content, SMS messages, social media posts, or any suspicious text...",
+                            lines=4,
+                            max_lines=12
+                        )
+                        
+                        with gr.Row():
+                            analyze_btn = gr.Button("🔍 Analyze Content", variant="primary", size="lg")
+                            clear_btn = gr.Button("🗑️ Clear", variant="secondary")
                     
-                    with gr.Row():
-                        analyze_btn = gr.Button("🔍 Analyze Content", variant="primary", size="lg")
-                        clear_btn = gr.Button("🗑️ Clear", variant="secondary")
+                    with gr.Column(scale=1):
+                        with gr.Group():
+                            classification_output = gr.Textbox(label="🎯 Classification", interactive=False)
+                            confidence_output = gr.Textbox(label="📊 Confidence Level", interactive=False)
+                            time_output = gr.Textbox(label="⚡ Analysis Time", interactive=False)
                 
-                with gr.Column(scale=1):
-                    with gr.Group():
-                        classification_output = gr.Textbox(label="🎯 Classification", interactive=False)
-                        confidence_output = gr.Textbox(label="📊 Confidence Level", interactive=False)
-                        time_output = gr.Textbox(label="⚡ Analysis Time", interactive=False)
-            
-            analysis_output = gr.Textbox(
-                label="🔬 Detailed AI Analysis",
-                lines=10,
-                max_lines=20,
-                interactive=False,
-                placeholder="Detailed analysis will appear here..."
-            )
-            
-            # Enhanced Examples section with categories
-            gr.Markdown("### 📚 Comprehensive Test Examples")
-            gr.Markdown("Try these diverse examples to explore the AI's detection capabilities:")
-            
-            with gr.Accordion("🏦 Banking & Finance", open=False):
-                gr.Examples(
-                    examples=[ex for ex in examples if any(keyword in ex[0].lower() for keyword in ['paypal', 'bank', 'chase', 'credit', 'wellsfargo', 'visa'])],
-                    inputs=[input_text],
-                    outputs=[classification_output, confidence_output, analysis_output, time_output],
-                    fn=analyze_phishing,
-                    cache_examples=False
-                )
-            
-            with gr.Accordion("🛒 E-commerce & Shopping", open=False):
-                gr.Examples(
-                    examples=[ex for ex in examples if any(keyword in ex[0].lower() for keyword in ['amazon', 'ebay', 'apple', 'microsoft', 'netflix'])],
-                    inputs=[input_text],
-                    outputs=[classification_output, confidence_output, analysis_output, time_output],
-                    fn=analyze_phishing,
-                    cache_examples=False
-                )
-            
-            with gr.Accordion("📧 Email Scams", open=False):
-                gr.Examples(
-                    examples=[ex for ex in examples if len(ex[0]) > 100 and any(keyword in ex[0].lower() for keyword in ['urgent', 'congratulations', 'won', 'grant', 'lottery'])],
-                    inputs=[input_text],
-                    outputs=[classification_output, confidence_output, analysis_output, time_output],
-                    fn=analyze_phishing,
-                    cache_examples=False
-                )
-            
-            with gr.Accordion("📱 SMS & Text Messages", open=False):
-                gr.Examples(
-                    examples=[ex for ex in examples if any(keyword in ex[0].lower() for keyword in ['alert', 'package', 'verification', 'expires', 'code'])],
-                    inputs=[input_text],
-                    outputs=[classification_output, confidence_output, analysis_output, time_output],
-                    fn=analyze_phishing,
-                    cache_examples=False
-                )
-            
-            with gr.Accordion("💻 Tech Support Scams", open=False):
-                gr.Examples(
-                    examples=[ex for ex in examples if any(keyword in ex[0].lower() for keyword in ['virus', 'infected', 'security warning', 'update required', 'antivirus'])],
-                    inputs=[input_text],
-                    outputs=[classification_output, confidence_output, analysis_output, time_output],
-                    fn=analyze_phishing,
-                    cache_examples=False
-                )
-            
-            with gr.Accordion("💰 Investment & Crypto Scams", open=False):
-                gr.Examples(
-                    examples=[ex for ex in examples if any(keyword in ex[0].lower() for keyword in ['bitcoin', 'crypto', 'investment', 'trading', 'returns'])],
-                    inputs=[input_text],
-                    outputs=[classification_output, confidence_output, analysis_output, time_output],
-                    fn=analyze_phishing,
-                    cache_examples=False
-                )
-            
-            with gr.Accordion("💼 Job & Employment Scams", open=False):
-                gr.Examples(
-                    examples=[ex for ex in examples if any(keyword in ex[0].lower() for keyword in ['work from home', 'job', 'employment', 'mystery shopper', 'remote'])],
-                    inputs=[input_text],
-                    outputs=[classification_output, confidence_output, analysis_output, time_output],
-                    fn=analyze_phishing,
-                    cache_examples=False
-                )
-            
-            with gr.Accordion("✅ Legitimate Content Examples", open=False):
-                gr.Examples(
-                    examples=[ex for ex in examples if any(keyword in ex[0].lower() for keyword in ['thank you', 'receipt', 'appointment', 'order confirmation', 'welcome'])],
-                    inputs=[input_text],
-                    outputs=[classification_output, confidence_output, analysis_output, time_output],
-                    fn=analyze_phishing,
-                    cache_examples=False
+                analysis_output = gr.Textbox(
+                    label="🔬 Detailed AI Analysis",
+                    lines=10,
+                    max_lines=20,
+                    interactive=False,
+                    placeholder="Detailed analysis will appear here..."
                 )
+                
+                # Enhanced Examples section with categories
+                gr.Markdown("### 📚 Comprehensive Test Examples", elem_classes=["container"])
+                gr.Markdown("Try these diverse examples to explore the AI's detection capabilities:", elem_classes=["container"])
+                
+                with gr.Accordion("🏦 Banking & Finance", open=False):
+                    gr.Examples(
+                        examples=[ex for ex in examples if any(keyword in ex[0].lower() for keyword in ['paypal', 'bank', 'chase', 'credit', 'wellsfargo', 'visa'])],
+                        inputs=[input_text],
+                        outputs=[classification_output, confidence_output, analysis_output, time_output],
+                        fn=analyze_phishing,
+                        cache_examples=False
+                    )
+                
+                with gr.Accordion("🛒 E-commerce & Shopping", open=False):
+                    gr.Examples(
+                        examples=[ex for ex in examples if any(keyword in ex[0].lower() for keyword in ['amazon', 'ebay', 'apple', 'microsoft', 'netflix'])],
+                        inputs=[input_text],
+                        outputs=[classification_output, confidence_output, analysis_output, time_output],
+                        fn=analyze_phishing,
+                        cache_examples=False
+                    )
+                
+                with gr.Accordion("📧 Email Scams", open=False):
+                    gr.Examples(
+                        examples=[ex for ex in examples if len(ex[0]) > 100 and any(keyword in ex[0].lower() for keyword in ['urgent', 'congratulations', 'won', 'grant', 'lottery'])],
+                        inputs=[input_text],
+                        outputs=[classification_output, confidence_output, analysis_output, time_output],
+                        fn=analyze_phishing,
+                        cache_examples=False
+                    )
+                
+                with gr.Accordion("📱 SMS & Text Messages", open=False):
+                    gr.Examples(
+                        examples=[ex for ex in examples if any(keyword in ex[0].lower() for keyword in ['alert', 'package', 'verification', 'expires', 'code'])],
+                        inputs=[input_text],
+                        outputs=[classification_output, confidence_output, analysis_output, time_output],
+                        fn=analyze_phishing,
+                        cache_examples=False
+                    )
+                
+                with gr.Accordion("💻 Tech Support Scams", open=False):
+                    gr.Examples(
+                        examples=[ex for ex in examples if any(keyword in ex[0].lower() for keyword in ['virus', 'infected', 'security warning', 'update required', 'antivirus'])],
+                        inputs=[input_text],
+                        outputs=[classification_output, confidence_output, analysis_output, time_output],
+                        fn=analyze_phishing,
+                        cache_examples=False
+                    )
+                
+                with gr.Accordion("💰 Investment & Crypto Scams", open=False):
+                    gr.Examples(
+                        examples=[ex for ex in examples if any(keyword in ex[0].lower() for keyword in ['bitcoin', 'crypto', 'investment', 'trading', 'returns'])],
+                        inputs=[input_text],
+                        outputs=[classification_output, confidence_output, analysis_output, time_output],
+                        fn=analyze_phishing,
+                        cache_examples=False
+                    )
+                
+                with gr.Accordion("💼 Job & Employment Scams", open=False):
+                    gr.Examples(
+                        examples=[ex for ex in examples if any(keyword in ex[0].lower() for keyword in ['work from home', 'job', 'employment', 'mystery shopper', 'remote'])],
+                        inputs=[input_text],
+                        outputs=[classification_output, confidence_output, analysis_output, time_output],
+                        fn=analyze_phishing,
+                        cache_examples=False
+                    )
+                
+                with gr.Accordion("✅ Legitimate Content Examples", open=False):
+                    gr.Examples(
+                        examples=[ex for ex in examples if any(keyword in ex[0].lower() for keyword in ['thank you', 'receipt', 'appointment', 'order confirmation', 'welcome'])],
+                        inputs=[input_text],
+                        outputs=[classification_output, confidence_output, analysis_output, time_output],
+                        fn=analyze_phishing,
+                        cache_examples=False
+                    )
         
         # Batch Analysis Tab
         with gr.TabItem("📊 Batch Analysis"):
-            gr.Markdown("### 📦 Analyze Multiple Items at Once")
-            gr.Markdown("Upload a text file with one URL, email, or content per line for bulk analysis")
-            
-            with gr.Row():
-                with gr.Column():
-                    file_input = gr.File(
-                        label="📁 Upload Text File (.txt)",
-                        file_types=[".txt"],
-                        type="text"
-                    )
-                    
-                    batch_btn = gr.Button("📊 Analyze Batch", variant="primary", size="lg")
-                    
-                    gr.Markdown("""
-                    **📋 File Format:**
-                    - One item per line
-                    - Supports URLs, emails, SMS content
-                    - Maximum 100 items per batch
-                    - Plain text format (.txt)
-                    """)
-            
-            batch_output = gr.Markdown(label="📈 Batch Analysis Results")
+            with gr.Column(elem_classes=["main-content"]):
+                gr.Markdown("### 📦 Analyze Multiple Items at Once", elem_classes=["container"])
+                gr.Markdown("Upload a text file with one URL, email, or content per line for bulk analysis", elem_classes=["container"])
+                
+                with gr.Row():
+                    with gr.Column():
+                        file_input = gr.File(
+                            label="📁 Upload Text File (.txt)",
+                            file_types=[".txt"],
+                            type="filepath"
+                        )
+                        
+                        batch_btn = gr.Button("📊 Analyze Batch", variant="primary", size="lg")
+                        
+                        gr.Markdown("""
+                        **📋 File Format:**
+                        - One item per line
+                        - Supports URLs, emails, SMS content
+                        - Maximum 100 items per batch
+                        - Plain text format (.txt)
+                        """, elem_classes=["container"])
+                
+                batch_output = gr.Markdown(label="📈 Batch Analysis Results")
         
         # Real-time Monitoring Tab
         with gr.TabItem("⚡ Quick Test"):
-            gr.Markdown("### 🚀 Quick Phishing Detection Test")
-            gr.Markdown("Instantly test common phishing scenarios with pre-loaded examples")
-            
-            with gr.Row():
-                with gr.Column():
-                    gr.Markdown("#### 🚨 Suspicious Content")
-                    suspicious_examples = [
-                        "Urgent: Your account will be suspended in 24 hours! Verify now: secure-verification.fake-bank.com",
-                        "Congratulations! You've won $10,000! Claim immediately: lottery-winner.scam-site.org",
-                        "Apple ID locked due to suspicious activity. Unlock now: apple-security.phishing-domain.net"
-                    ]
-                    
-                    for i, example in enumerate(suspicious_examples):
-                        if gr.Button(f"Test Suspicious #{i+1}", variant="stop"):
-                            input_text.value = example
+            with gr.Column(elem_classes=["main-content"]):
+                gr.Markdown("### 🚀 Quick Phishing Detection Test", elem_classes=["container"])
+                gr.Markdown("Instantly test common phishing scenarios with pre-loaded examples", elem_classes=["container"])
                 
-                with gr.Column():
-                    gr.Markdown("#### ✅ Legitimate Content")
-                    legitimate_examples = [
-                        "Your monthly statement is ready for download on our secure banking portal.",
-                        "Thank you for your purchase. Your order will ship within 2-3 business days.",
-                        "Appointment reminder: Your doctor's appointment is scheduled for tomorrow at 2 PM."
-                    ]
+                with gr.Row():
+                    with gr.Column():
+                        gr.Markdown("#### 🚨 Test Suspicious Content", elem_classes=["container"])
+                        suspicious_btn1 = gr.Button("🚨 Test: Fake Bank Alert", variant="stop")
+                        suspicious_btn2 = gr.Button("🚨 Test: Lottery Scam", variant="stop")
+                        suspicious_btn3 = gr.Button("🚨 Test: Apple ID Phishing", variant="stop")
                     
-                    for i, example in enumerate(legitimate_examples):
-                        if gr.Button(f"Test Legitimate #{i+1}", variant="primary"):
-                            input_text.value = example
+                    with gr.Column():
+                        gr.Markdown("#### ✅ Test Legitimate Content", elem_classes=["container"])
+                        legitimate_btn1 = gr.Button("✅ Test: Bank Statement", variant="primary")
+                        legitimate_btn2 = gr.Button("✅ Test: Order Confirmation", variant="primary")
+                        legitimate_btn3 = gr.Button("✅ Test: Appointment Reminder", variant="primary")
+                
+                with gr.Row():
+                    with gr.Column(scale=2):
+                        quick_input = gr.Textbox(
+                            label="📝 Quick Test Content",
+                            placeholder="Content from quick test buttons will appear here...",
+                            lines=3
+                        )
+                        
+                        quick_analyze_btn = gr.Button("🔍 Analyze Quick Test", variant="primary", size="lg")
+                
+                with gr.Row():
+                    with gr.Column():
+                        quick_classification = gr.Textbox(label="🎯 Classification", interactive=False)
+                        quick_confidence = gr.Textbox(label="📊 Confidence", interactive=False)
+                        quick_time = gr.Textbox(label="⚡ Time", interactive=False)
+                
+                quick_analysis = gr.Textbox(
+                    label="🔬 Quick Analysis Results",
+                    lines=8,
+                    interactive=False,
+                    placeholder="Analysis results will appear here..."
+                )
         
         # Statistics & Insights Tab
         with gr.TabItem("📈 Insights"):
@@ -679,14 +741,52 @@ with gr.Blocks(
         inputs=[file_input],
         outputs=[batch_output]
     )
+    
+    # Quick Test tab event handlers
+    suspicious_btn1.click(
+        fn=set_suspicious_1,
+        inputs=[],
+        outputs=quick_input
+    )
+    
+    suspicious_btn2.click(
+        fn=set_suspicious_2,
+        inputs=[],
+        outputs=quick_input
+    )
+    
+    suspicious_btn3.click(
+        fn=set_suspicious_3,
+        inputs=[],
+        outputs=quick_input
+    )
+    
+    legitimate_btn1.click(
+        fn=set_legitimate_1,
+        inputs=[],
+        outputs=quick_input
+    )
+    
+    legitimate_btn2.click(
+        fn=set_legitimate_2,
+        inputs=[],
+        outputs=quick_input
+    )
+    
+    legitimate_btn3.click(
+        fn=set_legitimate_3,
+        inputs=[],
+        outputs=quick_input
+    )
+    
+    quick_analyze_btn.click(
+        fn=analyze_phishing,
+        inputs=[quick_input],
+        outputs=[quick_classification, quick_confidence, quick_analysis, quick_time]
+    )
 
 # Launch the app
 if __name__ == "__main__":
     app.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        show_error=True,
-        share=False,
-        favicon_path=None,
-        show_tips=True
+        share=True
     )