Update app.py

app.py

@@ -1,34 +1,90 @@
 import gradio as gr
-from analyze_email_main import analyze_email
+from analyze_email_main import analyze
 
-def analyze_uploaded_email(file_path):
+def analyze_email(file_obj):
+    if file_obj is None:
+        return "<p style='color:red;'>Please upload a .eml file to analyze.</p>"
+
+    results = analyze(file_obj.name)
+
+    # Basic verdict info
+    attack_score = results[0]  # e.g., "Attack Score: 85"
+    attack_type = results[1]   # e.g., "Attack Type: Credential Harvesting"
+    verdict = results[2]       # e.g., "Final Verdict: Malicious"
+
+    score_val = int("".join(filter(str.isdigit, attack_score)))  # extract numeric score
+    if score_val > 100:  
+        score_val = 100
+
+    # Assign verdict color
+    if "Malicious" in verdict:
+        verdict_color = "#e74c3c"  # red
+    elif "Suspicious" in verdict:
+        verdict_color = "#e67e22"  # orange
+    elif "Spam" in verdict:
+        verdict_color = "#f1c40f"  # yellow
+    else:
+        verdict_color = "#2ecc71"  # green
+
+    # Extract report sections
     try:
-        results = analyze_email(file_path)
-
-        if isinstance(results, dict):
-            output = []
-            for section, findings in results.items():
-                output.append(f"--- {section.upper()} ---")
-                if findings:
-                    output.extend(findings)
-                else:
-                    output.append("No issues found")
-            return "\n".join(output)
-        elif isinstance(results, list):
-            return "\n".join(results)
-        else:
-            return str(results)
-
-    except Exception as e:
-        return f"❌ Error analyzing email: {e}"
-
-iface = gr.Interface(
-    fn=analyze_uploaded_email,
-    inputs=gr.File(type="filepath", label="Upload .eml file"),
-    outputs="text",
-    title="Email Analysis Tool",
-    description="Upload a .eml file to analyze headers, body, and URLs."
+        tags_index = results.index("---- Attack Analysis Tags ----")
+        findings_index = results.index("---- Detailed Findings ----")
+        body_index = results.index("---- Highlighted Body ----")
+    except ValueError:
+        return "<pre>" + "\n".join(results) + "</pre>"
+
+    tags = results[tags_index + 1 : findings_index]
+    findings = results[findings_index + 1 : body_index]
+    highlighted_body = results[body_index + 1 :]
+
+    # Styled HTML Report
+    html = f"""
+    <div style='font-family: Arial, sans-serif; padding: 20px; max-width: 800px; margin:auto;'>
+
+      <h2 style="color:#2c3e50;">📧 Email Security Report</h2>
+
+      <h3>Attack Score</h3>
+      <div style="background:#ecf0f1; border-radius:10px; overflow:hidden; height:24px; width:100%;">
+        <div style="width:{score_val}%; background:#3498db; height:100%; text-align:center; color:white; line-height:24px;">
+          {score_val}/100
+        </div>
+      </div>
+
+      <p><b>Attack Type:</b> {attack_type}</p>
+      <p><b>Final Verdict:</b> <span style="background:{verdict_color}; color:white; padding:4px 10px; border-radius:6px;">
+        {verdict.replace("Final Verdict:", "").strip()}
+      </span></p>
+
+      <h3>🔖 Attack Analysis Tags</h3>
+      <ul>
+        {''.join(f'<li>{tag}</li>' for tag in tags)}
+      </ul>
+
+      <h3>🕵️ Detailed Findings</h3>
+      <ul>
+        {''.join(f'<li>{finding}</li>' for finding in findings)}
+      </ul>
+
+      <h3>📩 Highlighted Email Body</h3>
+      <div style='border:1px solid #ccc; padding:15px; background:#fdfdfd; border-radius:8px;'>
+        {"<br>".join(highlighted_body)}
+      </div>
+
+    </div>
+    """
+
+    return html
+
+# Gradio Interface
+demo = gr.Interface(
+    fn=analyze_email,
+    inputs=gr.File(label="Upload .eml File", type="file"),
+    outputs=gr.HTML(label="Analysis Report"),
+    title="📧 Email Security Analyzer",
+    description="Upload an .eml file to detect phishing, spam, or malicious emails. "
+                "This analyzer combines AI, heuristics, and threat intelligence feeds."
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    demo.launch()