Update app.py

app.py

@@ -2,6 +2,9 @@ import gradio as gr
 import os
 import sys
 from pathlib import Path
+import os
+import numpy as np
+from gradio import *
 
 models = [
     {"name": "Stable Diffusion 2", "url": "stabilityai/stable-diffusion-2-1"},
@@ -78,6 +81,48 @@ with gr.Blocks(css='style.css') as myface:
         magic4 = gr.Textbox(label="Generated Prompt", lines=2)
         magic5 = gr.Textbox(label="Generated Prompt", lines=2)
         magic6 = gr.Textbox(label="Generated Prompt", lines=2)
+        
+
+# Set up the GUI
+with Blocks() as demo:
+    # Create a State variable to store the selected image
+    img = State()
+
+    # Define a function to load the image from the local storage directory
+    def load_image(directory):
+        # Get the list of files in the directory
+        filenames = os.listdir(directory)
+        
+        # Select the first image file (e.g. "image1.jpg")
+        filename = filenames[0]
+        
+        # Load the image using numpy.load()
+        img_data = np.load(os.path.join(directory, filename))
+        
+        # Convert the image data to a NumPy array
+        img = np.array(img_data)
+        return img
+
+    # Create a Gallery widget to display the loaded image
+    gallery = Gallery(directory='home/downloads/images')
+
+    # Add the image to the gallery
+    gallery.add(img)
+
+    # Create a Button widget to trigger the darkening of the image
+    darken_btn = Button("Darken Image")
+
+    # Define a function to darken the image
+    def darken_img(img):
+        # Darken the image by multiplying each pixel value by 0.8
+        darkened_img = np.round(img * 0.8).astype(np.uint8)
+        return darkened_img
+
+    # Connect the Button widget to the darken_img function
+    darken_btn.click(darken_img, [img], [gallery])
+
+# Launch the GUI
+demo.launch()    
 
     model_name1.change(set_model, inputs=model_name1, outputs=[output1, output2, output3, output4, output5, output6])