import os
from dotenv import load_dotenv
import google.generativeai as genai

load_dotenv()

SYS_INSTR = "You are a plant disease expert. You will be given queries regarding plant diseases. Always respond in Markdown"
TXT_PROMPT = "Suggest remedy for the disease in bullet points"
IMG_TXT_PROMPT = "Based on the given image, suggest the possible disease the plant is suffering from, along with the remedy in 150 words"


def llm_strategy(llm_name, disease_name, image_file=None):
    if llm_name.lower() == "gemini":
        return get_response_from_gemini(disease_name, image_file)
    else:
        raise ValueError(f"LLM {llm_name} not supported")


def get_response_from_gemini(disease_name, image_file=None) -> str:
    genai.configure(api_key=os.environ["GEMINI_API_KEY"])
    model = genai.GenerativeModel("gemini-1.5-flash", system_instruction=SYS_INSTR)

    generation_config = genai.GenerationConfig(max_output_tokens=300)

    prompt = [TXT_PROMPT, disease_name]
    if image_file:
        prompt = [IMG_TXT_PROMPT, image_file]
    response = model.generate_content(prompt, generation_config=generation_config)
    return response.text