Create app.py

app.py

@@ -0,0 +1,26 @@
+from fastapi import FastAPI, UploadFile, File, HTTPException
+import os
+import google.generativeai as genai
+from PIL import Image
+from io import BytesIO
+
+app = FastAPI()
+
+GOOGLE_API_KEY = os.getenv('GOOGLE_API_KEY')
+
+if not GOOGLE_API_KEY:
+    raise ValueError("GOOGLE_API_KEY environment variable not set")
+
+genai.configure(api_key=GOOGLE_API_KEY)
+model = genai.GenerativeModel('gemini-1.5-flash')
+
+@app.post("/generate-caption/")
+async def generate_caption(file: UploadFile = File(...)):
+    try:
+        image_data = await file.read()
+        image = Image.open(BytesIO(image_data))
+        
+        response = model.generate_content(image)
+        return {"caption": response.text}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))