Update src/module/AIP.py

src/module/AIP.py

@@ -12,15 +12,10 @@ import sys
 
 # --- 1. IMPORT SÉCURISÉ DES FONCTIONS ---
 try:
-    from Function.AIP_Function import (
-        calculate_occupancy_rate,
-        audit_data_quality,
-        get_stock_alerts,
-        get_patient_stats
-    )
+    from module import Validation_Contract
+    
 except ImportError as e:
-    st.error(f"Erreur interne : impossible de trouver AIP_Function. {e}")
-    calculate_occupancy_rate = audit_data_quality = get_stock_alerts = get_patient_stats = lambda x: "Fonction non chargée"
+    st.error(f"Erreur interne : impossible de trouver Validation_Contract. {e}")
 
 # --- STYLE EXTRAIT DU CODE 01 (GOTHAM THEME) ---
 def apply_custom_styles():
@@ -59,33 +54,46 @@ def apply_custom_styles():
     </style>
     """, unsafe_allow_html=True)
 
-class JasmineAgent:
-    def __init__(self, rdf_store, ontology_rules, dataframes=None):
-        self.google_key = os.environ.get("GOOGLE_API_KEY") or st.secrets.get("GOOGLE_API_KEY")
-        if self.google_key: 
-            genai.configure(api_key=self.google_key)
 
-        self.groq_key = os.environ.get("GROQ_API_KEY") or st.secrets.get("GROQ_API_KEY")
-        self.groq_client = Groq(api_key=self.groq_key) if self.groq_key else None
-        
+class JasmineAgent:
+    def __init__(self, rdf_store, ontology_rules, dataframes=None, gspread_client=None):
         self.rdf_store = rdf_store
         self.ontology_rules = ontology_rules
         self.dataframes = dataframes or {}
-        
-        self.MODEL_CASCADE = [
-            "gemini-2.0-flash-exp",
-            "llama-3.3-70b-versatile",
-            "gemini-2.0-flash-lite",
-            "llama-3.1-8b-instant"
-        ]
-        
-        self.available_functions = {
-            "calculate_occupancy_rate": calculate_occupancy_rate,
-            "audit_data_quality": audit_data_quality,
-            "get_stock_alerts": get_stock_alerts,
-            "get_patient_stats": get_patient_stats
+        self.gspread_client = gspread_client
+
+        # Liste des modules complets que l'agent peut exécuter
+        self.available_modules = {
+            "validation_contract": {
+                "class": Validation_Contract.ETL_Engine,
+                "start_method": "run_etl",
+                "args": {"gspread_client": self.gspread_client, "sheet_name": "bdd_mvp_faker"}
+            }
         }
 
+
+    def execute_module(self, module_name):
+    """
+    Lance un module complet et retourne les résultats.
+    """
+    if module_name not in self.available_modules:
+        return {"error": f"Module {module_name} inconnu"}
+
+    mod_info = self.available_modules[module_name]
+    try:
+        # Instancier le module
+        instance = mod_info["class"](**mod_info.get("args", {}))
+        # Lancer la méthode de démarrage
+        result, explorer = getattr(instance, mod_info["start_method"])()
+        # Stocker DataExplorer dans session_state
+        st.session_state.data_explorer = explorer
+        return {"success": True, "result": {"run_status": result, "data_explorer": explorer.get_summary()}}
+    except Exception as e:
+        import traceback
+        return {"error": str(e), "trace": traceback.format_exc()}
+
+
+    
     def build_base_system_prompt(self):
         return """
 ROLE: Tu es Jasmine, un agent intelligent d'analyse hospitalière.
@@ -145,10 +153,13 @@ FORMAT JSON STRICT :
                 tool = action.get("tool")
                 
                 func_res = None
+                tool = action.get("tool")
                 if tool and tool != "none":
-                    func_res = self.execute_function(tool, action.get("args", {}))
-                
-                return "", action, thought_trace, func_res
+                    if tool in self.available_functions:
+                        func_res = self.execute_function(tool, action.get("args", {}))
+                    elif tool in self.available_modules:
+                        func_res = self.execute_module(tool)
+
             except Exception as e:
                 last_error = str(e)
                 continue