ice_breaker.py

from dotenv import load_dotenv
from langchain.chains import LLMChain
from langchain_openai import ChatOpenAI
from langchain_core.prompts import PromptTemplate
from agents.linkedin_lookup_agent import lookup as linkedin_lookup_agent
from third_parties.linkedin import scrape_linkedin_profile
import streamlit as st
import json, os


if __name__ == "__main__":
    load_dotenv()

    # Streamlit app
    st.subheader("Get a summary of the person you'd to break the ice with")

    # Get OpenAI API key, PROXYCURL API key and SERPAPI KEY, and source document input
    with st.sidebar:
        openai_api_key = st.text_input("OpenAI API key", type="password")
        proxy_curl_api_key = st.text_input("Proxy Curl API key", type="password")
        serpapi_api_key = st.text_input("Serp API key", type="password")

    os.environ["OPENAI_API_KEY"] = openai_api_key
    os.environ["PROXYCURL_API_KEY"] = proxy_curl_api_key
    os.environ["SERPAPI_API_KEY"] = serpapi_api_key
            
    print("Hello LangChain!")

    default_name = "Ozer Ozdal"  # Default value for the Name input

    name = st.text_input("Name", default_name)
    
    default_info = """Ozer Ozdal is an ML Engineer with many years of expertise in designing and implementing machine learning solutions, predictive analytics, and data science applications.
    Demonstrated proficiency in model explainability, including LLM models, and a robust comprehension of machine learning algorithms for classification, regression, and clustering, driving toward strategic business objectives. Skilled in leveraging AWS cloud architectures. 
    Proficient in employing data-centric methodologies to extract insights and drive decision-making."""

    information = st.text_area("Provide some information about the person you would like to search", default_info)

    if st.button("Submit"):
        try:

            if name == default_name:
                # If the user searches "Ozer Ozdal", do not send API requests.
                # Fetch the information from the results/summary.json file as a cost-conscious application
                with open("results/summary.json") as f:
                    summary = json.load(f)
                    st.success(summary['text'])

            elif name != default_name and openai_api_key and proxy_curl_api_key and serpapi_api_key:

                linkedin_profile_url = linkedin_lookup_agent(name=name, openai_api_key=openai_api_key)

                summary_template = """
                given the information {information} about a person I want you to create:
                1. A short summary
                2. two interesting facts about them
                """

                summary_prompt_template = PromptTemplate(
                    input_variables=["information"], template=summary_template
                )

                # temperature will decide how creative the language model will be.
                # 0 means it won't be creative
                llm = ChatOpenAI(temperature=0, model_name="gpt-3.5-turbo", openai_api_key=openai_api_key)

                chain = LLMChain(llm=llm, prompt=summary_prompt_template)

                linkedin_data = scrape_linkedin_profile(
                    linkedin_profile_url=linkedin_profile_url, 
                    mock=False,
                    proxy_curl_api_key=proxy_curl_api_key)

                summary = chain.invoke(input={"information": linkedin_data})

                # Return the short summary and two interesting facts about the person
                st.success(summary['text'])

            else:
                st.error(f"""
                        Sorry we cannot provide a summary for {name}. \n
                        Please provide your OpenAl API, ProxyCurl API and SerpAPI API keys. \n
                        You can only search for the name Ozer Ozdal for free :)
                        """)

        except Exception as e:
            st.error(f"An error occurred: {e}")

    # Unset environment variables
    if 'OPENAI_API_KEY' in os.environ:
        del os.environ['OPENAI_API_KEY']
            
    if 'PROXYCURL_API_KEY' in os.environ:
        del os.environ['PROXYCURL_API_KEY']

    if 'SERPAPI_API_KEY' in os.environ:
        del os.environ['SERPAPI_API_KEY']