Update app.py

app.py

@@ -0,0 +1,57 @@
+import cv2
+import numpy as np
+import gradio as gr
+
+# Function to process the PSR moon image
+def process_image(input_image):
+    # Convert PIL Image to a NumPy array
+    image = np.array(input_image)
+
+    # Convert image to grayscale
+    gray_image = cv2.cvtColor(image, cv2.COLOR_RGB2GRAY)
+
+    # Apply CLAHE (Contrast Limited Adaptive Histogram Equalization) for enhanced contrast
+    clahe = cv2.createCLAHE(clipLimit=2.0, tileGridSize=(8, 8))  # Increased clipLimit for stronger contrast
+    cl1 = clahe.apply(gray_image)
+
+    # Create a mask for darker regions (e.g., craters) based on pixel intensity
+    _, mask = cv2.threshold(gray_image, 80, 255, cv2.THRESH_BINARY_INV)  # Adjusted threshold for more precise mask
+
+    # Apply CLAHE only to the masked regions (darker areas)
+    brightened_craters = cv2.bitwise_and(cl1, cl1, mask=mask)
+
+    # Combine brightened craters with the original image using weighted addition
+    combined = cv2.addWeighted(brightened_craters, 1.3, cl1, 0.9, 0)  # Adjusted weights for better blending
+
+    # Further brighten the final image by increasing alpha (contrast) and beta (brightness)
+    brightened_image = cv2.convertScaleAbs(combined, alpha=1.2, beta=30)  # Increased contrast and moderate brightness
+
+    # Define a mask for the darker spots that need additional brightening
+    dark_spots_mask = cv2.inRange(brightened_image, 0, 110)  # Refined range for more precise dark spot targeting
+
+    # Increase the brightness of only the dark areas by blending with a brighter image
+    brightened_dark_spots = cv2.addWeighted(brightened_image, 1.0, dark_spots_mask, 0.4, 40)
+
+    # Combine the brightened dark spots with the rest of the image
+    final_image = cv2.bitwise_or(brightened_image, brightened_dark_spots)
+
+    return final_image
+
+# Create a Gradio app using Blocks
+with gr.Blocks() as app:
+    gr.Markdown("## Enhanced Lunar PSR Image")
+
+    # Image upload component
+    img_input = gr.Image(label="Upload PSR Moon Image", type="pil")
+
+    # Button to trigger processing
+    process_button = gr.Button("Process Image")
+
+    # Output image component to display the processed image
+    img_output = gr.Image(label="Enhanced Image")
+
+    # Define the interaction between components
+    process_button.click(process_image, inputs=img_input, outputs=img_output)
+
+# Launch the app
+app.launch()