modified 1st summary button functionality

app.py

@@ -35,24 +35,26 @@ st.markdown(""" #### LorSor helps you through a simple 3 stage process.
 st.markdown('#')
 col1, col2 = st.columns(2)
 
-with col1:
-    col1.header("Step 1:")
-    raw_text = st.text_area('Paste the full article text to summarize here...')
-    st.button("Summarize this")
-
+@st.cache
 def get_response(input_text):
     batch = tokenizer([input_text],truncation=True,padding='longest',max_length=1024, return_tensors="pt").to('cpu')
     gen_out = model.generate(**batch,max_length=128,num_beams=5, num_return_sequences=1, temperature=1.5)
     output_text = tokenizer.batch_decode(gen_out, skip_special_tokens=True)
     return output_text
 
-if len(raw_text) < 10:
-    summary = "Lorem Ipsum is a long and boring piece of old latin text. What it means i have no idea"
-else:
-    summary = get_response(raw_text)
+with col1:
+    col1.header("Step 1:")
+    raw_text = st.text_area('Paste the full article text to summarize here...')
+    summary_button = st.button("Summarize this")
 
+    if summary_button:
+        if len(raw_text) < 10:
+            summary = "<< Add some text in ( Step 1 ) for me to summarize >>"
+        else:
+            summary = get_response(raw_text)
 
 with col2:
     col2.header("Step 2:")
     y = st.text_area("Here is the completed summary for you to edit", summary)
-    st.button("Submit edits")
+    st.button("Submit edits")
+    st.balloons()