src/pipeline/main.py

"""
project @ LearnableAI
created @ 2025-01-17
author  @ github.com/ishworrsubedii
"""
import os

import google.generativeai as genai
from groq import Groq

from src.services.use_case_one.word_to_sentence import UseCaseOne
from src.services.use_case_two.image_describer import ImageDescriptionGenerator


class LearnableAI:
    def __init__(self):
        self.groq_client = Groq(api_key=os.getenv("GROQ_API_KEY"))
        self.genai = genai.configure(api_key=os.getenv("GENAI_API_KEY"))

        self.use_case_one = UseCaseOne(client=self.groq_client)

        self.use_case_two = ImageDescriptionGenerator()

    def word_to_sentence(self, interest: str, difficulties: str, words: list):
        return self.use_case_one.generate_educational_content(interest, difficulties, words)

    def image_describer(self, image_path: str, difficulties):
        return self.use_case_two.process_image(image_path, difficulty_level=difficulties)