upgrade app.py

app.py

@@ -11,11 +11,8 @@ import subprocess
 import shutil
 from datetime import datetime
 from pathlib import Path
-
-# Importation et chargement des variables d'environnement depuis le fichier .env
 from dotenv import load_dotenv
 load_dotenv()
-
 import streamlit as st
 import pandas as pd
 import plotly.express as px
@@ -414,19 +411,12 @@ def save_qa_pair(question, answer, category, subcategory, source, attack_signatu
 
 def collect_kaggle_data(queries):
     logger.info("Début de la collecte des données Kaggle...")
-    kaggle_dir = Path(".kaggle")
-    kaggle_json = kaggle_dir / "kaggle.json"
-    if not kaggle_json.exists():
-        logger.warning("Fichier kaggle.json non trouvé. Veuillez le placer dans le dossier .kaggle/")
-        return
-    
-    os.environ['KAGGLE_CONFIG_DIR'] = str(kaggle_dir.absolute())
     
-    try:
-        kaggle.api.authenticate()
-    except Exception as e:
-        logger.error(f"Erreur d'authentification Kaggle: {str(e)}")
-        return
+    # Importation de Kaggle ici, après avoir configuré les variables d'environnement
+    os.environ['KAGGLE_USERNAME'] = os.getenv('KAGGLE_USERNAME')
+    os.environ['KAGGLE_KEY'] = os.getenv('KAGGLE_KEY')
+    import kaggle
+    kaggle.api.authenticate()
     
     search_queries = queries.split('\n') if queries else ["cybersecurity", "vulnerability"]