Update app.py

app.py

@@ -537,39 +537,33 @@ class UltraSupremeOptimizer:
             return False
             
 def optimize_image(self, image):
-    if image is None:
-        return None
+        if image is None:
+            return None
+            
+        if isinstance(image, np.ndarray):
+            image = Image.fromarray(image)
+        elif not isinstance(image, Image.Image):
+            image = Image.open(image)
         
-    if isinstance(image, np.ndarray):
-        image = Image.fromarray(image)
-    elif not isinstance(image, Image.Image):
-        image = Image.open(image)
-    
-    if image.mode != 'RGB':
-        image = image.convert('RGB')
-    
-    max_size = 768 if self.device != "cpu" else 512
-    if image.size[0] > max_size or image.size[1] > max_size:
-        image.thumbnail((max_size, max_size), Image.Resampling.LANCZOS)
-    
-    return image
+        if image.mode != 'RGB':
+            image = image.convert('RGB')
+        
+        max_size = 768 if self.device != "cpu" else 512
+        if image.size[0] > max_size or image.size[1] > max_size:
+            image.thumbnail((max_size, max_size), Image.Resampling.LANCZOS)
+        
+        return image
 
-@spaces.GPU
-def generate_ultra_supreme_prompt(self, image):
-    try:
-        if not self.is_initialized:
+    @spaces.GPU
+    def generate_ultra_supreme_prompt(self, image):
+        try:
+            if not self.is_initialized:
                 if not self.initialize_model():
                     return "❌ Model initialization failed.", "Please refresh and try again.", 0, {}
             
             if image is None:
                 return "❌ Please upload an image.", "No image provided.", 0, {}
             
-            self.usage_count += 1
-            
-            image = self.optimize_image(image)
-            if image is None:
-                return "❌ Image processing failed.", "Invalid image format.", 0, {}
-            
             start_time = datetime.now()
             
             # ULTRA SUPREME TRIPLE CLIP ANALYSIS