Update app.py

app.py

@@ -26,8 +26,9 @@ def main():
         text_input = st.text_input("Enter the text:")
         if st.button("Generate Story and Image") and text_input:
             #generate_image(text_input)
+            generated_text = generate_mask_from_result(text_input)
             st.success(f'Generated Caption: {text_input}')
-            generate_mask_from_result(text_input)
+            st.success(f'Generated Text: {generated_text}')
             
 
     # Input URL
@@ -36,8 +37,9 @@ def main():
         if st.button("Generate Story and Image") and image_url:
             image_text = image_to_text_from_url(image_url)
             #generate_image(image_text)
+            generated_text = generate_mask_from_result(image_text)
             st.success(f'Generated Caption: {image_text}')
-            generate_mask_from_result(image_text)
+            st.success(f'Generated Text: {generated_text}')
             
 
 def image_to_text_from_file(uploaded_file):