Update app.py

app.py

@@ -5,9 +5,8 @@ import os
 co = cohere.Client(os.getenv('COHERE_API_KEY')) 
 
 # Initialization
-if 'output' not in st.session_state:
-    st.session_state['output'] = 'Output:'
-
+if not output:
+    output = ""
 def generate_hashtags(input):
     if len(input) == 0:
         return None
@@ -23,7 +22,7 @@ def generate_hashtags(input):
     stop_sequences=["--"], 
     return_likelihoods='NONE') 
     
-    st.session_state['output'] = response.generations[0].text
+    output = response.generations[0].text
 
 
 st.title('Positive Reframing Generator')
@@ -31,4 +30,4 @@ st.write('''This is a simple **Streamlit** app that generates the input sentence
 
 input = st.text_area('Enter your post title caption here', height=100)
 st.button('Reframe', on_click = generate_hashtags(input))
-st.write(st.session_state.output)
+st.write('{output}')