test_setup.py

#!/usr/bin/env python3
"""
Script de test pour vérifier l'installation de Trading AI Analyzer PRO
"""

import sys

print("=" * 60)
print("🧪 Vérification des dépendances Trading AI Analyzer PRO")
print("=" * 60)
print()

# Test Python version
print(f"✓ Python {sys.version}")
print()

# Test des imports Python
dependencies = [
    ("streamlit", "Streamlit"),
    ("yfinance", "yfinance"),
    ("pandas", "Pandas"),
    ("pandas_ta", "Pandas-TA"),
    ("plotly", "Plotly"),
    ("ollama", "Ollama"),
    ("numpy", "NumPy"),
]

print("Vérification des packages Python :")
all_ok = True
for module_name, display_name in dependencies:
    try:
        __import__(module_name)
        print(f"  ✅ {display_name}")
    except ImportError:
        print(f"  ❌ {display_name} - NON INSTALLÉ")
        all_ok = False

print()

# Test Ollama
print("Vérification de Ollama :")
try:
    import ollama
    print("  ✅ Package Ollama installé")
    print()
    print("  Note: Vérifiez que Ollama est lancé :")
    print("        ollama run mistral")
except Exception as e:
    print(f"  ❌ Ollama non trouvé: {e}")
    all_ok = False

print()
print("=" * 60)

if all_ok:
    print("✅ Toutes les dépendances sont correctement installées!")
    print()
    print("Prochaines étapes :")
    print("1. Assurez-vous que Ollama est lancé avec Mistral:")
    print("   → ollama run mistral")
    print()
    print("2. Lancez l'application:")
    print("   → streamlit run app.py")
    print()
else:
    print("❌ Certaines dépendances manquent ou ne sont pas configurées")
    print()
    print("Solutions :")
    print("1. Réinstallez les packages: pip install -r requirements.txt")
    print("2. Installez Ollama depuis ollama.com")
    print()
    sys.exit(1)

print("=" * 60)