Update app.py

app.py

@@ -334,10 +334,14 @@ def main_app():
                               # Store image data in database
                               store_image_data_in_db(st.session_state.current_user, prompt, aspect_ratio, realism, cloud_storage_url, thumbnail_url)
                               st.success("Image stored to database successfully!")
-                              st.image(image, caption="Generated Image", use_column_width=True) # Display high res image on top
-                              download_path = download_image(image_url)
-                              if download_path:
-                                 st.download_button(label="Download Image", data = open(download_path, "rb"), file_name = f"image.png")
+                              with st.container(border=True):
+                                st.image(image, use_column_width=True)  # Display high res image on top
+                                st.write(f"**Prompt:** {prompt}")
+                                st.write(f"**Aspect Ratio:** {aspect_ratio}")
+                                st.write(f"**Realism:** {realism}")
+                                download_path = download_image(image_url)
+                                if download_path:
+                                  st.download_button(label="Download Image", data = open(download_path, "rb"), file_name = f"image.png")
                           else:
                                st.error("Failed to upload image to cloud storage.")
                       else:
@@ -393,8 +397,8 @@ def main_app():
                 if image_data.get('thumbnail_url') and image_data.get('image_url'):
                   if st.button("More", key = f"more_{i}"):
                       st.session_state.selected_image = image_data
-                      
-                  st.image(image_data['thumbnail_url'], width = 150)
+                  st.image(image_data['thumbnail_url'], width = 150) #display thumbnail
+
                 else:
                     st.image(image_data['image_url'], width = 150)
                     st.write(f"**Prompt:** {image_data['prompt']}")
@@ -407,7 +411,7 @@ def main_app():
 
     # Display modal if an image is selected
     if st.session_state.selected_image:
-       with st.container(border = True):
+        with st.container(border = True):
            st.image(st.session_state.selected_image['image_url'], use_column_width=True)
            st.write(f"**Prompt:** {st.session_state.selected_image['prompt']}")
            st.write(f"**Aspect Ratio:** {st.session_state.selected_image['aspect_ratio']}")
@@ -415,13 +419,13 @@ def main_app():
            download_path = download_image(st.session_state.selected_image['image_url'])
            if download_path:
               st.download_button(label="Download Image", data = open(download_path, "rb"), file_name = f"image.png")
-       if st.button("Close"):
+
+        if st.button("Close"):
           st.session_state.selected_image = None # close the modal when "close" is clicked
 
     # 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()
 else: