app.py

from fastapi import FastAPI
import google.generativeai as genai
import os

# Initialize FastAPI app
app = FastAPI()

# Load API key from Hugging Face Secrets
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
genai.configure(api_key=GEMINI_API_KEY)

@app.get("/")
def home():
    return {"message": "Welcome to Gemini FastAPI Wrapper!"}

@app.get("/generate/")
def generate_content(prompt: str):
    try:
        model = genai.GenerativeModel("gemini-1.5-flash")
        response = model.generate_content(prompt)
        return {"response": response.text}
    except Exception as e:
        return {"error": str(e)}