main.py

import asyncio
import logging
import os
import sys

from dotenv import load_dotenv

from web3_wrapped.bots.twitter.twitter_bot import Web3WrappedTwitterBot
from web3_wrapped.swarm import start_swarm_chat, web3_wrapped_swarm

load_dotenv()

log_level = os.getenv("LOG_LEVEL", "INFO")
logging.basicConfig(
    level=getattr(logging, log_level),
    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
    force=True,  # ensure config gets used
)

logger = logging.getLogger(__name__)


async def main():
    if len(sys.argv) > 1 and sys.argv[1] == "--twitter-bot":
        # Start Twitter bot mode
        bot = Web3WrappedTwitterBot(web3_wrapped_swarm)
        await bot.start()
    else:
        # Start interactive console mode
        await start_swarm_chat(web3_wrapped_swarm)


if __name__ == "__main__":
    asyncio.run(main())