Update app.py

app.py

@@ -44,19 +44,17 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             # Create a sketchpad for drawing
-            canvas = gr.Sketchpad(label="Draw Here")  
+            canvas = gr.Sketchpad(label="Draw Here")
             clear_btn = gr.Button("Clear")
             generate_btn = gr.Button("Generate Image")
         
         with gr.Column():
-            # Create color buttons for different terrains
-            brush_color = gr.Variable(default="black")  # Variable to store brush color
-
-            def change_color(color):
-                brush_color.set(color)  # Update the brush color
-
-            for color_name, color in color_mapping.items():
-                gr.Button(color_name, variant="primary").click(fn=change_color, inputs=color, outputs=None)
+            # Create a dropdown for different terrain colors
+            color_dropdown = gr.Dropdown(
+                label="Select Brush Color",
+                choices=list(color_mapping.keys()),
+                value="Water"  # Default color
+            )
 
             output_image = gr.Image(label="Generated Image", type="numpy")