Update main.py

main.py

@@ -1,10 +1,11 @@
 import base64
+from typing import List, Tuple
+
 from fastapi import FastAPI, UploadFile, File, HTTPException, Request
 from fastapi.responses import JSONResponse, HTMLResponse
 from fastapi.templating import Jinja2Templates
 
 import google.generativeai as genai
-import httpx
 
 # Configure the Google Generative AI API key.
 genai.configure(api_key="AIzaSyDcYyq3w21iwipYn17wCAQo3AYWhUIGDSI")
@@ -12,11 +13,17 @@ genai.configure(api_key="AIzaSyDcYyq3w21iwipYn17wCAQo3AYWhUIGDSI")
 app = FastAPI()
 templates = Jinja2Templates(directory="templates")
 
-def generate_content_from_image(image_bytes: bytes) -> str:
-    # Encode the image as base64.
-    image_data = base64.b64encode(image_bytes).decode("utf-8")
+def generate_content_from_images(images: List[Tuple[str, bytes]]) -> str:
+    """
+    Given a list of tuples (mime_type, image_bytes), encode each image
+    in base64, build a list of image dicts, append the fixed prompt,
+    and call the Gemini model.
+    """
+    image_data_list = []
+    for mime_type, image_bytes in images:
+        image_data = base64.b64encode(image_bytes).decode("utf-8")
+        image_data_list.append({'mime_type': mime_type, 'data': image_data})
     
-    # Create a prompt
     prompt = """analyse question , analyse option , and give me question , it's options and answer in json format 
 
 fixed json format
@@ -36,12 +43,12 @@ below is example of json format ,the json structure is same like below
 
 only json format nothing else"""
     
-    # Use the exact code you provided.
+    payload = image_data_list.copy()
+    payload.append(prompt)
+    
+    # Use your exact code snippet.
     model = genai.GenerativeModel(model_name="gemini-2.0-flash")
-    response = model.generate_content([
-        {'mime_type': 'image/jpeg', 'data': image_data},
-        prompt
-    ])
+    response = model.generate_content(payload)
     return response.text
 
 @app.get("/", response_class=HTMLResponse)
@@ -49,15 +56,20 @@ async def index(request: Request):
     return templates.TemplateResponse("index.html", {"request": request})
 
 @app.post("/upload", response_class=JSONResponse)
-async def upload_image(file: UploadFile = File(...)):
-    if not file.content_type.startswith("image/"):
-        raise HTTPException(status_code=400, detail="Only image files are accepted.")
-    try:
+async def upload_images(files: List[UploadFile] = File(...)):
+    if not files:
+        raise HTTPException(status_code=400, detail="No files uploaded.")
+    images = []
+    for file in files:
+        if not file.content_type.startswith("image/"):
+            raise HTTPException(status_code=400, detail="Only image files are accepted.")
         image_bytes = await file.read()
-        result = generate_content_from_image(image_bytes)
+        images.append((file.content_type, image_bytes))
+    try:
+        result = generate_content_from_images(images)
         return {"result": result}
     except Exception as e:
-        return JSONResponse(status_code=500, content={"detail": f"Error processing the image: {str(e)}"})
+        return JSONResponse(status_code=500, content={"detail": f"Error processing the image(s): {str(e)}"})
 
 if __name__ == "__main__":
     import uvicorn