import os
from api_key import apikey

import streamlit as st
from langchain.llms import OpenAI
from langchain.prompts import PromptTemplate
from langchain.memory import ConversationBufferMemory
from langchain.chains import LLMChain

os.environ['OPENAI_API_KEY'] = apikey

#APP FRAMEWORK
st.title('🦜️🔗 COVER-LETTER GPT CREATOR')
prompt = st.text_input("Enter the JD on which you want to generate Cover Letter")

#PROMPT RESUME TEMPLATE
resume_template = PromptTemplate(
    input_variables = ['topic'], 
    template='write me a Cover Letter about the Job Discription of {topic}'
)

# #COVER LETTER RESUME
# letter_template = PromptTemplate(
#     input_variables = ['title'], 
#     template='write me a cover letter based on the following resume: {title}'
# )

# MEMORY
resume_memory = ConversationBufferMemory(input_key='topic', memory_key='chat_history')
# letter_memory = ConversationBufferMemory(input_key='title', memory_key='chat_history')

# LLMS
llm = OpenAI(temperature=0.9) 
resume_chain = LLMChain(llm=llm, prompt=resume_template, verbose=False, output_key='title', memory=resume_memory)
# letter_chain = LLMChain(llm=llm, prompt=letter_template, verbose=True, output_key='script', memory=letter_memory)
# Sequential_Chain = SequentialChain(chains=[resume_chain, letter_chain],
#                                    input_variables=['topic'],
#                                    output_variable=['title','script'])

#SHOW STUFF ON THE SCREEN
if prompt:
    response = resume_chain.run(prompt,)
    st.write(response)
    # st.write(response['title'])
    # st.write(response['script'])