Update app.py

app.py

@@ -1,24 +1,9 @@
 import gradio as gr  
 import numpy as np  
 from PIL import Image  
-import torch  
-from basicsr.archs.rrdbnet_arch import RRDBNet  
-from realesrgan import RealESRGANer  
+import cv2  
 
-def load_model():  
-    model = RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64, num_block=23, num_grow_ch=32, scale=4)  
-    upsampler = RealESRGANer(  
-        scale=4,  
-        model_path='https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth',  
-        model=model,  
-        tile=0,  
-        tile_pad=10,  
-        pre_pad=0,  
-        half=False  
-    )  
-    return upsampler  
-
-def super_resolve_image(input_image):  
+def super_resolve_image(input_image, scale_factor=4):  
     if input_image is None:  
         return None  
     
@@ -27,17 +12,30 @@ def super_resolve_image(input_image):
         if isinstance(input_image, Image.Image):  
             input_image = np.array(input_image)  
         
-        # Ensure the image is in RGB format  
+        # Ensure image is in RGB  
         if len(input_image.shape) == 2:  
-            input_image = np.stack([input_image]*3, axis=-1)  
+            input_image = cv2.cvtColor(input_image, cv2.COLOR_GRAY2RGB)  
         
-        # Load model  
-        upsampler = load_model()  
+        # Upscaling methods  
+        upscaling_methods = [  
+            cv2.INTER_LINEAR,   # Bilinear interpolation  
+            cv2.INTER_CUBIC,    # Bicubic interpolation  
+            cv2.INTER_LANCZOS4  # Lanczos interpolation  
+        ]  
         
-        # Super-resolve  
-        output, _ = upsampler.enhance(input_image)  
+        # Try different interpolation methods  
+        for method in upscaling_methods:  
+            try:  
+                upscaled = cv2.resize(  
+                    input_image,   
+                    (input_image.shape[1] * scale_factor, input_image.shape[0] * scale_factor),   
+                    interpolation=method  
+                )  
+                return upscaled  
+            except Exception as e:  
+                print(f"Interpolation method failed: {method}")  
         
-        return output  
+        return None  
     
     except Exception as e:  
         print(f"Super-resolution error: {e}")  
@@ -45,17 +43,24 @@ def super_resolve_image(input_image):
 
 def create_gradio_interface():  
     with gr.Blocks() as demo:  
-        gr.Markdown("# 🖼️ Image Super-Resolution with Real-ESRGAN")  
+        gr.Markdown("# 🖼️ Simple Image Upscaling")  
         
         with gr.Row():  
             input_image = gr.Image(label="Input Image", type="pil")  
-            output_image = gr.Image(label="Super-Resolved Image")  
+            output_image = gr.Image(label="Upscaled Image")  
+        
+        with gr.Row():  
+            scale_dropdown = gr.Dropdown(  
+                choices=[2, 4, 8],   
+                value=4,   
+                label="Upscale Factor"  
+            )  
         
         enhance_btn = gr.Button("Enhance Image Resolution")  
         
         enhance_btn.click(  
             fn=super_resolve_image,   
-            inputs=input_image,   
+            inputs=[input_image, scale_dropdown],   
             outputs=output_image  
         )