Update app.py

app.py

@@ -1,29 +1,41 @@
 import gradio as gr
-from analyze_email_main import analyze_email  # Import main analysis function
+import tempfile
+from analyze_email_main import analyze_email
 
-def analyze_eml_file(file):
+def analyze_uploaded_email(file_obj):
     try:
-        with open(file.name, "rb") as f:
-            raw_email = f.read()
-        results = analyze_email(raw_email)
+        # Write uploaded file to a temporary .eml file
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".eml") as tmp:
+            tmp.write(file_obj.read())
+            tmp_path = tmp.name
 
-        # Format results for display
-        if isinstance(results, list):
-            output = "\n".join([f"{r['category']}: {r['description']}" for r in results])
+        results = analyze_email(tmp_path)
+
+        if isinstance(results, dict):
+            # Format dict output nicely
+            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:
-            output = str(results)
+            return str(results)
 
-        return output
     except Exception as e:
         return f"❌ Error analyzing email: {e}"
 
-demo = gr.Interface(
-    fn=analyze_eml_file,
-    inputs=gr.File(file_types=[".eml"], label="Upload an Email (.eml)"),
+iface = gr.Interface(
+    fn=analyze_uploaded_email,
+    inputs=gr.File(type="file", label="Upload .eml file"),
     outputs="text",
-    title="📧 Email Guardians - Free Email Analyzer",
-    description="Upload a .eml file to analyze headers, body, and URLs for phishing, spam, spoofing, and malicious content."
+    title="Email Analysis Tool",
+    description="Upload a .eml file to analyze headers, body, and URLs."
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    iface.launch()