from model_initial import initialize_models
from generate_caption import generate_caption
from PIL import Image

if __name__ == "__main__" or "get_ipython" in globals():
    print("Initializing models...")
    try:
        clip_model, image_adapter = initialize_models()
        print("Models initialized successfully!")
    except Exception as e:
        print(f"Error initializing models: {e}")
        print("You can still use the basic caption functionality with Gemini API only")
        clip_model, image_adapter = None, None