File size: 1,414 Bytes
01746ab 31b9335 01746ab c885edb 31b9335 c885edb 01746ab 31b9335 c885edb 01746ab c885edb 01746ab 31b9335 01746ab 31b9335 c885edb 31b9335 c885edb 31b9335 c885edb 01746ab 31b9335 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 | import os
import io
import base64
from fastapi import FastAPI, HTTPException, Header
from pydantic import BaseModel
from google import genai
from PIL import Image
app = FastAPI()
# Get the key from environment variables
GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
# Initialize Gemini Client
client = genai.Client(api_key=GEMINI_API_KEY)
class ImageRequest(BaseModel):
base64_image: str
@app.post("/explain-chart")
async def explain_chart(request: ImageRequest, authorization: str = Header(None)):
# Security check (matching the key sent from Streamlit)
if not authorization or authorization != f"Bearer {GEMINI_API_KEY}":
raise HTTPException(status_code=401, detail="Unauthorized")
try:
# 1. Decode the base64 string back into an image
image_data = base64.b64decode(request.base64_image)
img = Image.open(io.BytesIO(image_data))
# 2. Call Gemini 1.5 Flash (optimized for vision/charts)
response = client.models.generate_content(
model="gemini-1.5-flash",
contents=[
"You are an expert IPL data analyst. Explain this chart and provide the key insights in concise bullet points.",
img
]
)
return {"explanation": response.text}
except Exception as e:
print(f"Error detail: {e}")
raise HTTPException(status_code=500, detail=str(e)) |