from fastapi import FastAPI, Request
import os
import google.generativeai as genai
from dotenv import load_dotenv

load_dotenv()
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")

genai.configure(api_key=GEMINI_API_KEY)
model = genai.GenerativeModel("gemini-1.5-flash")

app = FastAPI()

@app.post("/story")
async def generate_story(request: Request):
    data = await request.json()
    prompt = data.get("prompt", "3分以内で眠れるおとぎ話をください。")
    response = model.generate_content(prompt)
    return {"story": response.text}