Update main.py

main.py

@@ -11,7 +11,8 @@ PY_MODULES = {
     "dark_circles.py": "DarkCircleDetector",
     "inflammation.py": "RednessDetector",
     "texture.py": "TextureDetector",
-    "skin_tone.py": "SkinToneDetector"
+    "skin_tone.py": "SkinToneDetector",
+    "oiliness.py": "OilinessDetector"  # Added oiliness module
 }
 
 def dynamic_import(module_path, class_name):
@@ -54,6 +55,14 @@ def analyze_skin(image: np.ndarray, analysis_type: str) -> np.ndarray:
         detector = detector_classes["SkinToneDetector"](image)
         result = detector.predict_json()
         output = result.get("output_image")
+    elif analysis_type == "oiliness":
+        detector = detector_classes["OilinessDetector"](image)
+        result = detector.predict_json()
+        if result.get("detected"):
+            output = result.get("overlay_image")
+            print(f"Oiliness scores: {result.get('scores')}")
+        else:
+            print(f"Oiliness detection error: {result.get('error')}")
     return output
 
 # --- Gradio Interface code ---
@@ -62,7 +71,7 @@ app = gr.Interface(
     inputs=[
         gr.Image(type="numpy", label="Upload your face image"),
         gr.Radio(
-            ["dark circles", "redness", "texture", "skin tone"],
+            ["dark circles", "redness", "texture", "skin tone", "oiliness"],  # Added oiliness option
             label="Select Skin Analysis Type"
         ),
     ],