Update app.py

app.py

@@ -3,11 +3,17 @@ import smtplib
 from email.mime.text import MIMEText
 from email.mime.multipart import MIMEMultipart
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
-from langchain_core.prompts import PromptTemplate
 from langchain_huggingface import HuggingFacePipeline
 from langchain.agents import create_react_agent, AgentExecutor, Tool
+from langchain.prompts import PromptTemplate
 from langchain.memory import ConversationBufferMemory
 
+# Email configuration
+SENDER_EMAIL = "[email protected]"  # Replace with your email
+SENDER_PASSWORD = "Achuta@86"  # Replace with your email password
+SMTP_SERVER = "smtp.gmail.com"  # Replace with your SMTP server
+SMTP_PORT = 587  # Replace with your SMTP port
+
 # Set up the open-source LLM
 @st.cache_resource
 def load_model():
@@ -24,131 +30,82 @@ def load_model():
 
 local_llm = load_model()
 
-# Function to send an actual email using SMTP
-def send_actual_email(to_email, subject, body):
-    from_email = "[email protected]"  # Replace with your email address
-    from_password = "Nneka@86"  # Replace with your email password
-
-    msg = MIMEMultipart()
-    msg['From'] = from_email
-    msg['To'] = to_email
-    msg['Subject'] = subject
-
-    msg.attach(MIMEText(body, 'plain'))
-
+def send_email(to_email, subject, body):
     try:
-        server = smtplib.SMTP('smtp.gmail.com', 587)  # Replace with your SMTP server details
-        server.starttls()
-        server.login(from_email, from_password)
-        text = msg.as_string()
-        server.sendmail(from_email, to_email, text)
-        server.quit()
-        return "Email sent successfully!"
+        message = MIMEMultipart()
+        message["From"] = SENDER_EMAIL
+        message["To"] = to_email
+        message["Subject"] = subject
+        message.attach(MIMEText(body, "plain"))
+
+        with smtplib.SMTP(SMTP_SERVER, SMTP_PORT) as server:
+            server.starttls()
+            server.login(SENDER_EMAIL, SENDER_PASSWORD)
+            server.send_message(message)
+        return "Email sent successfully"
     except Exception as e:
         return f"Failed to send email: {str(e)}"
 
-# Tool for sending the email
-def send_email_tool(name, email, preferred_date):
-    zoom_link = """
-    Lawrence Emenike is inviting you to a scheduled Zoom meeting.
-
-    Topic: Lawrence Emenike's Zoom Meeting
-    Time: Aug 10, 2024 05:00 PM West Central Africa
-
-    Join Zoom Meeting
-    https://us04web.zoom.us/j/73793374638?pwd=S0TEJ30da7dhQ8viOdafMzPfCVzoLJ.1
-
-    Meeting ID: 737 9337 4638
-    Passcode: 9cNPkn
-    """
-    
-    subject = "Invitation to Discuss Cybersecurity Program"
-    body = f"""
-    Hi {name},
-
-    Thank you for your interest in our cybersecurity program. We would like to invite you to a video call to discuss how our program can benefit your organization.
-
-    Here are the details:
-
-    Preferred Date: {preferred_date}
-    Zoom Meeting Link: {zoom_link}
-
-    Looking forward to our discussion!
-
-    Best regards,
-    Lawrence Emenike
-    """
-
-    # Send the actual email
-    result = send_actual_email(email, subject, body)
-    return result
-
-# Define the tools for the agent
 tools = [
     Tool(
         name="Send Email",
-        func=send_email_tool,
-        description="Sends an email with the Zoom meeting details"
+        func=send_email,
+        description="Sends an email. Args: to_email, subject, body"
     )
 ]
 
-# Define the prompt with clearer instructions for the LLM
 prompt = PromptTemplate.from_template(
-    """You are an AI CyberSecurity Program Advisor. Your goal is to engage with the user and collect their name, email address, and preferred date for a video call. Once collected, send an email invitation to a Zoom meeting.
+    """You are a helpful assistant scheduling cybersecurity program meetings. 
+    Your task is to collect the user's name, email, and preferred meeting date, then send a meeting invitation.
 
-    You have access to the following tools:
+    Use the following format:
+    Human: <user's input>
+    Assistant: <your response>
+    Human: <user's input>
+    Assistant: <your response>
+    ...
 
-    {tools}
+    Once you have all the information, use the Send Email tool to send the invitation.
+    The email should include this Zoom link: https://us04web.zoom.us/j/73793374638?pwd=S0TEJ30da7dhQ8viOdafMzPfCVzoLJ.1
+    And this Meeting ID: 737 9337 4638
 
-    Use the following format:
+    Begin the conversation by asking for the user's name.
 
-    Question: the input question you must answer
-    Thought: you should always think about what to do
-    Action: the action to take, should be one of [{tool_names}]
-    Action Input: the input to the action
-    Observation: the result of the action
-    ... (this Thought/Action/Action Input/Observation can repeat N times)
-    Thought: I now know the final answer
-    Final Answer: The invitation email has been sent successfully.
-
-    If you need to send an email, use the following format exactly:
-    Action: Send Email
-    Action Input: {{"name": "<name>", "email": "<email>", "preferred_date": "<date>"}}
-    
-    Begin!
-
-    Question: {input}
-    Thought: Let's approach this step-by-step:
+    {chat_history}
+    Human: {input}
+    Assistant: Let's proceed step by step:
     {agent_scratchpad}"""
 )
 
-# Create the React agent
-agent = create_react_agent(
-    llm=local_llm, 
-    tools=tools, 
-    prompt=prompt
-)
-
-# Create the agent executor with handle_parsing_errors=True
+agent = create_react_agent(local_llm, tools, prompt)
 agent_executor = AgentExecutor.from_agent_and_tools(
-    agent=agent, 
-    tools=tools, 
-    verbose=True, 
-    memory=ConversationBufferMemory(),
-    handle_parsing_errors=True  # Added to handle parsing errors
+    agent=agent, tools=tools, verbose=True, memory=ConversationBufferMemory(memory_key="chat_history", return_messages=True)
 )
 
 # Streamlit interface
-st.title("AI CyberSecurity Program Advisor")
+st.title("CyberSecurity Program Meeting Scheduler")
+
+st.write("Chat with the AI to schedule your meeting. The AI will ask for your name, email, and preferred meeting date.")
+
+if "messages" not in st.session_state:
+    st.session_state.messages = []
 
-st.write("The AI will collect your details and send you an invitation for a video call.")
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
 
-# Starting the conversation
-if st.button("Start Conversation"):
-    initial_message = "Please provide your name, email address, and preferred date for the video call."
-    with st.spinner("AI is processing..."):
-        response = agent_executor.invoke({"input": initial_message})
-        st.write(response['output'])  # Display the agent's final response
+if prompt := st.chat_input("Your message"):
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    with st.chat_message("user"):
+        st.markdown(prompt)
+
+    with st.chat_message("assistant"):
+        with st.spinner("Thinking..."):
+            response = agent_executor.run(prompt)
+            st.markdown(response)
+            st.session_state.messages.append({"role": "assistant", "content": response})
 
 st.sidebar.title("About")
-st.sidebar.info("This is a demo of an AI CyberSecurity Program Advisor using LangChain. It helps you schedule a video call and sends an email invitation automatically.")
+st.sidebar.info("This is an interactive CyberSecurity Program Meeting Scheduler. Chat with the AI to schedule your meeting. It will collect your information and send a meeting invitation via email.")
+
+# To run this script, use: streamlit run your_script_name.py