first commit

.gitignore

@@ -0,0 +1 @@
+.env

app.py

@@ -0,0 +1,108 @@
+import os
+import openai
+import streamlit as st
+from utils.mail_generator import AzureOpenAIEmailGenerator
+
+
+# Configurazione e costanti
+class Config:
+    PAGE_TITLE = "Draftlytics 📧"
+    PAGE_ICON = "📬"
+    HEADER_COLOR = "#9d03fc"
+    ENGINE = "gpt-4"
+    CUSTOM_CSS = f"""  
+      <style>  
+          .css-1egvi7u {{ margin-top: -4rem; }}  /* Move app further up and remove top padding */  
+          .css-znku1x a {{ color: {HEADER_COLOR}; }}  /* Change hyperlink href link color */  
+          .css-qrbaxs {{ min-height: 0.0rem; }}  /* Change height of text input fields headers */  
+          .stSpinner > div > div {{ border-top-color: {HEADER_COLOR}; }}  /* Change spinner color to primary color */  
+          .css-15tx938 {{ min-height: 0.0rem; }}  /* Change min height of text input box */  
+          header {{ visibility: hidden; }}  /* Hide top header line */  
+          #MainMenu {{ visibility: hidden; }}  /* Hide "made with streamlit" footer menu area */  
+          footer {{ visibility: hidden; }}  /* Hide footer */  
+      </style>  
+    """
+
+
+# Classe principale dell'applicazione
+class DraftlyticsApp:
+    def __init__(self):
+        self.email_generator = AzureOpenAIEmailGenerator(Config.ENGINE)
+        self.email_text = ""
+
+    def set_page_config(self):
+        st.set_page_config(page_title=Config.PAGE_TITLE, page_icon=Config.PAGE_ICON)
+
+    def apply_custom_css(self):
+        st.markdown(Config.CUSTOM_CSS, unsafe_allow_html=True)
+
+    def display_header(self):
+        #st.image('images/rephraise_logo.png', width=100)
+        st.markdown(f"# Welcome to {Config.PAGE_TITLE} 📧")
+        st.write("### Your Expert AI Email Assistant 🤖")
+
+    def get_user_input(self):
+        st.subheader("📌 What is the main subject of your email?")
+        st.write("Note: The generated email will be in English.")
+        with st.expander("✉️ Email Input", expanded=True):
+            input_c1 = st.text_input('Provide the main topic of your email', '')
+            input_c2 = st.text_input('Provide the secondary topic of your email (optional)', '')
+            col1, col2, col3, space, col4 = st.columns([5, 5, 5, 0.5, 5])
+
+            with col1:
+                input_sender = st.text_input('Sender Name', '')
+            with col2:
+                input_recipient = st.text_input('Recipient Name', '')
+            with col3:
+                input_style = st.selectbox('Writing Style', ('formal', 'motivated', 'concerned', 'disappointed'),
+                                           index=0)
+            with col4:
+                st.write("\n")  # add spacing
+                st.write("\n")  # add spacing
+                if st.button('Generate Email 🚀'):
+                    with st.spinner('Generating your email... ⏳'):
+                        self.validate_and_generate_email(input_c1, input_c2, input_sender, input_recipient, input_style)
+
+    def validate_and_generate_email(self, input_c1, input_c2, input_sender, input_recipient, input_style):
+        if not input_c1.strip():
+            st.error('⚠️ The main topic is required!')
+            return
+        if not input_sender.strip():
+            st.error('⚠️ The sender name is required!')
+            return
+
+        if not input_recipient.strip():
+            st.error('⚠️ The recipient name is required!')
+            return
+
+        input_contents = [input_c1.strip()]
+        if input_c2.strip():
+            input_contents.append(input_c2.strip())
+
+        self.email_text = self.email_generator.generate_mail_format(
+            input_sender.strip(), input_recipient.strip(), input_style, input_contents
+        )
+        self.display_email()
+
+    @st.dialog("📧 RESULT:")
+    def display_email(self):
+        if self.email_text:
+            st.write('\n')  # add spacing
+            st.markdown(self.email_text)
+
+    def run(self):
+        self.set_page_config()
+        self.apply_custom_css()
+        self.display_header()
+        self.get_user_input()
+
+    # Funzione principale per eseguire l'applicazione
+
+
+def main():
+    app = DraftlyticsApp()
+    app.run()
+
+
+if __name__ == '__main__':
+    main()

requirements.txt

@@ -0,0 +1,4 @@
+streamlit==1.39.0
+openai==0.27.4
+python-dotenv
+langchain

utils/mail_generator.py

@@ -0,0 +1,111 @@
+import os
+from abc import ABC, abstractmethod
+from dotenv import load_dotenv, find_dotenv
+import openai
+from typing import List
+
+
+class AIEmailGenerator(ABC):
+    """Abstract base class for generating emails using AI."""
+
+    @abstractmethod
+    def _configure_api(self):
+        """Configure the API settings."""
+        pass
+
+    @abstractmethod
+    def generate_mail_contents(self, email_contents: List[str]) -> List[str]:
+        """Generate more formal and elaborate content for each email topic."""
+        pass
+
+    @abstractmethod
+    def generate_mail_format(self, sender: str, recipient: str, style: str, email_contents: List[str]) -> str:
+        """Generate a formatted email with updated content."""
+        pass
+
+
+class AzureOpenAIEmailGenerator(AIEmailGenerator):
+    """Concrete class for generating emails using OpenAI on Azure."""
+
+    def __init__(self, engine: str):
+        """Initialize the AzureOpenAIEmailGenerator class by loading environment variables and configuring OpenAI."""
+        self.engine = engine
+        self._load_environment()
+        self._configure_api()
+
+    def _load_environment(self):
+        """Load environment variables from .env file."""
+        load_dotenv(find_dotenv())
+
+    def _configure_api(self):
+        """Configure OpenAI API settings."""
+        openai.api_type = "azure"
+        openai.api_base = os.getenv('OPENAI_API_BASE')
+        openai.api_version = "2023-03-15-preview"
+        openai.api_key = os.getenv("OPENAI_API_KEY")
+
+    def generate_mail_contents(self, email_contents: List[str]) -> List[str]:
+        """Generates more formal and elaborate content for each email topic."""
+        for i, input_text in enumerate(email_contents):
+            prompt = (
+                f"Rewrite the text to be elaborate and polite.\n"
+                f"Abbreviations need to be replaced.\n"
+                f"Text: {input_text}\n"
+                f"Rewritten text:"
+            )
+            response = openai.ChatCompletion.create(
+                engine=self.engine,
+                messages=[
+                    {"role": "system", "content": "You are an AI assistant that helps people find information."},
+                    {"role": "user", "content": prompt}
+                ],
+                temperature=0,
+                max_tokens=len(input_text) * 3,
+                top_p=0.95,
+                frequency_penalty=0,
+                presence_penalty=0
+            )
+            email_contents[i] = response['choices'][0]['message']['content']
+        return email_contents
+
+    def generate_mail_format(self, sender: str, recipient: str, style: str, email_contents: List[str]) -> str:
+        """Generates a formatted email with updated content."""
+        email_contents = self.generate_mail_contents(email_contents)
+        contents_str = ""
+        contents_length = 0
+
+        for i, content in enumerate(email_contents):
+            contents_str += f"\nContent{i + 1}: {content}"
+            contents_length += len(content)
+
+        prompt = (
+            f"Write a professional email that sounds {style} and includes Content1 and Content2 in that order.\n\n"
+            f"Sender: {sender}\n"
+            f"Recipient: {recipient}{contents_str}\n\n"
+            f"Email Text:"
+        )
+        response = openai.ChatCompletion.create(
+            engine=self.engine,
+            messages=[
+                {"role": "system", "content": "You are an AI assistant that helps people find information."},
+                {"role": "user", "content": prompt}
+            ],
+            temperature=0,
+            max_tokens=contents_length * 2,
+            top_p=0.95,
+            frequency_penalty=0,
+            presence_penalty=0
+        )
+        return response['choices'][0]['message']['content']
+
+        # Esempio di utilizzo:
+
+if __name__ == "__main__":
+    engine = "gpt-4"  # Specificare l'engine desiderato
+    email_generator = AzureOpenAIEmailGenerator(engine)
+    sender = "example_sender@example.com"
+    recipient = "example_recipient@example.com"
+    style = "formal"
+    email_contents = ["This is the first content.", "This is the second content."]
+    formatted_email = email_generator.generate_mail_format(sender, recipient, style, email_contents)
+    print(formatted_email)