Update app.py

app.py

@@ -102,8 +102,8 @@ def main():
     st.header("Generate image with ControllNet")
     with st.sidebar:
         st_lottie(lottie_penguin, height=200)
-        choose = option_menu("Generate image", ["Upload", "Canvas", "Image Gallery"],
-                            icons=['cloud-upload', 'file-plus', 'collection'],
+        choose = option_menu("Generate image", ["Upload", "Canvas"],
+                            icons=['cloud-upload', 'file-plus'],
                             menu_icon="infinity", default_index=0,
                             styles={
                                 "container": {"padding": ".0rem", "font-size": "14px"},
@@ -167,11 +167,13 @@ def main():
 
     elif choose == 'Canvas':
         with st.form(key='canvas_generate_form'):
+            
             # Specify canvas parameters in application
             stroke_width = st.sidebar.slider("Stroke width: ", 1, 25, 3)
             stroke_color = st.sidebar.color_picker("Stroke color hex: ")
             bg_color = st.sidebar.color_picker("Background color hex: ", "#eee")
             realtime_update = st.sidebar.checkbox("Update in realtime", True)
+            
             # Create a canvas component
             col31, col32 = st.columns(2)
             with col31:
@@ -209,7 +211,7 @@ def main():
                     n_prompt = st.text_input(label="Negative Prompt",
                                           value='longbody, lowres, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality')
 
-            # Do something interesting with the image data and paths
+            # Generate image from canvas
             generate_button = st.form_submit_button(label='Generate Image')
             if generate_button:
                 if canvas_result.image_data is not None:
@@ -219,19 +221,18 @@ def main():
                         H, W, C = input_image.shape
                         output_image = cv2.resize(results[0], (W, H), interpolation=cv2.INTER_AREA)
                         col32.image(output_image, channels='RGB', width=None, clamp=True, caption='Generated image')
-    
-    elif choose == "Image Gallery":
-        with st.expander('Image gallery', expanded=True):
-            col01, col02, = st.columns(2)
-            with col01:
-                st.image('demo/example_1.jpg', caption="Sport car")
-                st.image('demo/example_2.jpg', caption="Dog house")
-                st.image('demo/example_3.jpg', caption="Guitar")
-            with col02:
-                st.image('demo/example_4.jpg', caption="Sport car")
-                st.image('demo/example_5.jpg', caption="Dog house")
-                st.image('demo/example_6.jpg', caption="Guitar")
-
+                        
+            # Image gallery         
+            with st.expander('Image gallery', expanded=True):
+                col01, col02, = st.columns(2)
+                with col01:
+                    st.image('demo/example_1.jpg', caption="Sport car")
+                    st.image('demo/example_2.jpg', caption="Dog house")
+                    st.image('demo/example_3.jpg', caption="Guitar")
+                with col02:
+                    st.image('demo/example_4.jpg', caption="Sport car")
+                    st.image('demo/example_5.jpg', caption="Dog house")
+                    st.image('demo/example_6.jpg', caption="Guitar")
         
 if __name__ == '__main__':
     main()