Spaces:
Runtime error
Runtime error
File size: 1,607 Bytes
603c081 1a06de0 603c081 1a06de0 603c081 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
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'])
|