File size: 4,255 Bytes
62a0596 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 |
#!/bin/bash
# --- Couleurs pour l'affichage ---
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m'
# --- Fonctions d'affichage ---
step() { echo -e "${GREEN}[ÉTAPE]${NC} $1"; }
info() { echo -e "${BLUE}[INFO]${NC} $1"; }
warning() { echo -e "${YELLOW}[AVERTISSEMENT]${NC} $1"; }
error() { echo -e "${RED}[ERREUR]${NC} $1"; }
# Vérifier si on est root
if [[ $EUID -eq 0 ]]; then
error "Ne pas exécuter ce script en tant que root. Veuillez utiliser un utilisateur normal."
exit 1
fi
# Supprimer l'arborescence et l'environnement existants pour une installation propre
step "Suppression de l'environnement existant et de l'arborescence..."
rm -rf data logs config server scripts models .kaggle venv
info "Nettoyage terminé."
# --- Étape 1 : Création de l'arborescence du projet ---
step "Création de l'arborescence du projet..."
mkdir -p data/devsecops/qa
mkdir -p data/security/qa
mkdir -p data/development/qa
mkdir -p data/data_analysis/qa
mkdir -p logs
mkdir -p config
mkdir -p server
mkdir -p scripts
mkdir -p models
mkdir -p llama.cpp # Ajout du dossier pour llama.cpp
mkdir -p .kaggle
info "Arborescence créée."
# --- Étape 2 : Installation des dépendances système ---
step "Installation des dépendances système (peut nécessiter votre mot de passe)..."
sudo apt update
sudo apt install -y python3 python3-pip python3-venv cmake build-essential git aria2
info "Dépendances système installées."
# --- Étape 3 : Création et activation de l'environnement virtuel ---
step "Création de l'environnement virtuel..."
python3 -m venv venv
info "Environnement virtuel créé."
source venv/bin/activate
info "Environnement virtuel activé."
# --- Étape 4 : Création du fichier requirements.txt et installation des dépendances ---
step "Génération et installation des dépendances Python..."
# Note : j'ai ajouté 'python-dotenv' pour gérer les variables d'environnement
cat << EOF > requirements.txt
streamlit
pandas
plotly
beautifulsoup4
html2text
requests
kaggle
python-dotenv
lxml
matplotlib
EOF
pip install -r requirements.txt
info "Dépendances Python installées."
# --- Étape 5 : Création des fichiers de configuration ---
step "Création du fichier de configuration 'config/app_config.py'..."
cat << EOF > config/app_config.py
# app_config.py
import os
from pathlib import Path
import streamlit as st
from dotenv import load_dotenv
def setup_dotenv():
env_dir = Path("config")
env_dir.mkdir(exist_ok=True)
env_path = env_dir / ".env"
if not env_path.exists():
with open(env_path, 'w') as f:
f.write("# Configuration du serveur LLM et des clés API\n")
f.write("LLM_SERVER_URL=http://localhost:8080/completion\n")
f.write("GITHUB_API_TOKEN=\n")
f.write("HUGGINGFACE_API_TOKEN=\n")
f.write("NVD_API_KEY=\n")
f.write("STACK_EXCHANGE_API_KEY=\n")
load_dotenv(dotenv_path=env_path)
setup_dotenv()
LLM_SERVER_URL = os.getenv('LLM_SERVER_URL', 'http://localhost:8080/completion')
def init_session_state():
if 'bot_status' not in st.session_state:
st.session_state.bot_status = "Arrêté"
if 'server_status' not in st.session_state:
st.session_state.server_status = "Inactif"
if 'total_qa_pairs' not in st.session_state:
st.session_state.total_qa_pairs = 0
if 'logs' not in st.session_state:
st.session_state.logs = []
if 'qa_data' not in st.session_state:
st.session_state.qa_data = []
if 'enable_enrichment' not in st.session_state:
st.session_state.enable_enrichment = True
if 'min_relevance' not in st.session_state:
st.session_state.min_relevance = 70
if 'num_queries' not in st.session_state:
st.session_state.num_queries = 5
if 'temperature' not in st.session_state:
st.session_state.temperature = 0.7
if 'n_predict' not in st.session_state:
st.session_state.n_predict = 512
REQUEST_COUNT = 0
MAX_REQUESTS_BEFORE_PAUSE = 15
MIN_PAUSE = 2
MAX_PAUSE = 5
USE_API_KEYS = True
EOF
info "Fichier de configuration créé."
# --- Étape 6 : Lancement de l'application Streamlit ---
step "Lancement de l'application app.py..."
./venv/bin/streamlit run app.py
|