Spaces:

buchi-stdesign
/

Gemini_API_iPhone

Sleeping

Upload 5 files

c8ffa6f verified over 1 year ago

548 Bytes

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}

	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}