File size: 1,472 Bytes
231f8b8 af03415 231f8b8 58cf445 231f8b8 766f4ca 231f8b8 73eeafc 231f8b8 |
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 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
from fastapi import FastAPI, File, Form, UploadFile, Request, HTTPException
from typing import Annotated, Optional
from fastapi.responses import JSONResponse
from pydantic import BaseModel
from PIL import Image
import os
import google.generativeai as genai
import google.ai.generativelanguage as glm
from typing import Dict
import io
from fastapi.middleware.cors import CORSMiddleware
from imageFunct import get_image_data
# Initialize FastAPI app
app = FastAPI()
# Configure API key for Google Generative AI
os.environ['GOOGLE_API_KEY'] = "AIzaSyDAG_Xl66vh4ceY81UXe3vrdwP6wIAkpBs"
genai.configure(api_key=os.environ['GOOGLE_API_KEY'])
# Configure CORS
origins = [
"http://localhost:3000",
]
app.add_middleware(
CORSMiddleware,
allow_origins=origins,
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
@app.get("/")
async def read_index():
return "This is the index page"
@app.post('/classify')
async def main(
image: UploadFile = File(...),
action_id: Optional[str] = Form(None)
):
try:
image_data = await image.read()
image = Image.open(io.BytesIO(image_data))
result = get_image_data(image)
return JSONResponse(content=result)
except Exception as e:
raise HTTPException(status_code=500, detail=f"Error generating content: {e}")
# Run the application
if __name__ == "__main__":
import uvicorn
uvicorn.run(app, host="0.0.0.0", port=7980)
|