from utils.logging_utils import setup_logging
from scripts.train_viral_potential import train_viral_potential
from scripts.train_engagement_rate import train_engagement_rate
from scripts.train_promotion_strategy import train_promotion_strategy
from scripts.train_time_series import train_time_series
from scripts.analyze_image import analyze_image_url
from scripts.analyze_engagement import analyze_engagement

# Set up logging
setup_logging()

# Main application logic
if __name__ == "__main__":
    # Train and export models
    train_viral_potential()
    train_engagement_rate()
    train_promotion_strategy()
    train_time_series()

    # Analyze engagement data
    analyze_engagement()

    # Analyze an example image
    image_url = "https://instagram.flos5-3.fna.fbcdn.net/v/t51.29350-15/461962570_446497237879604_125381930282480249_n.jpg?stp=dst-jpg_e15_tt6&efg=eyJ2ZW5jb2RlX3RhZyI6ImltYWdlX3VybGdlbi43MjB4MTI4MC5zZHIuZjI5MzUwLmRlZmF1bHRfY292ZXJfZnJhbWUifQ&_nc_ht=instagram.flos5-3.fna.fbcdn.net&_nc_cat=108&_nc_ohc=I22rbM-bFp4Q7kNvgEwCOwX&_nc_gid=22ddc0b11dfd44968a15fc79be74c9b9&edm=AEhyXUkBAAAA&ccb=7-5&ig_cache_key=MzQ3MjEwOTAyNDM0NzA1NTM4NzE3MjMxODMzOTg0NDgzOTg%3D.3-ccb7-5&oh=00_AYCYZr-aNIHjjQ86_HzvYiLDaExV26TEpndfIl9-rjyFcg&oe=678FAD53&_nc_sid=8f1549"
    caption = "This is a beautiful sunset!"
    analyze_image_url(image_url, caption)