Add application files

app.py

@@ -1341,16 +1341,13 @@ def chat_with_ai(message: str, history: list) -> tuple:
                 response += "To find contacts, go to **Prospects Tab** and run **Find Prospects** again."
             return history + [[message, response]], ""
 
-    # Handle "draft email" for a known prospect
-    if any(kw in msg_lower for kw in ["draft", "write", "compose", "email to", "create email"]):
+    # Handle "show email" - just viewing existing drafts
+    if any(kw in msg_lower for kw in ["show email", "existing email", "what email", "see email", "view email"]):
         if mentioned_prospect:
             p_name = mentioned_prospect["name"]
             p_name_lower = p_name.lower()
-
-            # Check if we already have an email for this prospect
             existing_emails = [e for e in knowledge_base["emails"]
                               if p_name_lower in e.get("prospect_company", "").lower()]
-
             if existing_emails:
                 email = existing_emails[0]
                 response = f"## ✉️ Existing Email Draft for {p_name}\n\n"
@@ -1359,25 +1356,100 @@ def chat_with_ai(message: str, history: list) -> tuple:
                 response += f"---\n\n{email.get('body', 'No content')}\n\n"
                 response += "---\n\n*This email was drafted during prospect discovery.*"
             else:
-                # Generate a template based on prospect info
-                contacts = [c for c in knowledge_base["contacts"]
-                           if p_name_lower in c.get("company", "").lower()]
-                contact = contacts[0] if contacts else None
+                response = f"No existing email drafts found for **{p_name}**."
+            return history + [[message, response]], ""
 
-                to_email = contact.get("email", "contact@" + p_name.lower().replace(" ", "") + ".com") if contact else f"contact@{p_name.lower().replace(' ', '')}.com"
-                contact_name = contact.get("name", "there").split()[0] if contact else "there"
+    # Handle "draft/write/compose email" - create custom email based on user's request
+    if any(kw in msg_lower for kw in ["draft", "write", "compose", "create email", "email to", "send email", "mail to"]):
+        if mentioned_prospect:
+            p_name = mentioned_prospect["name"]
+            p_name_lower = p_name.lower()
 
-                response = f"## ✉️ Email Template for {p_name}\n\n"
-                response += f"**To:** {to_email}\n"
-                response += f"**Subject:** {client_name} + {p_name}: A Potential Partnership\n\n"
+            # Get contact info
+            contacts = [c for c in knowledge_base["contacts"]
+                       if p_name_lower in c.get("company", "").lower()
+                       or c.get("company", "").lower() in p_name_lower]
+            contact = contacts[0] if contacts else None
+            to_email = contact.get("email", f"contact@{p_name.lower().replace(' ', '')}.com") if contact else f"contact@{p_name.lower().replace(' ', '')}.com"
+            contact_name = contact.get("name", "").split()[0] if contact and contact.get("name") else "there"
+            contact_title = contact.get("title", "") if contact else ""
+
+            # Extract specific details from user's message
+            import re
+
+            # Check if this is a meeting request
+            is_meeting_request = any(kw in msg_lower for kw in ["meeting", "call", "demo", "schedule", "appointment"])
+
+            # Extract date/time info
+            date_match = re.search(r'(\d{1,2}(?:st|nd|rd|th)?\s+(?:jan|feb|mar|apr|may|jun|jul|aug|sep|oct|nov|dec)[a-z]*\s+\d{4}|\w+day(?:\s+next\s+week)?|\d{1,2}[/-]\d{1,2}[/-]\d{2,4})', msg_lower)
+            time_match = re.search(r'(\d{1,2}:\d{2}|\d{1,2}\s*(?:am|pm))', msg_lower)
+            duration_match = re.search(r'(\d+)\s*(?:min|minute|hour)', msg_lower)
+
+            date_str = date_match.group(1).title() if date_match else ""
+            time_str = time_match.group(1) if time_match else ""
+            duration_str = duration_match.group(0) if duration_match else ""
+
+            # Extract the purpose/topic from the message
+            # Remove common words to find the custom content
+            custom_content = message
+            for word in ["draft", "write", "compose", "email", "mail", "to", p_name.lower(), "asking", "that", "can", "we", "a", "an", "the", "for", "about"]:
+                custom_content = re.sub(rf'\b{word}\b', '', custom_content, flags=re.IGNORECASE)
+            custom_content = ' '.join(custom_content.split()).strip()
+
+            # Generate custom email based on context
+            response = f"## ✉️ Custom Email Draft for {p_name}\n\n"
+            response += f"**To:** {to_email}\n"
+
+            if is_meeting_request:
+                # Meeting request email
+                subject = f"Meeting Request: {client_name} x {p_name}"
+                if date_str:
+                    subject = f"Meeting Request for {date_str} - {client_name} x {p_name}"
+                response += f"**Subject:** {subject}\n\n"
+                response += f"---\n\n"
+                response += f"Dear {contact_name},\n\n"
+                response += f"I hope this email finds you well.\n\n"
+                response += f"I'm reaching out from {client_name} regarding a potential collaboration with {p_name}. "
+                response += f"Based on our research, we believe there's a strong synergy between our companies, "
+                response += f"particularly in the {mentioned_prospect.get('industry', 'your industry')} space.\n\n"
+
+                if date_str or time_str or duration_str:
+                    response += f"I would like to propose a meeting"
+                    if date_str:
+                        response += f" on **{date_str}**"
+                    if time_str:
+                        response += f" at **{time_str}**"
+                    if duration_str:
+                        response += f" for **{duration_str}**"
+                    response += f" to discuss how {client_name} can help {p_name} achieve its goals.\n\n"
+                else:
+                    response += f"Would you be available for a brief call this week to discuss how {client_name} can support {p_name}'s growth?\n\n"
+
+                response += f"During our conversation, I'd love to explore:\n"
+                response += f"- How {client_name}'s solutions align with {p_name}'s current initiatives\n"
+                response += f"- Specific ways we can add value to your {mentioned_prospect.get('industry', 'business')}\n"
+                response += f"- Next steps for a potential partnership\n\n"
+                response += f"Please let me know if this time works for you, or suggest an alternative that fits your schedule.\n\n"
+            else:
+                # General outreach with custom content
+                subject = f"{client_name} + {p_name}: Let's Connect"
+                response += f"**Subject:** {subject}\n\n"
                 response += f"---\n\n"
                 response += f"Dear {contact_name},\n\n"
-                response += f"I came across {p_name} and was impressed by your work in {mentioned_prospect.get('industry', 'your industry')}.\n\n"
-                response += f"{mentioned_prospect.get('summary', '')}\n\n"
-                response += f"At {client_name}, we help companies like yours {mentioned_prospect.get('fit_reason', 'grow their business')}.\n\n"
-                response += f"Would you be open to a brief call to explore how we might work together?\n\n"
-                response += f"Best regards,\n[Your Name]\n{client_name}\n\n"
-                response += "---\n\n*This is a template. Customize it before sending.*"
+                response += f"I'm reaching out from {client_name} regarding {p_name}.\n\n"
+                if custom_content:
+                    response += f"{custom_content}\n\n"
+                response += f"Based on our research into {p_name}'s work in {mentioned_prospect.get('industry', 'your industry')}, "
+                response += f"we believe {client_name} can provide significant value.\n\n"
+                response += f"**About {p_name}:** {mentioned_prospect.get('summary', '')}\n\n"
+                response += f"**Why we're reaching out:** {mentioned_prospect.get('fit_reason', 'We see great potential for collaboration.')}\n\n"
+                response += f"Would you be open to a conversation about how we can work together?\n\n"
+
+            response += f"Best regards,\n"
+            response += f"[Your Name]\n"
+            response += f"{client_name}\n\n"
+            response += f"---\n\n"
+            response += f"*📝 This is a custom draft based on your request. Edit as needed before sending.*"
 
             return history + [[message, response]], ""
 
@@ -1976,11 +2048,11 @@ def create_app():
 
                         gr.HTML("""
                         <div style="background: var(--bg-tertiary); padding: 12px; border-radius: 8px; margin-bottom: 8px;">
-                            <strong>✉️ Outreach</strong>
+                            <strong>✉️ Custom Emails</strong>
                             <ul style="margin: 8px 0 0 0; padding-left: 16px; font-size: 13px;">
-                                <li>Draft email to [prospect]</li>
-                                <li>Find contacts at [company]</li>
-                                <li>Show email for [prospect]</li>
+                                <li>Draft meeting email to [prospect] for Wednesday at 3pm</li>
+                                <li>Write email to [prospect] about partnership</li>
+                                <li>Show existing email for [prospect]</li>
                             </ul>
                         </div>
                         """)