Update app.py

app.py

@@ -1,32 +1,4 @@
-import streamlit as st
-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.memory import ConversationBufferMemory
-
-# Mock lead data
-LEADS = [
-    {"name": "John Doe", "email": "[email protected]", "company": "TechCorp"},
-]
-
-# Set up the open-source LLM
-@st.cache_resource
-def load_model():
-    model_name = "google/flan-t5-large"
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
-    pipe = pipeline(
-        "text2text-generation",
-        model=model, 
-        tokenizer=tokenizer, 
-        max_length=512
-    )
-    return HuggingFacePipeline(pipeline=pipe)
-
-local_llm = load_model()
-
-# Define the tools for the agent
+# Mock function to send an email
 def send_email(to_email, subject, body):
     # For demo purposes, we'll just print the email details
     st.write(f"Email sent to: {to_email}")
@@ -34,68 +6,55 @@ def send_email(to_email, subject, body):
     st.write(f"Body: {body}")
     return "Email sent successfully"
 
-tools = [
-    Tool(
-        name="Send Email",
-        func=send_email,
-        description="Useful for sending emails to leads"
-    )
-]
+# Streamlit interface
+st.title("AI CyberSecurity Program Advisor")
 
-# Define the prompt
-prompt = PromptTemplate.from_template(
-    """You are an AI CyberSecurity Program Advisor. Your goal is to engage with leads and get them to book a video call for an in-person sales meeting. You have access to a list of leads and can send emails.
+st.write("Welcome! Please provide your details to schedule a video call to discuss our cybersecurity program.")
 
-You have access to the following tools:
+# Collect user information
+name = st.text_input("Your Name")
+email = st.text_input("Your Email Address")
+preferred_date = st.date_input("Preferred Date for the Video Call")
 
-{tools}
+# Zoom meeting details
+zoom_link = """
+Lawrence Emenike is inviting you to a scheduled Zoom meeting.
 
-Use the following format:
+Topic: Lawrence Emenike's Zoom Meeting
+Time: Aug 10, 2024 05:00 PM West Central Africa
 
-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: [Insert your final response here]
-Begin!
+Join Zoom Meeting
+https://us04web.zoom.us/j/73793374638?pwd=S0TEJ30da7dhQ8viOdafMzPfCVzoLJ.1
 
-Question: {input}
-Thought: Let's approach this step-by-step:
-{agent_scratchpad}"""
-)
+Meeting ID: 737 9337 4638
+Passcode: 9cNPkn
+"""
 
-# Create the React agent
-agent = create_react_agent(
-    llm=local_llm, 
-    tools=tools, 
-    prompt=prompt
-)
+if st.button("Schedule Video Call"):
+    if name and email and preferred_date:
+        # Compose the email
+        subject = "Invitation to Discuss Cybersecurity Program"
+        body = f"""
+        Hi {name},
 
-# Create the agent executor
-agent_executor = AgentExecutor.from_agent_and_tools(
-    agent=agent, 
-    tools=tools, 
-    verbose=True, 
-    memory=ConversationBufferMemory()
-)
+        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.
 
-# Streamlit interface
-st.title("AI CyberSecurity Program Advisor Demo")
+        Here are the details:
+
+        Preferred Date: {preferred_date}
+        Zoom Meeting Link: {zoom_link}
 
-st.write("This demo showcases an AI agent that can engage with leads and attempt to book video calls for sales meetings.")
+        Looking forward to our discussion!
 
-# Start a conversation with a predefined lead
-lead = LEADS[0]
-initial_message = f"Hello {lead['name']}, I'd like to discuss our cybersecurity program with {lead['company']}. Are you available for a quick video call?"
+        Best regards,
+        Lawrence Emenike
+        """
 
-if st.button("Start Conversation"):
-    with st.spinner("AI is generating a response..."):
-        response = agent_executor.invoke({"input": initial_message})
-        st.write("AI Response:")
-        st.write(response)  # Display the full response dictionary
+        # Send the email
+        send_email(email, subject, body)
+        st.success("Your video call has been scheduled, and an email has been sent!")
+    else:
+        st.error("Please fill in all the required fields.")
 
 st.sidebar.title("About")
-st.sidebar.info("This is a demo of an AI CyberSecurity Program Advisor using an open-source LLM and LangChain. It's designed to engage with leads and attempt to book video calls for sales meetings.")
+st.sidebar.info("This is a demo of an AI CyberSecurity Program Advisor. It's designed to help you schedule a video call to discuss our cybersecurity program.")