need to fix prompts

app.py

@@ -4,7 +4,6 @@ from streamlit_chat import message
 import os
 import openai
 
-
 #Import LLM wrapper
 from langchain.llms import OpenAI
 #Import Prompt Template
@@ -16,37 +15,40 @@ from langchain.chains import SimpleSequentialChain
 
 
 
-def load_chain():
+def load_chain(api_key):
     os.environ["OPENAI_API_KEY"] = api_key
     llm = OpenAI(temperature=0.3)
     #user_in = input("Please describe your Python project in one to two sentences:  ")
 
     first_prompt = PromptTemplate(
         input_variables=["user_in"],
-        template= "You are a senior Python Engineer. List five steps required to develop the software project specified. Use the common libraries available in Python3. Be verbose in the code and opinionated about framework choice.:\n\n {user_in}"
+        template= "You are a senior Python Engineer. List five steps required to develop the software project specified. Only include the 5 steps as a high level outline. Explain that this is only a guide. \n\n {user_in}"
     )
     #First chain
     chain = LLMChain(llm=llm, prompt=first_prompt)
 
     second_prompt = PromptTemplate(
         input_variables=["project"],
-        template= "Produce the python3 code for each step of the software {project} described. Use appropriate style, classes and variables. Be verbose."
+        template= "As the senior engineer, Produce the python3 code for each step of the software {project} described. Use appropriate style, classes and variables. Be verbose in the code and opinionated about framework choice.:"
     )
 
     chain_two = LLMChain(llm=llm, prompt=second_prompt)
 
     overall_chain = SimpleSequentialChain(chains=[chain, chain_two], verbose=True)
 
-    code_output = overall_chain.run(user_in)
 
-    return chain
+    return overall_chain
+
+
 
-chain = load_chain()
 
 # From here down is all the StreamLit UI.
 st.set_page_config(page_title="Python Project Generator", page_icon=":robot:")
 st.header("Python Project Generator")
 
+st.write("Enter your API key below:")
+openai_api_key = st.text_input("openai_api_key")
+
 if "generated" not in st.session_state:
     st.session_state["generated"] = []
 
@@ -56,18 +58,14 @@ if "past" not in st.session_state:
 
 def get_text():
     user_input = st.text_input("Please describe your python project in 1-2 sentences", key="input")
-    return user_input, get_key
-
-def get_key():
-    get_key = st.text_input("Please enter your OpenAPI API Key", key="input")
-    return get_key
+    return user_input
 
 
-api_key = get_key()
 user_input = get_text()
 
 
 if user_input:
+    chain = load_chain(openai_api_key)
     output = chain.run(input=user_input)
 
     st.session_state.past.append(user_input)