Update app.py

app.py

@@ -148,20 +148,9 @@ class ImageAnalyzer:
             logger.error(f"Model initialization error: {e}")
             return None
 
-    @st.cache_data  # Cache preprocessed images
     def preprocess_image(self, image_bytes):
         """Preprocess image for model input"""
-        try:
-            image = Image.open(image_bytes)
-            if image.mode != 'RGB':
-                image = image.convert('RGB')
-            
-            width, height = 224, 224
-            image = image.resize((width, height), Image.Resampling.LANCZOS)
-            return image
-        except Exception as e:
-            logger.error(f"Image preprocessing error: {e}")
-            return None
+        return _cached_preprocess_image(image_bytes)
 
     def analyze_image(self, image):
         """Analyze image for defects"""
@@ -198,6 +187,20 @@ class ImageAnalyzer:
             return None
 
 @st.cache_data
+def _cached_preprocess_image(image_bytes):
+    """Cached version of image preprocessing"""
+    try:
+        image = Image.open(image_bytes)
+        if image.mode != 'RGB':
+            image = image.convert('RGB')
+        
+        width, height = 224, 224
+        image = image.resize((width, height), Image.Resampling.LANCZOS)
+        return image
+    except Exception as e:
+        logger.error(f"Image preprocessing error: {e}")
+        return None
+        
 def get_groq_response(query: str, context: str) -> str:
     """Get response from Groq LLM with caching"""
     try: