main

app.py

@@ -0,0 +1,46 @@
+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('🦜️🔗 RESUME GPT CREATOR')
+prompt = st.text_input("Enter the JD on which you want to generate Resume")
+
+#PROMPT RESUME TEMPLATE
+resume_template = PromptTemplate(
+    input_variables = ['topic'], 
+    template='write me a resume 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'])
+