Second model version

README.md

@@ -1,10 +1,10 @@
 ---
-title: Cold Email 2.0
-emoji: 🔥
+title: Email Genie Part Two
+emoji: 💻
 colorFrom: pink
-colorTo: gray
+colorTo: pink
 sdk: streamlit
-sdk_version: 1.39.0
+sdk_version: 1.38.0
 app_file: app.py
 pinned: false
 ---

app.py

@@ -0,0 +1,57 @@
+import streamlit as st
+import pandas as pd
+import traceback
+import backend
+from lib.support.generate_email_support import render_gen_email
+from lib.support.template_email_support import render_templates
+from lib.support.user_profile_support import render_user_profile
+
+
+
+
+st.set_page_config(page_title="Cold Email Generator", layout="wide")
+
+# ------------------------------------- Custom CSS -------------------------------------------------------------
+st.markdown(
+    """
+    <style>
+    .css-1d391kg {  /* Adjust this class based on your Streamlit version */
+        padding-left: 10px;
+        padding-right: 10px;
+    }
+    </style>
+    """,
+    unsafe_allow_html=True
+)
+# --------------------------------------------------------------------------------------------------
+
+
+def main():
+    try:
+
+        if not st.session_state.get("excel_file"):
+            st.session_state.excel_file = "cold_email_data.xlsx"
+
+        st.title("Cold Email Generator")
+
+        if not st.session_state.get("user_id"):
+            render_user_profile()
+        else:
+            if not st.session_state.get("page"):
+                page = st.sidebar.radio("Navigate", ["Generate Email", "Template Emails", "User Profiles"])
+            else:
+                page = st.session_state.page
+
+            if page == "Generate Email":
+                render_gen_email()
+            elif page == "Template Emails":
+                render_templates()
+            elif page == "User Profiles":
+                render_user_profile()
+
+    except Exception as err:
+        traceback.print_exc()
+        print(err)
+        
+if __name__ == "__main__":
+    main()

backend.py

@@ -0,0 +1,26 @@
+
+import llm
+import traceback
+
+def collect_context_for_email(industry, role, tone, context, sender_details, rec_details):
+
+    try:
+                   
+        user_input = {
+            "industry": industry,
+            "recipient_role": rec_details["rec_name"],
+            "tone": tone,
+            "context": context,
+            "sender_name": sender_details["sender_name"],
+            "receiver_name": rec_details["rec_name"],
+            "receiver_designation": rec_details["rec_designation"],
+            "receiver_company_name": rec_details["rec_company_name"]
+        }
+
+        llm_output = llm.chat_with_groq(user_input)
+
+        return llm_output
+    except Exception as err:
+        traceback.print_exc()
+        print(err)
+    

lib/common/database_support.py

@@ -0,0 +1,94 @@
+import streamlit as st
+import pandas as pd
+from openpyxl import load_workbook, Workbook
+from datetime import datetime
+import uuid
+
+
+def load_or_create_excel(file_path):
+    try:
+        wb = load_workbook(file_path)
+    except FileNotFoundError:
+        wb = Workbook() 
+        wb.create_sheet("Users")
+        wb.create_sheet("GeneratedEmails")
+        wb.create_sheet("TemplateEmails")
+        
+        users_sheet = wb["Users"]
+        users_sheet.append(["UserID", "Name", "Email", "Company", "Role"])
+        
+        generated_emails_sheet = wb["GeneratedEmails"]
+        generated_emails_sheet.append(["EmailID", "UserID", "Subject", "Body", "Timestamp"])
+        
+        template_emails_sheet = wb["TemplateEmails"]
+        template_emails_sheet.append(["TemplateID", "Name", "Subject", "Body"])
+        
+        wb.save(file_path)
+    return wb
+
+# Function to add user to Excel
+def add_user_to_excel(file_path, user_data):
+    wb = load_or_create_excel(file_path)
+    sheet = wb["Users"]
+    user_id = str(uuid.uuid4())
+    sheet.append([user_id, user_data['name'], user_data['email'], user_data['company'], user_data['role']])
+    wb.save(file_path)
+    return user_id
+
+# Function to save generated email
+def save_email_to_excel(file_path, user_id, subject, body):
+    wb = load_or_create_excel(file_path)
+    sheet = wb["GeneratedEmails"]
+    email_id = str(uuid.uuid4())
+    sheet.append([email_id, user_id, subject, body, datetime.now().strftime("%Y-%m-%d %H:%M:%S")])
+    wb.save(file_path)
+
+# Function to add template email
+def add_template_email(file_path, name, subject, body):
+    wb = load_or_create_excel(file_path)
+    sheet = wb["TemplateEmails"]
+    template_id = str(uuid.uuid4())
+    sheet.append([template_id, name, subject, body])
+    wb.save(file_path)
+
+# Function to get all template emails
+def get_template_emails(file_path):
+    df = pd.read_excel(file_path, sheet_name="TemplateEmails")
+    return df.to_dict('records')
+
+# Function to generate email using Groq LLM (placeholder)
+def generate_email(prompt):
+    # Implement your Groq LLM email generation logic here
+    return f"Subject: Generated Email\n\nBody: {prompt}"
+
+def get_user_by_id(file_path, user_id):
+    wb = load_workbook(file_path)
+    sheet = wb["Users"]
+    for row in sheet.iter_rows(min_row=2, values_only=True):
+        if row[0] == user_id:
+            return {'name': row[1], 'email': row[2], 'company': row[3], 'role': row[4]}
+    return None
+
+def update_user_in_excel(file_path, user_id, user_data):
+    wb = load_workbook(file_path)
+    sheet = wb["Users"]
+    for row in sheet.iter_rows(min_row=2):
+        if row[0].value == user_id:
+            row[1].value = user_data['name']
+            row[2].value = user_data['email']
+            row[3].value = user_data['company']
+            row[4].value = user_data['role']
+            break
+    wb.save(file_path)
+
+def store_sent_email(file_path, sender_email, receiver_email, subject, body):
+    wb = load_or_create_excel(file_path)
+    sheet = wb["GeneratedEmails"]
+    email_id = str(uuid.uuid4())
+    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    
+    # Append the sent email details to the sheet
+    sheet.append([email_id, sender_email, receiver_email, subject, body, timestamp])
+    
+    wb.save(file_path)
+    return email_id

lib/support/generate_email_support.py

@@ -0,0 +1,86 @@
+import streamlit as st
+import pandas as pd
+import traceback
+import backend
+from lib.common.database_support import save_email_to_excel
+import time
+
+def render_gen_email():
+    try:
+
+        st.title("Generate Email")
+        excel_file = st.session_state.get("excel_file", "Error Not Found")
+
+        if excel_file == "Error Not Found":
+            raise Exception("Error Not Found")
+        
+        if not st.session_state.get("user_id"):
+            st.warning("Please fill in your profile information first.")
+            return
+
+        split_screen_col1, split_screen_col2 = st.columns(2)
+        # Input fields
+        with split_screen_col1:
+            industry_col, recipient_col = st.columns(2)
+            with industry_col:
+                industry = st.text_input("Industry")
+            with recipient_col:
+                role = st.text_input("Role")
+            
+            details_col1, details_col2 = st.columns(2)
+            with details_col1:
+                sender_name = st.text_input("Senders Name")
+
+            with details_col2:
+                rec_name = st.text_input("Receivers Name")
+                rec_designation = st.text_input("Receivers Designation")
+                rec_company_name = st.text_input("Receivers Company Name")
+            
+            sender_details = {
+                "sender_name": sender_name,
+            }
+            rec_details = {
+                "rec_name": rec_name,
+                "rec_designation": rec_designation,
+                "rec_company_name": rec_company_name
+            }
+
+        
+            tone = st.selectbox("Tone", ["Formal", "Casual", "Friendly", "Professional"])
+            context = st.text_area("Context")
+
+            is_generate_clicked = st.button("Generate Email")
+        
+        with split_screen_col2:
+
+            if is_generate_clicked:
+
+                generated_email = backend.collect_context_for_email(industry, role, tone, context, sender_details,rec_details )
+                st.session_state.generated_email = generated_email
+                st.text_area("Generated Email", value=generated_email, height=300, key="editable_email")
+                st.session_state.is_generate_clicked = True
+            else:
+                st.text_area("Generated Email", height=300, disabled=True)
+
+            # Save to templates button
+            if st.button("Save to templates"):
+                if st.session_state.get("generated_email"):
+                    edited_email = st.session_state.editable_email
+                    timestamp = int(time.time())
+                    new_template = {
+                        "Name": f"Generated Email {timestamp}",
+                        "Subject": "New Generated Email",
+                        "Body": edited_email
+                    }
+                    st.session_state.new_template = new_template  # Store the new template in session state
+                    st.session_state.show_new_template = True
+                    st.session_state.page = "Template Emails"
+                    st.rerun()
+                else:
+                    st.warning("Please generate an email first.")
+
+        st.markdown("---")
+
+    except Exception as err:
+        traceback.print_exc()
+        print(err)

lib/support/template_email_support.py

@@ -0,0 +1,145 @@
+import streamlit as st
+import traceback
+import time
+from lib.common.database_support import get_template_emails, add_template_email, store_sent_email, save_email_to_excel
+
+def get_sample_emails():
+    return [
+        {
+            "Name": "Professional Introduction",
+            "Subject": "Introduction - [Your Name] from [Your Company]",
+            "Body": "Dear [Recipient],\n\nI hope this email finds you well. My name is [Your Name], and I am reaching out to introduce myself and [Your Company]. We specialize in [brief description of your services/products].\n\nI would love the opportunity to discuss how we might be able to collaborate. Would you be available for a brief call next week?\n\nBest regards,\n[Your Name]\n[Your Position]\n[Your Company]"
+        },
+        {
+            "Name": "Friendly Invitation",
+            "Subject": "Let's catch up over coffee!",
+            "Body": "Hey [Friend's Name]!\n\nIt's been too long since we last caught up. How about we grab a coffee this weekend? I'd love to hear what you've been up to!\n\nLet me know if Saturday afternoon works for you.\n\nCheers,\n[Your Name]"
+        },
+        {
+            "Name": "Customer Support",
+            "Subject": "Re: Your Recent Order #12345",
+            "Body": "Dear [Customer Name],\n\nThank you for contacting our support team regarding your recent order #12345. We apologize for any inconvenience you may have experienced.\n\nWe have reviewed your case and [explanation of the solution or next steps].\n\nIf you have any further questions, please don't hesitate to reach out.\n\nBest regards,\n[Your Name]\nCustomer Support Team"
+        },
+        {
+            "Name": "Job Application",
+            "Subject": "Application for [Position] - [Your Name]",
+            "Body": "Dear Hiring Manager,\n\nI am writing to express my strong interest in the [Position] role at [Company Name], as advertised on [where you found the job posting].\n\nWith my background in [relevant skills/experience], I am confident that I would be a valuable addition to your team. [Brief highlight of your qualifications].\n\nPlease find attached my resume and portfolio. I look forward to the opportunity to discuss how my skills and experiences align with your needs.\n\nThank you for your time and consideration.\n\nSincerely,\n[Your Name]"
+        },
+        {
+            "Name": "Event Invitation",
+            "Subject": "You're Invited: [Event Name]",
+            "Body": "Dear [Recipient],\n\nWe are delighted to invite you to [Event Name], taking place on [Date] at [Time] at [Location].\n\n[Brief description of the event and its purpose]\n\nPlease RSVP by [RSVP deadline] to [contact email/phone].\n\nWe look forward to seeing you there!\n\nBest regards,\n[Your Name]\n[Organization]"
+        },
+        {
+            "Name": "Thank You Note",
+            "Subject": "Thank You for Your Support",
+            "Body": "Dear [Name],\n\nI wanted to take a moment to express my sincere gratitude for [reason for thanking - e.g., your recent donation, attending our event, your ongoing support].\n\n[Personalized message about the impact of their action/support]\n\nThank you again for your generosity and kindness.\n\nWarmest regards,\n[Your Name]"
+        },
+        {
+            "Name": "Business Proposal",
+            "Subject": "Business Proposal: [Brief Description]",
+            "Body": "Dear [Recipient],\n\nI hope this email finds you well. I am reaching out to propose a business opportunity that I believe would be mutually beneficial for [Your Company] and [Their Company].\n\n[Brief overview of the proposal]\n\nKey benefits of this partnership include:\n- [Benefit 1]\n- [Benefit 2]\n- [Benefit 3]\n\nI would welcome the opportunity to discuss this proposal in more detail. Are you available for a call next week?\n\nBest regards,\n[Your Name]\n[Your Position]\n[Your Company]"
+        },
+        {
+            "Name": "Apology Email",
+            "Subject": "Our Sincere Apologies",
+            "Body": "Dear [Customer Name],\n\nOn behalf of [Company Name], I want to sincerely apologize for [description of the issue or mistake].\n\nWe understand the inconvenience this has caused you and take full responsibility for our error. We are taking the following steps to rectify the situation:\n\n[List of actions being taken]\n\nAs a gesture of our commitment to making this right, we would like to offer you [compensation or solution].\n\nWe value your business and hope to regain your trust.\n\nSincerely,\n[Your Name]\n[Your Position]\n[Company Name]"
+        }
+    ]
+
+def render_templates():
+    try:
+        excel_file = st.session_state.get("excel_file", "Error Not Found")
+
+        if excel_file == "Error Not Found":
+            raise Exception("Error Not Found")
+        
+        if not st.session_state.get("user_id"):
+            st.warning("Please fill in your profile information first.")
+            return
+
+        st.title("Email Templates")
+
+        new_template = st.session_state.get("new_template", None)
+        show_only_new = st.session_state.get("show_new_template", False)
+
+        if show_only_new and new_template:
+            templates_to_show = [new_template]
+        else:
+            if "templates" not in st.session_state:
+                st.session_state.templates = get_sample_emails()
+            templates_to_show = st.session_state.templates
+
+        template_names = [template['Name'] for template in templates_to_show]
+        
+        if not show_only_new:
+            template_names.insert(0, "Select a template")
+
+        selected_template_name = st.selectbox("Choose a template", template_names, index=0 if show_only_new else None)
+
+        if selected_template_name and selected_template_name != "Select a template":
+            selected_template = next((template for template in templates_to_show if template['Name'] == selected_template_name), None)
+            # print(selected_template)
+            print(selected_template, "\n\nselc\n\n")
+            if selected_template:
+                # Use session state to store editable fields
+                if 'current_template' not in st.session_state:
+                    print("inside")
+
+                    st.session_state.current_template = {
+                        'sender_email': '',
+                        'receiver_email': '',
+                        'subject': selected_template['Subject'],
+                        'body': selected_template['Body']
+                    }
+
+                col1, col2 = st.columns(2)
+                with col1:
+                    st.session_state.current_template['sender_email'] = st.text_input("Sender's Email", value=st.session_state.current_template['sender_email'])
+                    st.session_state.current_template['subject'] = st.text_input("Subject", value=st.session_state.current_template['subject'])
+                with col2:
+                    st.session_state.current_template['receiver_email'] = st.text_input("Receiver's Email", value=st.session_state.current_template['receiver_email'])
+                
+                st.session_state.current_template['body'] = st.text_area("Email Body", value=st.session_state.current_template['body'], height=200)
+
+                col1, col2 = st.columns(2)
+                # with col1:
+                #     if st.button("Save Template"):
+                #         if st.session_state.current_template['subject'] and st.session_state.current_template['body']:
+                #             save_email_to_excel(excel_file, st.session_state.user_id, st.session_state.current_template['subject'], st.session_state.current_template['body'])
+                #             st.success("Template saved successfully!")
+                #         else:
+                #             st.warning("Please fill in the subject and body before saving.")
+
+                with col2:
+                    if st.button("Send Email"):
+                        if all(st.session_state.current_template.values()):
+                            store_sent_email(excel_file, st.session_state.current_template['sender_email'], st.session_state.current_template['receiver_email'], st.session_state.current_template['subject'], st.session_state.current_template['body'])
+                            st.success("Email sent successfully")
+                            
+                            if st.session_state.get("new_template"):
+                                st.session_state.pop("new_template")
+                            if st.session_state.get("show_new_template"):
+                                st.session_state.pop("show_new_template")
+
+                            if 'current_template' in st.session_state:
+                                st.session_state.pop("current_template")
+                            time.sleep(1)
+                            st.session_state.page = "Generate Email"
+                            st.rerun()
+                        else:
+                            st.warning("Please fill in all fields before sending.")
+            else:
+                st.warning("Selected template not found. Please choose a valid template.")
+
+        if show_only_new:
+            if st.button("Back to All Templates"):
+                st.session_state.pop("new_template", None)
+                st.session_state.pop("show_new_template", None)
+                st.session_state.pop("current_template",None)
+                st.rerun()
+
+    except Exception as err:
+        traceback.print_exc()
+        print(err)
+

lib/support/user_profile_support.py

@@ -0,0 +1,44 @@
+import streamlit as st
+import traceback
+from lib.common.database_support import add_user_to_excel, get_user_by_id, update_user_in_excel
+
+
+def render_user_profile(): 
+    try:
+        excel_file = st.session_state.get("excel_file", "Error Not Found")
+
+        if excel_file == "Error Not Found":
+            raise Exception("Error Not Found")
+
+        st.title("User Profile")
+        
+        user_id = st.session_state.get("user_id")
+        if not user_id:
+            st.warning("Please fill in your profile information to continue.")
+            user_data = {'name': '', 'email': '', 'company': '', 'role': ''}
+        else:
+            user_data = get_user_by_id(excel_file, user_id)
+
+        with st.form("user_form"):
+            name = st.text_input("Name", value=user_data['name'])
+            email = st.text_input("Email", value=user_data['email'])
+            company = st.text_input("Company", value=user_data['company'])
+            role = st.text_input("Role", value=user_data['role'])
+            submitted = st.form_submit_button("Save Profile")
+            
+            if submitted:
+                updated_user_data = {'name': name, 'email': email, 'company': company, 'role': role}
+                if user_id:
+                    update_user_in_excel(excel_file, user_id, updated_user_data)
+                    st.success("Profile updated successfully!")
+                else:
+                    user_id = add_user_to_excel(excel_file, updated_user_data)
+                    st.session_state['user_id'] = user_id
+                    st.success("Profile saved! You can now access other features.")
+                st.session_state['user_email'] = email
+                st.rerun()
+
+    except Exception as err:
+        traceback.print_exc()
+        print(err)
+

llm.py

@@ -0,0 +1,59 @@
+# Default
+import os
+from groq import Groq
+from dotenv import load_dotenv
+import traceback
+
+load_dotenv()
+
+client = Groq(
+    # This is the default and can be omitted
+    api_key=os.environ.get("GROQ_API_KEY"),
+)
+
+def chat_with_groq(user_input, additional_info=None):
+    
+    try:
+
+        template_prompt = f"""
+        
+        You are a professional email writer who specialises in writing cold email.
+        Generate a cold email to {user_input['recipient_role']} based on the following information:
+        
+        Industry: {user_input['industry']}
+        Tone: {user_input['tone']}
+        Context: {user_input['context']}
+        Sender's Name: {user_input['sender_name']}
+        Receiver's Name: {user_input['receiver_name']}
+        Receiver's Designation: {user_input['receiver_designation']}
+        Receiver's Company Name: {user_input['receiver_company_name']}
+
+        Incorporate all information and write a compelling email.
+
+        """
+        # test purposes:
+        # return template_prompt
+        chat_completion = client.chat.completions.create(
+            messages=[
+                {
+                    "role": "system",
+                    "content": "You are an expert email writer specializing in cold emails.",
+                },
+                {
+                    "role": "user",
+                    "content": template_prompt,
+                }
+            ],
+            model="mixtral-8x7b-32768",
+        )
+
+        generated_email = chat_completion.choices[0].message.content
+        return generated_email
+
+
+    except Exception as err:
+        traceback.print_exc()
+        print(err)
+
+
+

requirements.txt

@@ -0,0 +1,6 @@
+streamlit
+pandas
+groq
+python-dotenv
+pandas
+openpyxl