Update app.py

app.py

@@ -2,14 +2,11 @@ import gradio as gr
 from groq import Groq
 import os
 
-# ⚠️ IMPORTANT: Set your Groq API key in Hugging Face Spaces "Secrets"
-# os.environ["GROQ_API_KEY"] = "your_api_key_here"
-
 client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
 def analyze_email(email_text):
     if not email_text.strip():
-        return "⚠️ Please paste an email to analyze.", "", ""
+        return "⚠️ **Please paste an email to analyze.**", "", ""
 
     # Step 1: Detect if phishing
     detection_prompt = f"""
@@ -45,7 +42,12 @@ def analyze_email(email_text):
 
     rewritten_email = rewrite_response.choices[0].message.content.strip()
 
-    return email_text, classification_result, rewritten_email
+    # Return formatted Markdown
+    original_md = f"### ✉️ Original Email\n\n```\n{email_text}\n```"
+    classification_md = f"### 🛡️ Detection Result\n\n{classification_result}"
+    rewritten_md = f"### ✅ Safe Rewritten Email\n\n{rewritten_email}"
+
+    return original_md, classification_md, rewritten_md
 
 
 # 🚀 Gradio UI
@@ -63,9 +65,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     analyze_btn = gr.Button("🔍 Analyze Email")
 
     with gr.Row():
-        original_out = gr.Textbox(label="Original Email", lines=12)
-        classification_out = gr.Textbox(label="Phishing Detection Result", lines=4)
-        rewritten_out = gr.Textbox(label="Safe Rewritten Email", lines=12)
+        original_out = gr.Markdown()
+        classification_out = gr.Markdown()
+        rewritten_out = gr.Markdown()
 
     analyze_btn.click(
         analyze_email,