Update app.py

app.py

@@ -444,14 +444,14 @@ def main_app():
 
     # Display modal if an image is selected
     if st.session_state.selected_image:
-        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']}")
-           st.write(f"**Realism:** {st.session_state.selected_image['realism']}")
-           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", key=f"download_overlay_{uuid.uuid4()}")
+        with st.container(border=True):
+            st.image(cloud_storage_url, use_column_width=True)  # Display high-res image from URL
+            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", key=f"download_high_res_{uuid.uuid4()}")
 
         if st.button("Close"):
           st.session_state.selected_image = None # close the modal when "close" is clicked