Update app.py

app.py

@@ -383,16 +383,16 @@ def main_app():
 
         cols = st.columns(num_images_to_display)
         for i, image_data in enumerate(images_for_window):
-          with cols[i]:
-             if image_data.get('thumbnail_url'):
-                  if st.button(image_data.get('prompt') ,key=f"thumbnail_{i}"):
-                    st.image(image_data['image_url'], width = 200)
-                  st.image(image_data['thumbnail_url'], width = 150)
-                  st.write(f"**Prompt:** {image_data['prompt']}")
-                  st.write(f"**Aspect Ratio:** {image_data['aspect_ratio']}")
-                  st.write(f"**Realism:** {image_data['realism']}")
-                  st.markdown("---")
-             else:
+            with cols[i]:
+              if image_data.get('thumbnail_url'):
+                 with st.expander("More", expanded=False):
+                      st.image(image_data['image_url'], width = 200)
+                      st.write(f"**Prompt:** {image_data['prompt']}")
+                      st.write(f"**Aspect Ratio:** {image_data['aspect_ratio']}")
+                      st.write(f"**Realism:** {image_data['realism']}")
+                      st.markdown("---")
+                 st.image(image_data['thumbnail_url'], width = 150)
+              else:
                  st.image(image_data['image_url'], width = 150)
                  st.write(f"**Prompt:** {image_data['prompt']}")
                  st.write(f"**Aspect Ratio:** {image_data['aspect_ratio']}")
@@ -404,7 +404,6 @@ def main_app():
     # Logout button
     if st.button("Logout", on_click=logout_callback):
         pass
-
 # Render the appropriate screen based on login status
 if st.session_state.logged_in:
     main_app()