First model version

README.md

@@ -1,10 +1,10 @@
 ---
-title: Cold Email
-emoji: 🌍
-colorFrom: purple
+title: Cold Email Part One
+emoji: 🐠
+colorFrom: red
 colorTo: red
 sdk: streamlit
-sdk_version: 1.39.0
+sdk_version: 1.38.0
 app_file: app.py
 pinned: false
 ---

app.py

@@ -0,0 +1,103 @@
+
+import streamlit as st
+import pandas as pd
+import traceback
+import backend
+
+
+st.set_page_config(page_title="Cold Email Generator", layout="wide",initial_sidebar_state='collapsed')
+
+
+def main():
+    try:
+        # ------------------------------------- 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
+        )
+        # --------------------------------------------------------------------------------------------------
+
+
+        st.header("Cold Email Generator")
+        
+        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:
+                recipient_role = st.text_input("Recipient Name")
+            
+            details_col1, details_col2 = st.columns(2)
+            with details_col1:
+                details_type = st.radio("Select Details Type", ["Personal Details", "Company Details"])
+            
+            with details_col2:
+                name = st.text_input("Senders Name")
+                designation = ""
+                company_name = ""
+
+                if details_type == "Personal Details":
+                    designation = st.text_input("Designation/Role")
+                else:
+                    company_name = st.text_input("Company Name")
+                    designation = st.text_input("Designation/Role")
+            
+        
+            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,recipient_role, tone, context, details_type, name, designation, company_name)
+                st.session_state.edited_email  = st.text_area("Generated Email", value = generated_email,  height=300, disabled=False)
+                st.session_state.is_generate_clicked = True
+            else:
+                st.text_area("Generated Email",  height=300, disabled=True)
+
+            # Edit and Send button
+
+            if st.button("Completed Editing"):
+                st.session_state.is_cmp_editing_clicked = True
+
+        st.markdown("---")
+
+        if st.session_state.get("is_cmp_editing_clicked")  and st.session_state.get("is_generate_clicked") :
+            st.subheader("Chat to refine your email")
+
+            
+            email_col1, email_col2 = st.columns(2)
+            with email_col1:
+                sender_email = st.text_input("Sender's Email")
+            with email_col2:
+                receiver_email = st.text_input("Receiver's Email")
+            
+            subject = st.text_input("Subject")
+            st.text_area("Generated Email", value=st.session_state.edited_email,  height=300, disabled=False)
+
+        is_send_clicked = ""
+        if st.session_state.get("is_cmp_editing_clicked") :
+           is_send_clicked = st.button("Send Email")
+
+        if is_send_clicked:
+            st.subheader("All done, real-time email sending will be implemented in part2 😊 ")
+                    
+
+    except Exception as err:
+        traceback.print_exc()
+        print(err)
+        
+if __name__ == "__main__":
+    main()

backend.py

@@ -0,0 +1,27 @@
+
+import llm
+import traceback
+
+def collect_context_for_email(industry, rec_name, tone, context, details_type, name, designation, company_name =""):
+
+    try:
+
+        print("number of calls")
+        user_input = {
+            "industry": industry,
+            "recipient_name": rec_name,
+            "tone": tone,
+            "context": context,
+            "details_type": details_type,
+            "name": name,
+            "designation": designation,
+            "company_name": company_name
+        }
+
+        llm_output = llm.chat_with_groq(user_input)
+
+        return llm_output
+    except Exception as err:
+        traceback.print_exc()
+        print(err)
+    

llm.py

@@ -0,0 +1,55 @@
+# Default
+import os
+from groq import Groq
+import traceback
+
+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_name']} based on the following information:
+        
+        Industry: {user_input['industry']}
+        Tone: {user_input['tone']}
+        Context: {user_input['context']}
+        Senders Name: {user_input['name']}
+        Designation: {user_input['designation']}
+
+        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)
+
+
+

pages/1_❤️_Chat_Page.py

@@ -0,0 +1,8 @@
+import streamlit as st
+import pandas as pd
+import traceback
+import llm
+
+st.header("To be implemented in part 2")
+
+

pages/2_🗒_Templates_Page.py

@@ -0,0 +1,6 @@
+import streamlit as st
+import pandas as pd
+import traceback
+import llm
+
+st.header("To be implemented in part 2")

requirements.txt

@@ -0,0 +1,4 @@
+streamlit
+pandas
+groq
+python-dotenv