updated app

app.py

@@ -1,9 +1,9 @@
 """Python file to serve as the frontend"""
-#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
 import streamlit as st
 from streamlit_chat import message
 import os
 import openai
+import time
 
 #Import LLM wrapper
 from langchain.llms import OpenAI
@@ -30,7 +30,7 @@ def load_chain(api_key):
 
     second_prompt = PromptTemplate(
         input_variables=["program"],
-        template= '''Produce the python3 code for each step of the {program} described. Use python3 style. Be concise in the code and opinionated about framework choice.'''
+        template= '''Write the python3 code for each step of the {program} described. Use python3 style. Be concise in the code and opinionated about framework choice.'''
     )
 
     chain_two = LLMChain(llm=llm, prompt=second_prompt)
@@ -67,8 +67,11 @@ user_input = get_text()
 
 if user_input:
     chain = load_chain(openai_api_key)
-    output = chain.run(input=user_input)
-
+    
+    with st.spinner('Wait for it...'):
+        output = chain.run(input=user_input)
+        time.sleep(5)
+    st.success('Done!')
     st.session_state.past.append(user_input)
     st.session_state.generated.append(output)