Update app.py

app.py

@@ -548,66 +548,66 @@ def optimize_image(self, image):
     if image.mode != 'RGB':
         image = image.convert('RGB')
     
-    max_size = 768 if self.device != "cpu" else 512
+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:
-            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
-        logger.info("ULTRA SUPREME ANALYSIS - Maximum intelligence deployment")
-        
-        clip_fast = self.interrogator.interrogate_fast(image)
-        clip_classic = self.interrogator.interrogate_classic(image) 
-        clip_best = self.interrogator.interrogate(image)
-        
-        logger.info(f"ULTRA CLIP Results:\nFast: {clip_fast}\nClassic: {clip_classic}\nBest: {clip_best}")
-        
-        # ULTRA SUPREME ANALYSIS
-        ultra_analysis = self.analyzer.ultra_supreme_analysis(clip_fast, clip_classic, clip_best)
-        
-        # BUILD ULTRA SUPREME FLUX PROMPT
-        optimized_prompt = self.analyzer.build_ultra_supreme_prompt(ultra_analysis, [clip_fast, clip_classic, clip_best])
-        
-        # CALCULATE ULTRA SUPREME SCORE
-        score, breakdown = self.analyzer.calculate_ultra_supreme_score(optimized_prompt, ultra_analysis)
-        
-        end_time = datetime.now()
-        duration = (end_time - start_time).total_seconds()
-        
-        # Memory cleanup
-        if self.device == "cpu":
-            gc.collect()
-        else:
-            torch.cuda.empty_cache()
-        
-        # ULTRA COMPREHENSIVE ANALYSIS REPORT
-        gpu_status = "⚡ ZeroGPU" if torch.cuda.is_available() else "💻 CPU"
-        
-        # Format detected elements
-        features = ", ".join(ultra_analysis["facial_ultra"]["facial_hair"]) if ultra_analysis["facial_ultra"]["facial_hair"] else "None detected"
-        cultural = ", ".join(ultra_analysis["demographic"]["cultural_religious"]) if ultra_analysis["demographic"]["cultural_religious"] else "None detected"
-        clothing = ", ".join(ultra_analysis["clothing_accessories"]["eyewear"] + ultra_analysis["clothing_accessories"]["headwear"]) if ultra_analysis["clothing_accessories"]["eyewear"] or ultra_analysis["clothing_accessories"]["headwear"] else "None detected"
-        
-        analysis_info = f"""**🚀 ULTRA SUPREME ANALYSIS COMPLETE**
+
+    @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
+            logger.info("ULTRA SUPREME ANALYSIS - Maximum intelligence deployment")
+            
+            clip_fast = self.interrogator.interrogate_fast(image)
+            clip_classic = self.interrogator.interrogate_classic(image) 
+            clip_best = self.interrogator.interrogate(image)
+            
+            logger.info(f"ULTRA CLIP Results:\nFast: {clip_fast}\nClassic: {clip_classic}\nBest: {clip_best}")
+            
+            # ULTRA SUPREME ANALYSIS
+            ultra_analysis = self.analyzer.ultra_supreme_analysis(clip_fast, clip_classic, clip_best)
+            
+            # BUILD ULTRA SUPREME FLUX PROMPT
+            optimized_prompt = self.analyzer.build_ultra_supreme_prompt(ultra_analysis, [clip_fast, clip_classic, clip_best])
+            
+            # CALCULATE ULTRA SUPREME SCORE
+            score, breakdown = self.analyzer.calculate_ultra_supreme_score(optimized_prompt, ultra_analysis)
+            
+            end_time = datetime.now()
+            duration = (end_time - start_time).total_seconds()
+            
+            # Memory cleanup
+            if self.device == "cpu":
+                gc.collect()
+            else:
+                torch.cuda.empty_cache()
+            
+            # ULTRA COMPREHENSIVE ANALYSIS REPORT
+            gpu_status = "⚡ ZeroGPU" if torch.cuda.is_available() else "💻 CPU"
+            
+            # Format detected elements
+            features = ", ".join(ultra_analysis["facial_ultra"]["facial_hair"]) if ultra_analysis["facial_ultra"]["facial_hair"] else "None detected"
+            cultural = ", ".join(ultra_analysis["demographic"]["cultural_religious"]) if ultra_analysis["demographic"]["cultural_religious"] else "None detected"
+            clothing = ", ".join(ultra_analysis["clothing_accessories"]["eyewear"] + ultra_analysis["clothing_accessories"]["headwear"]) if ultra_analysis["clothing_accessories"]["eyewear"] or ultra_analysis["clothing_accessories"]["headwear"] else "None detected"
+            
+            analysis_info = f"""**🚀 ULTRA SUPREME ANALYSIS COMPLETE**
 **Processing:** {gpu_status} • {duration:.1f}s • Triple CLIP Ultra Intelligence  
 **Ultra Score:** {score}/100 • Breakdown: Structure({breakdown.get('structure',0)}) Features({breakdown.get('features',0)}) Cultural({breakdown.get('cultural',0)}) Emotional({breakdown.get('emotional',0)}) Technical({breakdown.get('technical',0)})  
 **Generation:** #{self.usage_count}  
@@ -624,12 +624,12 @@ def generate_ultra_supreme_prompt(self, image):
 - **Classic:** {clip_classic[:50]}...  
 - **Best:** {clip_best[:50]}...
 **⚡ ULTRA OPTIMIZATION:** Applied absolute maximum depth analysis with Pariente AI research rules"""
-        
-        return optimized_prompt, analysis_info, score, breakdown
-        
-    except Exception as e:
-        logger.error(f"Ultra supreme generation error: {e}")
-        return f"❌ Error: {str(e)}", "Please try with a different image.", 0, {}
+            
+            return optimized_prompt, analysis_info, score, breakdown
+            
+        except Exception as e:
+            logger.error(f"Ultra supreme generation error: {e}")
+            return f"❌ Error: {str(e)}", "Please try with a different image.", 0, {}
 
 # Initialize the optimizer
 optimizer = UltraSupremeOptimizer()
@@ -701,6 +701,7 @@ def create_interface():
         position: relative;
         overflow: hidden;
     }
+    
     .main-header::before {
         content: '';
         position: absolute;