Remove image part

src/app.py

@@ -6,7 +6,6 @@ from utils.ui_components import (
     render_header,
     render_controls,
     render_image_columns,
-    render_example_gallery,
     render_info_sections
 )
 
@@ -44,6 +43,5 @@ render_image_columns(
     enhanced_image=st.session_state.get('enhanced_image')
 )
 
-# Render example gallery and information sections
-render_example_gallery()
+# Render information sections
 render_info_sections()

src/utils/ui_components.py

@@ -73,61 +73,6 @@ def render_image_columns(base_image=None, enhanced_image=None):
         if enhanced_image:
             st.image(enhanced_image, use_column_width=True)
 
-def render_example_gallery():
-    st.markdown("""
-    <div class="dark-theme">
-        <h2>🎆 Example Gallery</h2>
-        <p>Compare original and enhanced versions for each style:</p>
-    </div>
-    """, unsafe_allow_html=True)
-
-    try:
-        output_dir = Path("Outputs")
-        original_dir = output_dir
-        enhanced_dir = output_dir / "Color_Enhanced"
-
-        if enhanced_dir.exists():
-            original_images = {
-                Path(f).stem.split('_example')[0]: f 
-                for f in original_dir.glob("*.webp") 
-                if '_example' in f.name
-            }
-            enhanced_images = {
-                Path(f).stem.split('_example')[0]: f 
-                for f in enhanced_dir.glob("*.webp") 
-                if '_example' in f.name
-            }
-
-            styles = [
-                ("dhoni", "Dhoni Style"),
-                ("mickey_mouse", "Mickey Mouse Style"),
-                ("balloon", "Balloon Style"),
-                ("lion_king", "Lion King Style"),
-                ("rose_flower", "Rose Flower Style")
-            ]
-
-            for style_key, style_name in styles:
-                if style_key in original_images and style_key in enhanced_images:
-                    st.markdown(f"### {style_name}")
-                    col1, col2 = st.columns(2)
-                    
-                    with col1:
-                        st.image(
-                            str(original_images[style_key]),
-                            caption="Original",
-                            use_column_width=True
-                        )
-                    with col2:
-                        st.image(
-                            str(enhanced_images[style_key]),
-                            caption="Color Enhanced",
-                            use_column_width=True
-                        )
-                    st.markdown("<hr>", unsafe_allow_html=True)
-
-    except Exception as e:
-        st.error(f"Error loading example gallery: {str(e)}")
-
 def render_info_sections():
     col1, col2 = st.columns(2)