Update app.py

app.py

@@ -1,127 +1,118 @@
-# app.py (Gradio GPT Newsletter with Mautic Integration)
 import gradio as gr
-import openai
 import os
-import requests
 import re
+import requests
 from dotenv import load_dotenv
 from openai import OpenAI
 
-# Load secrets from .env
 load_dotenv()
-openai.api_key = os.getenv("OPENAI_API_KEY")
+
+# Environment variables
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 MAUTIC_BASE_URL = os.getenv("MAUTIC_BASE_URL")
 MAUTIC_USERNAME = os.getenv("MAUTIC_USERNAME")
 MAUTIC_PASSWORD = os.getenv("MAUTIC_PASSWORD")
 
+client = OpenAI(api_key=OPENAI_API_KEY)
+
+SECTORS = [
+    "Law", "HR", "Marketing", "Finance", "Healthcare", "Education", "Retail",
+    "Manufacturing", "Real Estate", "Construction", "Transport", "Hospitality",
+    "Media", "Entertainment", "Telecom", "Government", "Energy", "Insurance",
+    "Agriculture", "Technology"
+]
+
 PROMPTS = {
-    "Law": "Generate a UK cybersecurity insight for professionals in the legal sector.",
-    "HR": "Generate a UK cybersecurity update for HR professionals.",
-    "Marketing": "Generate a UK cybersecurity tip for marketing professionals.",
-    "General": "Generate a brief UK-relevant cybersecurity insight for business users."
+    sector: f"Generate a UK cybersecurity insight for the {sector} sector." for sector in SECTORS
 }
 
+# Spam moderation
+SPAM_KEYWORDS = ["free money", "urgent click", "buy now", "no cost"]
 def moderate_content(text):
-    spam_keywords = ["free money", "urgent click", "buy now", "no cost"]
-    for keyword in spam_keywords:
-        if re.search(keyword, text, re.IGNORECASE):
-            return False
-    return True
+    return not any(re.search(kw, text, re.IGNORECASE) for kw in SPAM_KEYWORDS)
 
-def send_to_mautic(name, email):
+def send_to_mautic(name, email, content):
     try:
-        url = f"{MAUTIC_BASE_URL}/api/contacts/new"
-        data = {
+        contact_data = {
             "firstname": name,
             "email": email,
-            "tags": "GPT Generated"
+            "tags": "A",
+            "custom_fields[ai_insight]": content,
+            "segment[]": ["ai-contacts"]
         }
-        response = requests.post(url, auth=(MAUTIC_USERNAME, MAUTIC_PASSWORD), data=data)
-        return response.status_code == 200
+        response = requests.post(
+            f"{MAUTIC_BASE_URL}/api/contacts/new",
+            auth=(MAUTIC_USERNAME, MAUTIC_PASSWORD),
+            data=contact_data
+        )
+        if response.status_code in [200, 201]:
+            return True, "Contact synced and added to Mautic."
+        else:
+            return False, f"Mautic error: {response.text}"
     except Exception as e:
-        return False
+        return False, f"Error syncing to Mautic: {str(e)}"
 
-
-client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 def generate_insight(name, email, sector, tone, temperature):
-    base_prompt = PROMPTS.get(sector, PROMPTS["General"])
-    tone_injection = f" The tone should be {tone.lower()} and professional."
-    full_prompt = base_prompt + tone_injection
+    base_prompt = PROMPTS.get(sector, PROMPTS["Technology"])
+    tone_prompt = f" Make the tone {tone.lower()} and professional."
+    full_prompt = base_prompt + tone_prompt
 
     try:
         response = client.chat.completions.create(
             model="gpt-4",
             messages=[
-                {"role": "system", "content": "You are a UK-based cybersecurity assistant."},
+                {"role": "system", "content": "You are a UK cybersecurity newsletter assistant."},
                 {"role": "user", "content": full_prompt}
             ],
             temperature=temperature,
             max_tokens=300
         )
-
-        # This is the correct way to access the message content
         insight = response.choices[0].message.content.strip()
 
         if not moderate_content(insight):
-            return "<p style='color:red;'>Insight flagged as potentially spammy. Please regenerate or adjust tone/temperature.</p>"
-
-        send_to_mautic(name, email)
-        
-        html_template = f"""
-        <html>
-        <body style='font-family:Arial; color:#333;'>
-          <div style='max-width:600px; margin:auto; background:#f9f9f9; padding:20px; border-radius:8px;'>
-            <h2 style='background:#004080; color:white; padding:10px;'>Cyber Insight Briefing - {sector}</h2>
-            <p>Hello {name},</p>
-            <p>Here are the latest cybersecurity insights tailored for your sector ({sector}) with a {tone.lower()} tone:</p>
-            <p><strong>AI Insight:</strong><br>{insight}</p>
-            <p>Stay secure,<br>CyberIntel UK</p>
-            <hr>
-            <p style='font-size:12px;'>You are receiving this because you subscribed to CyberIntel. <a href='#'>Unsubscribe</a> | <a href='#'>Manage Preferences</a></p>
-          </div>
-        </body>
-        </html>
-        """
-
-        return html_template
-    except Exception as e:
-        return f"<p style='color:red;'>Error: {str(e)}</p>"
-
-with gr.Blocks() as demo:
-    with gr.Tab("Home"):
-        gr.Markdown("""
-        # UK Cybersecurity Newsletter Demo
-        Use this tool to generate AI-powered, sector-specific insights with compliance filters, tone control, and automatic Mautic contact creation.
-        """)
-
-    with gr.Tab("Generate Newsletter"):
-        with gr.Row():
-            name = gr.Textbox(label="Your Name", placeholder="e.g., Alice")
-            email = gr.Textbox(label="Your Email", placeholder="alice@example.com")
-
-        sector = gr.Dropdown(label="Choose Sector", choices=["Law", "HR", "Marketing"], value="Law")
-        tone = gr.Radio(label="Tone", choices=["Formal", "Friendly", "Urgent", "Compliant"], value="Formal")
-        temperature = gr.Slider(0.0, 1.0, value=0.7, step=0.1, label="Creativity (Temperature)")
-
-        generate_btn = gr.Button("Generate Email Insight")
-        email_preview = gr.HTML(label="HTML Email Preview")
-
-        generate_btn.click(
-            generate_insight,
-            inputs=[name, email, sector, tone, temperature],
-            outputs=email_preview
-        )
-
-    with gr.Tab("About"):
-        gr.Markdown("""
-        This demo integrates GPT-4 with tone and moderation control to generate UK-sector-focused cybersecurity insights.
-        Includes Mautic contact sync.
-        """)
-
-if __name__ == "__main__":
-    demo.launch()
-
+            return "", "Content flagged as spammy. Please retry."
 
+        return insight, "AI-generated content ready. You can edit before sending."
 
+    except Exception as e:
+        return "", f"OpenAI error: {str(e)}"
+
+# Gradio UI
+def build_interface():
+    with gr.Blocks() as demo:
+        with gr.Tab("Generate Insight"):
+            name = gr.Textbox(label="Name")
+            email = gr.Textbox(label="Email")
+            sector = gr.Dropdown(SECTORS, label="Sector", value="Law")
+            tone = gr.Radio(["Formal", "Friendly", "Urgent", "Compliant"], label="Tone", value="Formal")
+            temperature = gr.Slider(0.1, 1.0, value=0.7, step=0.1, label="Creativity")
+            generate_btn = gr.Button("Generate Insight")
+            status = gr.Textbox(label="Status", interactive=False)
+            raw_output = gr.Textbox(label="Generated Content", lines=10)
+
+            generate_btn.click(fn=generate_insight,
+                               inputs=[name, email, sector, tone, temperature],
+                               outputs=[raw_output, status])
+
+        with gr.Tab("Preview & Send"):
+            edited = gr.Textbox(label="Editable Email Content", lines=10)
+            send_btn = gr.Button("Send via Mautic")
+            send_status = gr.Textbox(label="Mautic Send Log")
+
+            send_btn.click(fn=send_to_mautic,
+                           inputs=[name, email, edited],
+                           outputs=[None, send_status])
+
+        with gr.Tab("About"):
+            gr.Markdown("""
+            ### AI Cybersecurity Newsletter Demo
+            This app uses OpenAI GPT to generate sector-specific insights, allows editing, and integrates with Mautic.
+
+            Contacts are added to the `ai-contacts` segment and emailed through `Campaign A` using the `Email Test` template.
+            """)
+
+    return demo
 
+if __name__ == "__main__":
+    build_interface().launch()