Add example image

src/app.py

@@ -6,7 +6,8 @@ from utils.ui_components import (
     render_header,
     render_controls,
     render_image_columns,
-    render_info_sections
+    render_info_sections,
+    render_example_gallery
 )
 
 # Initialize the application
@@ -44,4 +45,7 @@ render_image_columns(
 )
 
 # Render information sections
-render_info_sections()
+render_info_sections()
+
+# Display example gallery
+render_example_gallery()

src/utils/ui_components.py

@@ -58,6 +58,8 @@ def render_controls(style_names):
             index=0
         )
         
+        st.warning("⚠️ Please note: Image generation may take several hours in the free environment due to limited computational resources.")
+        
         return prompt, selected_style
 
 def render_image_columns(base_image=None, enhanced_image=None):
@@ -66,12 +68,12 @@ def render_image_columns(base_image=None, enhanced_image=None):
     with col1:
         st.markdown("### Original Style")
         if base_image:
-            st.image(base_image, use_column_width=True)
+            st.image(base_image, use_container_width=True)
     
     with col2:
         st.markdown("### Color Enhanced")
         if enhanced_image:
-            st.image(enhanced_image, use_column_width=True)
+            st.image(enhanced_image, use_container_width=True)
 
 def render_info_sections():
     col1, col2 = st.columns(2)
@@ -123,4 +125,23 @@ def render_info_sections():
                 <li>Preserve artistic style while boosting vibrancy</li>
             </ul>
         </div>
-        """, unsafe_allow_html=True)
+        """, unsafe_allow_html=True)
+def render_example_gallery():
+    st.markdown("""
+    <div class="dark-theme">
+        <h2>🖼️ Example Gallery</h2>
+        <p>Here are some examples of AI-generated artwork:</p>
+    </div>
+    """, unsafe_allow_html=True)
+    
+    # Get all jpg files from the generated_images directory
+    image_dir = Path(__file__).parent.parent / "generated_images"
+    image_files = list(image_dir.glob("*.jpg"))
+    
+    # Create columns for the gallery
+    cols = st.columns(3)
+    
+    # Display images in the columns
+    for idx, image_file in enumerate(image_files):
+        with cols[idx % 3]:
+            st.image(str(image_file), caption=image_file.stem.replace("_", " ").title(), use_container_width=True)