Update app.py

app.py

@@ -1,8 +1,8 @@
 import os
 import gradio as gr
-from groq import Groq
-import pandas as pd
-from datetime import datetime
+from langchain.llms import Groq
+from langchain.prompts import PromptTemplate
+from langchain.chains import LLMChain
 from sendgrid import SendGridAPIClient
 from sendgrid.helpers.mail import Mail
 import json
@@ -37,7 +37,7 @@ class UserProfile:
         self._ensure_storage_exists()
     
     def _ensure_storage_exists(self):
-        if not os.path.exists(self.storage_path):
+        if not os.path.exists(self.storage_path):   
             with open(self.storage_path, 'w') as f:
                 json.dump({"profiles": []}, f)
     
@@ -108,7 +108,7 @@ class EmailTemplate:
 
 class EmailGenie:
     def __init__(self):
-        self.client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
+        self.llm = Groq(api_key=os.environ.get("GROQ_API_KEY"))
         self.sendgrid_client = initialize_email_client()
         self.sender_email = os.environ.get("SENDER_EMAIL")
         self.user_profile = UserProfile()
@@ -120,13 +120,13 @@ class EmailGenie:
         return self.current_profile is not None
     
     def structure_info(self, info: str) -> str:
-        prompt = f"Please structure and summarize the following information about a person or company: {info}"
-        chat_completion = self.client.chat.completions.create(
-            messages=[{"role": "user", "content": prompt}],
-            model="llama-3.1-8b-instant",
+        structure_prompt = PromptTemplate(
+            input_variables=["info"],
+            template="Please structure and summarize the following information about a person or company: {info}"
         )
-        return chat_completion.choices[0].message.content
-
+        structure_chain = LLMChain(llm=self.llm, prompt=structure_prompt)
+        return structure_chain.run(info)
+    
     def generate_email(self, template_name: str, context: Dict) -> str:
         if not self.current_profile:
             return "Please set up your profile first."
@@ -134,22 +134,30 @@ class EmailGenie:
         template = self.email_template.get_template(template_name)
         structured_info = self.structure_info(context.get('recipient_info', ''))
         
-        prompt = f"""
-        Using the following template and information, create a personalized email:
-        Template: {template}
-        Sender: {self.current_profile['name']} ({self.current_profile['industry']})
-        Sender Background: {self.current_profile['background']}
-        Recipient: {context['recipient']}
-        Subject: {context['email_subject']}
-        Recipient Information: {structured_info}
-        """
+        email_prompt = PromptTemplate(
+            input_variables=["template", "sender_name", "sender_industry", "sender_background", "recipient", "subject", "structured_info"],
+            template="""
+            Using the following template and information, create a personalized email:
+            Template: {template}
+            Sender: {sender_name} ({sender_industry})
+            Sender Background: {sender_background}
+            Recipient: {recipient}
+            Subject: {subject}
+            Recipient Information: {structured_info}
+            """
+        )
         
-        chat_completion = self.client.chat.completions.create(
-            messages=[{"role": "user", "content": prompt}],
-            model="llama-3.1-8b-instant",
+        email_chain = LLMChain(llm=self.llm, prompt=email_prompt)
+        return email_chain.run(
+            template=template,
+            sender_name=self.current_profile['name'],
+            sender_industry=self.current_profile['industry'],
+            sender_background=self.current_profile['background'],
+            recipient=context['recipient'],
+            subject=context['email_subject'],
+            structured_info=structured_info
         )
-        return chat_completion.choices[0].message.content
-
+    
     def preview_email(self, email_content: str) -> str:
         if not self.current_profile:
             return "Please set up your profile first."