Spaces:

syedaoon
/

ZeroIG

Running

App Files Files Community

syedaoon commited on Jul 27

Commit

4e9fc8c

verified ·

1 Parent(s): 9aea794

Update app.py

Browse files

Files changed (1) hide show

app.py +106 -39

app.py CHANGED Viewed

@@ -11,6 +11,10 @@ import time
 app = Flask(__name__)
 HTML_TEMPLATE = """
 <!DOCTYPE html>
 <html>
@@ -35,6 +39,7 @@ HTML_TEMPLATE = """
         .error { color: red; }
         .processing { color: orange; font-weight: bold; }
         .method-info { background: #e3f2fd; padding: 15px; margin: 10px 0; border-radius: 5px; }
     </style>
     <script>
         function selectMethod(method) {
@@ -59,7 +64,6 @@ HTML_TEMPLATE = """
             });
         }
-        // Initialize with ZeroIG selected
         window.onload = function() {
             selectMethod('zeroig');
         }
@@ -68,7 +72,12 @@ HTML_TEMPLATE = """
 <body>
     <div class="container">
         <h1>🌟 Multi-Method Low-Light Enhancement</h1>
-        <p>Choose your enhancement method and upload a low-light image!</p>
         <div class="method-selection">
             <h3>📚 Select Enhancement Method:</h3>
@@ -83,18 +92,18 @@ HTML_TEMPLATE = """
             </button>
             <div class="method-info" data-method="zeroig">
-                <strong>ZeroIG:</strong> Zero-shot illumination-guided joint denoising and adaptive enhancement.
-                Fast, no training data required, prevents over-enhancement.
             </div>
             <div class="method-info" data-method="cfwd">
-                <strong>CFWD:</strong> CLIP-Fourier Guided Wavelet Diffusion for low-light enhancement.
-                State-of-the-art quality using diffusion models and CLIP guidance.
             </div>
             <div class="method-info" data-method="both">
-                <strong>Comparison Mode:</strong> Process with both methods to see the differences.
-                Takes longer but shows you the best of both approaches.
             </div>
         </div>
@@ -114,10 +123,6 @@ HTML_TEMPLATE = """
         <div class="status">{{ status }}</div>
         {% endif %}
-        {% if processing %}
-        <div class="processing">{{ processing }}</div>
-        {% endif %}
         {% if error %}
         <div class="error">{{ error }}</div>
         {% endif %}
@@ -170,14 +175,14 @@ HTML_TEMPLATE = """
             <div style="margin: 15px 0;">
                 <strong>🎯 ZeroIG (CVPR 2024):</strong>
-                <p>Zero-shot illumination-guided joint denoising and adaptive enhancement. Fast and reliable.</p>
                 <p>📄 <a href="https://openaccess.thecvf.com/content/CVPR2024/papers/Shi_ZERO-IG_Zero-Shot_Illumination-Guided_Joint_Denoising_and_Adaptive_Enhancement_for_Low-Light_CVPR_2024_paper.pdf">Paper</a> |
                    💻 <a href="https://github.com/Doyle59217/ZeroIG">Code</a></p>
             </div>
             <div style="margin: 15px 0;">
                 <strong>🌊 CFWD (2024):</strong>
-                <p>CLIP-Fourier Guided Wavelet Diffusion for state-of-the-art low-light enhancement.</p>
                 <p>📄 <a href="https://arxiv.org/abs/2401.03788">Paper</a> |
                    💻 <a href="https://github.com/hejh8/CFWD">Code</a></p>
             </div>
@@ -187,6 +192,47 @@ HTML_TEMPLATE = """
 </html>
 """
 class MultiMethodProcessor:
     def __init__(self):
         self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
@@ -195,7 +241,7 @@ class MultiMethodProcessor:
         print(f"Multi-method processor initialized on {self.device}")
     def load_zeroig(self):
-        """Load ZeroIG model"""
         try:
             from model import Finetunemodel, Network
@@ -230,15 +276,33 @@ class MultiMethodProcessor:
             return None
     def load_cfwd(self):
-        """Load CFWD model - placeholder for now"""
         try:
-            # TODO: Implement CFWD model loading
-            # This is where you'll add CFWD model loading code
-            print("⚠️ CFWD model not yet implemented")
-            return None
         except Exception as e:
-            print(f"❌ CFWD not available: {e}")
             return None
     def enhance_with_zeroig(self, image):
@@ -282,26 +346,24 @@ class MultiMethodProcessor:
             return None, f"ZeroIG failed: {str(e)}"
     def enhance_with_cfwd(self, image):
-        """Enhance image using CFWD - placeholder for now"""
         if self.cfwd_model is None:
-            # TODO: Replace this with actual CFWD implementation
-            # For now, return a simple enhancement as placeholder
-            arr = np.array(image).astype(np.float32)
-            enhanced = np.clip(arr * 2.2, 0, 255).astype(np.uint8)
-            enhanced_image = Image.fromarray(enhanced)
-            return enhanced_image, "CFWD placeholder (not yet implemented)"
         try:
-            # TODO: Implement actual CFWD processing
             pass
         except Exception as e:
             return None, f"CFWD failed: {str(e)}"
-    def simple_enhance(self, image):
-        """Fallback simple enhancement"""
-        arr = np.array(image).astype(np.float32)
-        enhanced = np.clip(arr * 1.8, 0, 255).astype(np.uint8)
-        return Image.fromarray(enhanced)
 # Initialize processor
 print("🚀 Loading multi-method processor...")
@@ -319,10 +381,12 @@ def index():
     results = None
     status = None
     error = None
-    processing = None
     method_used = None
     processing_time = None
     if request.method == 'POST':
         try:
             file = request.files['image']
@@ -359,12 +423,15 @@ def index():
                     zeroig_enhanced, zeroig_msg = processor.enhance_with_zeroig(image)
                     cfwd_enhanced, cfwd_msg = processor.enhance_with_cfwd(image)
                     if zeroig_enhanced:
                         results['zeroig'] = image_to_base64(zeroig_enhanced)
                     if cfwd_enhanced:
                         results['cfwd'] = image_to_base64(cfwd_enhanced)
-                    status = f"✅ Both methods completed"
                 end_time = time.time()
                 processing_time = round(end_time - start_time, 2)
@@ -375,13 +442,13 @@ def index():
             print(f"Error: {e}")
     return render_template_string(HTML_TEMPLATE,
                                   results=results,
                                   status=status,
                                   error=error,
-                                  processing=processing,
                                   method_used=method_used,
                                   processing_time=processing_time)
 if __name__ == '__main__':
-    print("🚀 Starting Multi-Method Enhancement App...")
     app.run(host='0.0.0.0', port=7860)

 app = Flask(__name__)
+# Add paths for different methods
+sys.path.append('.')  # For ZeroIG (root folder)
+sys.path.append('./CFWD')  # For CFWD method
 HTML_TEMPLATE = """
 <!DOCTYPE html>
 <html>
         .error { color: red; }
         .processing { color: orange; font-weight: bold; }
         .method-info { background: #e3f2fd; padding: 15px; margin: 10px 0; border-radius: 5px; }
+        .folder-status { background: #fff3cd; padding: 10px; margin: 10px 0; border-radius: 5px; font-family: monospace; }
     </style>
     <script>
         function selectMethod(method) {
             });
         }
         window.onload = function() {
             selectMethod('zeroig');
         }
 <body>
     <div class="container">
         <h1>🌟 Multi-Method Low-Light Enhancement</h1>
+        <p>Professional low-light image enhancement with multiple state-of-the-art methods</p>
+        <div class="folder-status">
+            <strong>📁 Method Status:</strong><br>
+            {{ method_status }}
+        </div>
         <div class="method-selection">
             <h3>📚 Select Enhancement Method:</h3>
             </button>
             <div class="method-info" data-method="zeroig">
+                <strong>🎯 ZeroIG:</strong> Zero-shot illumination-guided joint denoising and adaptive enhancement.
+                Fast processing, no training data required, prevents over-enhancement artifacts.
             </div>
             <div class="method-info" data-method="cfwd">
+                <strong>🌊 CFWD:</strong> CLIP-Fourier Guided Wavelet Diffusion for low-light enhancement.
+                Uses diffusion models with CLIP guidance for state-of-the-art quality.
             </div>
             <div class="method-info" data-method="both">
+                <strong>🔄 Comparison Mode:</strong> Process with both methods to see the differences.
+                Takes longer but shows you the best of both approaches side-by-side.
             </div>
         </div>
         <div class="status">{{ status }}</div>
         {% endif %}
         {% if error %}
         <div class="error">{{ error }}</div>
         {% endif %}
             <div style="margin: 15px 0;">
                 <strong>🎯 ZeroIG (CVPR 2024):</strong>
+                <p>Zero-shot illumination-guided joint denoising and adaptive enhancement for low-light images.</p>
                 <p>📄 <a href="https://openaccess.thecvf.com/content/CVPR2024/papers/Shi_ZERO-IG_Zero-Shot_Illumination-Guided_Joint_Denoising_and_Adaptive_Enhancement_for_Low-Light_CVPR_2024_paper.pdf">Paper</a> |
                    💻 <a href="https://github.com/Doyle59217/ZeroIG">Code</a></p>
             </div>
             <div style="margin: 15px 0;">
                 <strong>🌊 CFWD (2024):</strong>
+                <p>Low-light Image Enhancement via CLIP-Fourier Guided Wavelet Diffusion.</p>
                 <p>📄 <a href="https://arxiv.org/abs/2401.03788">Paper</a> |
                    💻 <a href="https://github.com/hejh8/CFWD">Code</a></p>
             </div>
 </html>
 """
+def check_method_availability():
+    """Check which methods are available"""
+    status = []
+    # Check ZeroIG (root folder)
+    zeroig_files = ['model.py', 'loss.py', 'utils.py']
+    zeroig_ok = all(os.path.exists(f) for f in zeroig_files)
+    if zeroig_ok:
+        # Check for weights
+        weights_found = False
+        for weight_path in ["./weights/LOL.pt", "./weights/model.pt"]:
+            if os.path.exists(weight_path):
+                weights_found = True
+                break
+        if weights_found:
+            status.append("✅ ZeroIG: Ready with trained weights")
+        else:
+            status.append("⚠️ ZeroIG: Available but no trained weights found")
+    else:
+        status.append("❌ ZeroIG: Missing required files")
+    # Check CFWD folder
+    cfwd_folder_exists = os.path.exists('./CFWD')
+    if cfwd_folder_exists:
+        cfwd_files = os.listdir('./CFWD')
+        status.append(f"📁 CFWD folder: Found with {len(cfwd_files)} files")
+        # Check for common CFWD files
+        expected_cfwd_files = ['test.py', 'model.py', 'datasets.py']
+        found_files = [f for f in expected_cfwd_files if f in cfwd_files]
+        if found_files:
+            status.append(f"✅ CFWD files found: {', '.join(found_files)}")
+        else:
+            status.append("⚠️ CFWD: Folder exists but expected files not found")
+    else:
+        status.append("❌ CFWD: Folder not found - please create ./CFWD/ and upload CFWD files")
+    return "<br>".join(status)
 class MultiMethodProcessor:
     def __init__(self):
         self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
         print(f"Multi-method processor initialized on {self.device}")
     def load_zeroig(self):
+        """Load ZeroIG model from root folder"""
         try:
             from model import Finetunemodel, Network
             return None
     def load_cfwd(self):
+        """Load CFWD model from CFWD folder"""
         try:
+            # Check if CFWD folder exists
+            if not os.path.exists('./CFWD'):
+                print("❌ CFWD folder not found")
+                return None
+            # Add CFWD folder to path and try importing
+            sys.path.insert(0, './CFWD')
+            # Try to import CFWD modules (will depend on actual CFWD structure)
+            # This is a placeholder - you'll need to adapt based on actual CFWD files
+            try:
+                # Example imports - replace with actual CFWD imports
+                # from model import CFWDModel  # Replace with actual CFWD model
+                # model = CFWDModel()
+                # model.load_weights('./CFWD/weights/cfwd_weights.pt')
+                print("⚠️ CFWD: Import structure not yet implemented")
+                return None
+            except ImportError as e:
+                print(f"❌ CFWD import failed: {e}")
+                return None
         except Exception as e:
+            print(f"❌ CFWD loading error: {e}")
             return None
     def enhance_with_zeroig(self, image):
             return None, f"ZeroIG failed: {str(e)}"
     def enhance_with_cfwd(self, image):
+        """Enhance image using CFWD"""
         if self.cfwd_model is None:
+            # Placeholder enhancement until CFWD is implemented
+            try:
+                # Simple placeholder - replace with actual CFWD processing
+                arr = np.array(image).astype(np.float32)
+                enhanced = np.clip(arr * 2.2, 0, 255).astype(np.uint8)
+                enhanced_image = Image.fromarray(enhanced)
+                return enhanced_image, "CFWD placeholder (upload CFWD files to ./CFWD/ folder for real processing)"
+            except Exception as e:
+                return None, f"CFWD placeholder failed: {str(e)}"
         try:
+            # TODO: Implement actual CFWD processing when files are uploaded
+            # This will depend on the actual CFWD model structure
             pass
         except Exception as e:
             return None, f"CFWD failed: {str(e)}"
 # Initialize processor
 print("🚀 Loading multi-method processor...")
     results = None
     status = None
     error = None
     method_used = None
     processing_time = None
+    # Check method availability
+    method_status = check_method_availability()
     if request.method == 'POST':
         try:
             file = request.files['image']
                     zeroig_enhanced, zeroig_msg = processor.enhance_with_zeroig(image)
                     cfwd_enhanced, cfwd_msg = processor.enhance_with_cfwd(image)
+                    status_msgs = []
                     if zeroig_enhanced:
                         results['zeroig'] = image_to_base64(zeroig_enhanced)
+                        status_msgs.append(f"ZeroIG: {zeroig_msg}")
                     if cfwd_enhanced:
                         results['cfwd'] = image_to_base64(cfwd_enhanced)
+                        status_msgs.append(f"CFWD: {cfwd_msg}")
+                    status = "✅ " + " | ".join(status_msgs)
                 end_time = time.time()
                 processing_time = round(end_time - start_time, 2)
             print(f"Error: {e}")
     return render_template_string(HTML_TEMPLATE,
+                                  method_status=method_status,
                                   results=results,
                                   status=status,
                                   error=error,
                                   method_used=method_used,
                                   processing_time=processing_time)
 if __name__ == '__main__':
+    print("🚀 Starting Organized Multi-Method Enhancement App...")
     app.run(host='0.0.0.0', port=7860)