app/gen_ai.py

# from transformers import pipeline

# generator = pipeline("text-generation", model="gpt2")

# def generate_response(prompt: str):
#     result = generator(prompt, max_length=100, do_sample=True, temperature=0.7)
#     return result[0]["generated_text"]

from google import genai

client = genai.Client(api_key="AIzaSyAYEaSiZq7lcT5eJaVwnZJJ_UMoUTtpjJk")

def generate_response(prompt: str):
    result = client.models.generate_content(model="gemini-2.0-flash", contents=prompt)
    return result.text