Update app.py

app.py

@@ -282,6 +282,9 @@ if 'text_block' not in st.session_state:
 if 'on_topic' not in st.session_state:
     st.session_state['on_topic'] = ''
 
+if 'image_paths' not in st.session_state:
+    st.session_state['image_paths'] = []
+
 st.set_page_config(page_title='Clarifai Story Teller', layout='wide')
 
 
@@ -319,14 +322,15 @@ with tab2:
         cols = st.columns(4)
         with st.spinner('Generating Content...'):
             for i, prompt in enumerate(prompts):
-                image, img_error = generate_image(prompt+st.session_state.on_topic )
+                image_path, img_error = generate_image(prompt+st.session_state.on_topic )
                 audio, audio_error = generate_audio(prompt)
     
                 with cols[i % 4]:
                     if img_error:
                         st.error(img_error)
                     else:
-                        st.image(image, prompt, use_column_width=True)
+                        st.session_state['image_paths'].append(image_path)
+                        st.image(image_path, prompt, use_column_width=True)
     
                     if audio_error:
                         st.error(audio_error)
@@ -335,8 +339,14 @@ with tab2:
 
 # Tab 3: User Input and Results
 with tab3:
-    st.header("Interactive User Input")
-    user_input = st.text_input("Enter some information")
-    st.write(f"You entered: {user_input}")
-
-
+    if 'image_paths' in st.session_state:
+        for i in range(0, len(st.session_state['image_paths']), 4):
+            cols = st.columns(4)
+            for j in range(4):
+                idx = i + j
+                if idx < len(st.session_state['image_paths']):
+                    with cols[j]:
+                        st.image(st.session_state['image_paths'][idx], use_column_width=True)
+                        if st.button(f"Details {idx}", key=f"button_{idx}"):
+                            st.write(f"You clicked on image {idx}")
+                            # Add more actions for the click event here