hotfix: Ajuste na tipagem

.gitignore

@@ -1,2 +1,3 @@
 *.pickle
-*.pyc
+*.pyc
+src/__pycache__/*.pyc

src/main.py

@@ -1,4 +1,6 @@
 import gradio as gr
+from typing import Tuple, List, Optional, Any
+from PIL.Image import Image as PILImage
 from models.inference import (
     run_inference,
     get_available_model_choices,
@@ -34,35 +36,35 @@ custom_css = """
 # ==========================================================
 # HELPERS (Nomes de variáveis e lógica preservados)
 # ==========================================================
-def format_groups(enabled_groups):
+def format_groups(enabled_groups: List[str]) -> str:
     if not enabled_groups: return "No metadata group selected."
     label_map = dict(GROUP_CHOICES)
     return " | ".join([label_map.get(g, g) for g in enabled_groups])
 
-def safe_bool(value):
+def safe_bool(value: Any) -> bool:
     return bool(value)
 
-def build_values_dict(age, gender, region, diameter1, diameter2, itch, grew, hurt, changed, bleed, elevation):
+def build_values_dict(age: float, gender: str, region: str, diameter1: float, diameter2: float, itch: bool, grew: bool, hurt: bool, changed: bool, bleed: bool, elevation: bool) -> dict:
     return dict(
         age=age, gender=gender, region=region, diameter_1=diameter1, diameter_2=diameter2,
         itch=safe_bool(itch), grew=safe_bool(grew), hurt=safe_bool(hurt),
         changed=safe_bool(changed), bleed=safe_bool(bleed), elevation=safe_bool(elevation),
     )
 
-def build_metadata_preview(enabled_groups, age, gender, region, diameter1, diameter2, itch, grew, hurt, changed, bleed, elevation):
+def build_metadata_preview(enabled_groups: List[str], age: float, gender: str, region: str, diameter1: float, diameter2: float, itch: bool, grew: bool, hurt: bool, changed: bool, bleed: bool, elevation: bool) -> Tuple[str, str]:
     values = build_values_dict(age, gender, region, diameter1, diameter2, itch, grew, hurt, changed, bleed, elevation)
     metadata_csv = build_metadata_csv(values, enabled_groups)
     groups_text = format_groups(enabled_groups)
     return metadata_csv, groups_text
 
-def validate_inputs(image, enabled_groups, age, diameter1, diameter2):
+def validate_inputs(image: Optional[PILImage], enabled_groups: List[str], age: Optional[float], diameter1: Optional[float], diameter2: Optional[float]) -> None:
     if image is None: raise gr.Error("Please upload a dermoscopic image first.")
     if not enabled_groups: raise gr.Error("Please select at least one metadata group.")
     if age is None or age < 0: raise gr.Error("Age must be a valid non-negative number.")
     if diameter1 is None or diameter1 < 0: raise gr.Error("Diameter 1 must be a valid non-negative number.")
     if diameter2 is None or diameter2 < 0: raise gr.Error("Diameter 2 must be a valid non-negative number.")
 
-def gradio_predict(image, selected_model_key, enabled_groups, age, gender, region, diameter1, diameter2, itch, grew, hurt, changed, bleed, elevation):
+def gradio_predict(image: PILImage, selected_model_key: str, enabled_groups: List[str], age: float, gender: str, region: str, diameter1: float, diameter2: float, itch: bool, grew: bool, hurt: bool, changed: bool, bleed: bool, elevation: bool) -> Tuple[PILImage, PILImage, str, str, str]:
     validate_inputs(image, enabled_groups, age, diameter1, diameter2)
     values = build_values_dict(age, gender, region, diameter1, diameter2, itch, grew, hurt, changed, bleed, elevation)
     metadata_csv = build_metadata_csv(values, enabled_groups)
@@ -82,13 +84,13 @@ def gradio_predict(image, selected_model_key, enabled_groups, age, gender, regio
     )
     return image, heatmap_img, pretty_prediction, metadata_csv, groups_text
 
-def clear_all():
+def clear_all() -> Tuple[None, List[str], str, float, str, str, float, float, bool, bool, bool, bool, bool, bool, None, None, str, str, str]:
     default_model = DEFAULT_MODEL_KEY
     if default_model is None and MODEL_CHOICES:
         default_model = MODEL_CHOICES[0][1]
 
     return (
-        None, DEFAULT_GROUPS, default_model, 55, "FEMALE", "NECK", 6, 5, 
+        None, DEFAULT_GROUPS, default_model, 55.0, "FEMALE", "NECK", 6.0, 5.0, 
         False, False, False, False, False, False,
         None, None, "### Prediction Result\n\nRun the model to see the output here.",
         "", format_groups(DEFAULT_GROUPS)