Update app.py

app.py

@@ -34,10 +34,21 @@ def initialize_enhancer():
         # Force CPU usage for Hugging Face compatibility
         device = torch.device('cpu')
         
+        # Check if model weights file exists
+        weights_path = 'weights/realesr-general-x4v3.pth'
+        if not os.path.exists(weights_path):
+            print(f"Model weights not found at {weights_path}")
+            # Create the directory if it doesn't exist
+            os.makedirs('weights', exist_ok=True)
+            # You'd normally download weights here, but for this example
+            # we'll just use a placeholder
+            print("ERROR: Model weights file not found!")
+            return False
+        
         # Initialize the upsampler
         upsampler = RealESRGANer(
             scale=4,
-            model_path='weights/realesr-general-x4v3.pth',
+            model_path=weights_path,
             model=model,
             tile=0,  # Set to 0 for small images, increase for large images
             tile_pad=10,
@@ -58,11 +69,14 @@ def initialize_enhancer():
         print(error_msg)
         return False
 
+# Global init flag to track if we've attempted initialization
+init_attempted = False
+
 # Initialize model immediately at startup - this blocks until model is ready
 if not initialize_enhancer():
-    print("ERROR: Model failed to initialize. Server will exit.")
-    # In production, you might prefer to exit when model fails
-    # os.exit(1)
+    print("ERROR: Model failed to initialize. Server will continue running but enhancement won't work.")
+    # We'll keep running but mark that we attempted initialization
+    init_attempted = True
 
 @app.route('/enhance', methods=['POST'])
 def enhance_image():
@@ -110,10 +124,20 @@ def enhance_image():
 
 @app.route('/health', methods=['GET'])
 def health_check():
-    global upsampler
+    global upsampler, init_attempted
+    
+    # If we have the upsampler, we're ready
+    if upsampler is not None:
+        status = 'ready'
+    # If we tried to initialize and failed, report failure
+    elif init_attempted:
+        status = 'failed'
+    # Otherwise we're still in an unknown state
+    else:
+        status = 'initializing'
     
     status_info = {
-        'status': 'ready' if upsampler is not None else 'failed',
+        'status': status,
         'timestamp': time.time()
     }