Update app.py

app.py

@@ -1,9 +1,10 @@
 import os
 import gradio as gr
-from langchain_community.llms import Groq
-from langchain_core.prompts import PromptTemplate
-from langchain_core.output_parsers import StrOutputParser
-from langchain_core.runnables import RunnablePassthrough
+from langchain.llms import Groq
+from langchain.prompts import PromptTemplate
+from langchain.chains import LLMChain
+import pandas as pd
+from datetime import datetime
 from sendgrid import SendGridAPIClient
 from sendgrid.helpers.mail import Mail
 import json
@@ -121,12 +122,13 @@ class EmailGenie:
         return self.current_profile is not None
     
     def structure_info(self, info: str) -> str:
-        structure_prompt = PromptTemplate.from_template(
-            "Please structure and summarize the following information about a person or company: {info}"
+        prompt = PromptTemplate(
+            input_variables=["info"],
+            template="Please structure and summarize the following information about a person or company: {info}"
         )
-        chain = structure_prompt | self.llm | StrOutputParser()
-        return chain.invoke({"info": info})
-    
+        chain = LLMChain(llm=self.llm, prompt=prompt)
+        return chain.run(info=info)
+
     def generate_email(self, template_name: str, context: Dict) -> str:
         if not self.current_profile:
             return "Please set up your profile first."
@@ -134,8 +136,9 @@ class EmailGenie:
         template = self.email_template.get_template(template_name)
         structured_info = self.structure_info(context.get('recipient_info', ''))
         
-        email_prompt = PromptTemplate.from_template(
-            """
+        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})
@@ -146,17 +149,17 @@ class EmailGenie:
             """
         )
         
-        chain = email_prompt | self.llm | StrOutputParser()
-        return chain.invoke({
-            "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
-        })
-    
+        chain = LLMChain(llm=self.llm, prompt=email_prompt)
+        return 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
+        )
+
     def preview_email(self, email_content: str) -> str:
         if not self.current_profile:
             return "Please set up your profile first."