Update app.py

app.py

@@ -121,15 +121,15 @@ def main():
                 # Store descriptions in session state
                 st.session_state["all_output_texts"] = all_output_texts
 
-    # Check if descriptions are available in session state
-    if "all_output_texts" in st.session_state and st.session_state["all_output_texts"]:
-        st.write("Generate story")
-        generate_story_button = st.button("Generate Story", key="generate_story")
-
-        if generate_story_button:
-            story = generate_story(st.session_state["all_output_texts"])
-            st.write("Generated Story:")
-            st.write(story)
+            # Check if descriptions are available in session state
+            if "all_output_texts" in st.session_state and st.session_state["all_output_texts"]:
+                st.write("Generate story")
+                generate_story_button = st.button("Generate Story", key="generate_story")
+        
+                if generate_story_button:
+                    story = generate_story(st.session_state["all_output_texts"])
+                    st.write("Generated Story:")
+                    st.write(story)
 
 if __name__ == "__main__":
     main()