Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 import os
 import requests
-import base64
 
 # Environment variables
 SENDGRID_KEY = os.environ["SENDGRID_API_KEY"]
@@ -10,7 +9,6 @@ FROM_EMAIL = "epicamp.response@gmail.com"
 def send_report(to_email, pdf_b64, label, confidence, sequence):
     html_content = f"""
     <div style="font-family: 'Segoe UI', Arial, sans-serif; max-width:600px; margin:auto; color:#1a2540; line-height:1.6">
-
       <!-- Header -->
       <div style="background:#00285a; padding:22px; border-radius:10px 10px 0 0;">
         <h2 style="color:#ffffff; margin:0">EPIC-AMP Analysis Report</h2>
@@ -18,59 +16,60 @@ def send_report(to_email, pdf_b64, label, confidence, sequence):
           Explainable Antimicrobial Peptide Classification Platform
         </p>
       </div>
-
       <!-- Body -->
       <div style="border:1px solid #d0daea; border-top:none; padding:24px; border-radius:0 0 10px 10px">
-
-        <p>Dear User,</p>
-
+        <p>Dear Researcher,</p>
         <p>
           Your peptide sequence has been successfully analyzed using the <strong>EPIC-AMP platform</strong>.
-          Please find your detailed report attached as a PDF file.
+          Please find your detailed PDF report attached to this email.
         </p>
-
-        <p>Below is a summary of your analysis:</p>
-
-        <!-- Table -->
-        <table style="width:100%; border-collapse:collapse; margin:18px 0; font-size:14px">
+        <p>Below is a summary of your analysis results:</p>
+        <!-- Results Table -->
+        <table style="width:100%; border-collapse:collapse; margin:18px 0; font-size:14px; border-radius:6px; overflow:hidden">
+          <tr style="background:#00285a; color:#ffffff">
+            <td style="padding:10px 14px; font-weight:bold; width:40%">Metric</td>
+            <td style="padding:10px 14px; font-weight:bold">Result</td>
+          </tr>
           <tr style="background:#f0f4fa">
-            <td style="padding:10px; font-weight:bold; width:40%">Classification</td>
-            <td style="padding:10px">{label}</td>
+            <td style="padding:10px 14px; font-weight:bold">Classification</td>
+            <td style="padding:10px 14px; font-weight:bold; color:{'#2e7d32' if 'AMP' in label and 'Non' not in label else '#c62828'}">{label}</td>
           </tr>
           <tr>
-            <td style="padding:10px; font-weight:bold">Confidence Score</td>
-            <td style="padding:10px">{confidence}</td>
+            <td style="padding:10px 14px; font-weight:bold">Confidence Score</td>
+            <td style="padding:10px 14px">{confidence}</td>
           </tr>
           <tr style="background:#f0f4fa">
-            <td style="padding:10px; font-weight:bold">Input Sequence</td>
-            <td style="padding:10px; font-family:monospace; font-size:12px; word-break:break-all">
-              {sequence}
-            </td>
+            <td style="padding:10px 14px; font-weight:bold">Input Sequence</td>
+            <td style="padding:10px 14px; font-family:monospace; font-size:12px; word-break:break-all; color:#003f7d">{sequence}</td>
           </tr>
         </table>
-
         <p>
-          The attached report provides a comprehensive explanation of the prediction,
-          including interpretability insights to support further research and analysis.
+          The attached PDF report includes a full breakdown of the prediction, LIME feature attributions,
+          MIC predictions (if applicable), and global SHAP feature importance — providing complete
+          interpretability to support your research.
         </p>
-
         <p>
-          If you have any questions or need assistance, feel free to contact us.
+          If you have any questions, suggestions, or need further assistance, please don't hesitate
+          to reach out to us at
+          <a href="mailto:epicamp.sup@gmail.com" style="color:#1a6fc4">epicamp.sup@gmail.com</a>.
         </p>
-
-        <p style="margin-top:24px">
+        <p style="margin-top:20px">
           Best regards,<br>
-          <strong>EPIC-AMP Team</strong>
+          <strong>The EPIC-AMP Team</strong><br>
+          <span style="color:#666; font-size:13px">Bioinformatics &amp; Computational Biology Unit (BCBU)</span>
         </p>
-
         <!-- Footer -->
         <hr style="border:none; border-top:1px solid #e0e6f0; margin:24px 0">
-
-        <p style="color:#666; font-size:12px">
-          Zewail City of Science and Technology · Biotechnology & Bioinformatics Unit (BCBU)<br>
-          📧 <a href="mailto:epicamp.sup@gmail.com" style="color:#1a6fc4">epicamp.sup@gmail.com</a>
-        </p>
-
+        <div style="display:flex; justify-content:space-between; align-items:center; flex-wrap:wrap; gap:8px">
+          <p style="color:#888; font-size:11px; margin:0">
+            Zewail City of Science and Technology · BCBU<br>
+            Ahmed Zewail Street, October Gardens, Giza, Egypt
+          </p>
+          <p style="color:#888; font-size:11px; margin:0; text-align:right">
+            📧 <a href="mailto:epicamp.sup@gmail.com" style="color:#1a6fc4">epicamp.sup@gmail.com</a><br>
+            <span style="color:#bbb">This is an automated message — please do not reply directly.</span>
+          </p>
+        </div>
       </div>
     </div>
     """
@@ -95,20 +94,24 @@ def send_report(to_email, pdf_b64, label, confidence, sequence):
         },
         timeout=20
     )
-
     return "sent" if resp.status_code == 202 else f"error:{resp.status_code}:{resp.text}"
 
 
-# Gradio interface
-gr.Interface(
-    fn=send_report,
-    inputs=[
-        gr.Text(label="Recipient Email"),
-        gr.Text(label="PDF Base64"),
-        gr.Text(label="Label"),
-        gr.Text(label="Confidence"),
-        gr.Text(label="Sequence")
-    ],
-    outputs=gr.Text(label="Status"),
-    title="EPIC-AMP Mailer"
-).launch()
+# ── Gradio Blocks (fixes "Could not get API info" error) ──────────────
+with gr.Blocks() as demo:
+    with gr.Row():
+        to_email   = gr.Text(label="Recipient Email")
+        pdf_b64    = gr.Text(label="PDF Base64")
+        label_in   = gr.Text(label="Label")
+        confidence = gr.Text(label="Confidence")
+        sequence   = gr.Text(label="Sequence")
+    result = gr.Text(label="Status")
+    btn = gr.Button("Send")
+    btn.click(
+        fn=send_report,
+        inputs=[to_email, pdf_b64, label_in, confidence, sequence],
+        outputs=result,
+        api_name="send"          # ← makes /send the stable API endpoint
+    )
+
+demo.launch(ssr_mode=False)      # ← fixes the SSR startup warning