Spaces:
Running
Running
File size: 3,472 Bytes
5b02b7b b682b48 5b02b7b |
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 |
import os
import subprocess
import sys
from pathlib import Path
import venv
import platform
import logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)
def create_virtual_environment(venv_path):
"""Create a virtual environment."""
try:
subprocess.run([sys.executable, "-m", "venv", str(venv_path)], check=True)
logger.info(f"Created virtual environment at {venv_path}")
# Get pip path
pip_path = venv_path / "bin" / "pip"
# Upgrade pip
logger.info("Upgrading pip...")
subprocess.run([str(pip_path), "install", "--upgrade", "pip"], check=True)
except subprocess.CalledProcessError as e:
logger.error(f"Failed to create virtual environment: {e}")
raise
def install_dependencies(venv_path):
"""Install project dependencies."""
try:
pip_path = venv_path / "bin" / "pip"
# Install core dependencies first
logger.info("Installing core dependencies...")
subprocess.run([
str(pip_path), "install", "-e", "."
], check=True)
# Install dev dependencies separately
logger.info("Installing development dependencies...")
subprocess.run([
str(pip_path), "install",
"pytest==8.3.4",
"pytest-mockito==0.0.4",
"black==24.2.0",
"isort==5.13.0",
"flake8==7.0.0"
], check=True)
# Install playwright
logger.info("Installing and setting up Playwright...")
subprocess.run([
str(pip_path), "install", "playwright"
], check=True)
playwright_path = venv_path / "bin" / "playwright"
subprocess.run([str(playwright_path), "install"], check=True)
except subprocess.CalledProcessError as e:
logger.error(f"Failed to install dependencies: {e}")
logger.error(f"Exit code: {e.returncode}")
logger.error(f"Output: {e.output if hasattr(e, 'output') else 'No output'}")
raise
def create_env_file():
"""Create .env file if it doesn't exist."""
env_file = Path(".env")
if not env_file.exists():
with open(env_file, "w") as f:
f.write("GROQ_API_KEY=\n")
f.write("OLLAMA_API_TOKEN=\n")
logger.info("Created .env file")
def main():
"""Main setup function."""
try:
# Check Python version
if sys.version_info < (3, 8):
raise RuntimeError("Python 3.8 or higher is required")
# Get project root directory
project_root = Path(__file__).parent
venv_path = project_root / ".venv"
# Create virtual environment if it doesn't exist
if not venv_path.exists():
create_virtual_environment(venv_path)
# Install dependencies
install_dependencies(venv_path)
# Create .env file
create_env_file()
logger.info("\nSetup completed successfully!")
logger.info("\nNext steps:")
logger.info("1. Update the .env file with your API keys")
logger.info("2. Activate the virtual environment:")
logger.info(" source .venv/bin/activate")
logger.info("3. Run the application: python -m streamlit run src/crawlgpt/ui/chat_app.py")
except Exception as e:
logger.error(f"Setup failed: {e}")
sys.exit(1)
if __name__ == "__main__":
main() |