Update src/ai_processor.py

src/ai_processor.py

@@ -213,7 +213,10 @@ def load_yolo_model():
     return model
 def load_segmentation_model():
     load_model = _import_tf_loader()
-    return load_model(SEG_MODEL_PATH, compile=False, custom_objects={'InputLayer': _import_tf_loader().layers.InputLayer})
+    # Need to import tf.keras.layers within the no_cuda_env for custom_objects
+    with _no_cuda_env():
+        import tensorflow as tf
+    return load_model(SEG_MODEL_PATH, compile=False, custom_objects={'InputLayer': tf.keras.layers.InputLayer})
 
 def load_classification_pipeline():
     pipe = _import_hf_cls()
@@ -542,3 +545,22 @@ class AIProcessor:
         except Exception as e:
             logging.error(f"Pipeline error: {e}", exc_info=True)
             return f"❌ Error: {e}"
+
+# Global processor instance
+processor = None
+
+def get_processor():
+    """Get or create the global processor instance"""
+    global processor
+    if processor is None:
+        processor = AIProcessor()
+    return processor
+
+# Convenience function for external use
+@_SPACES_GPU(enable_queue=True, duration=180)
+def analyze_wound(image: Image.Image, questionnaire_data: Dict[str, Any]) -> str:
+    """
+    Main entry point for wound analysis
+    """
+    proc = get_processor()
+    return proc.full_analysis_pipeline(image, questionnaire_data)