Update main.py

main.py

@@ -1,81 +1,63 @@
-import os
-import shutil
-import tempfile
-from typing import List
-
-from fastapi import FastAPI, File, UploadFile, Form, HTTPException, Request
+import base64
+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 API key from environment variables.
-if not os.environ.get("GOOGLE_API_KEY"):
-    raise Exception("GOOGLE_API_KEY environment variable not set.")
-genai.configure(api_key=os.environ.get("GOOGLE_API_KEY"))
-
-# Choose the Gemini model.
-model = genai.GenerativeModel(model_name="gemini-1.5-pro-latest")
+# Configure the Google Generative AI API key.
+genai.configure(api_key="AIzaSyDcYyq3w21iwipYn17wCAQo3AYWhUIGDSI")
 
 app = FastAPI()
 templates = Jinja2Templates(directory="templates")
 
-
-def generate_content(file_paths: List[str], prompt: str) -> str:
-    """
-    Given a list of image file paths and a prompt, call the Gemini model
-    to generate content. The list provided to generate_content will contain
-    the image file paths followed by the prompt.
-    """
-    # Build the contents list: first the file paths, then the prompt.
-    contents = file_paths.copy()
-    contents.append(prompt)
+def generate_content_from_image(image_bytes: bytes) -> str:
+    # Encode the image as base64.
+    image_data = base64.b64encode(image_bytes).decode("utf-8")
     
-    try:
-        response = model.generate_content(contents)
-        if not response or not response.text:
-            raise Exception("No response text received.")
-        return response.text
-    except Exception as e:
-        print("Error generating content:", e)
-        raise e
+    # Create a prompt
+    prompt = """analyse question , analyse option , and give me question , it's options and answer in json format 
+
+fixed json format
+
+below is example of json format ,the json structure is same like below
 
+{
+  \"question\": \"What is the capital of France?\",
+  \"options\": [
+    \"a. Berlin\",
+    \"b. Madrid\",
+    \"c. Paris\",
+    \"d. Rome\"
+  ],
+  \"answer\": \"c\"
+}
+
+only json format nothing else"""
+    
+    # Use the exact code you provided.
+    model = genai.GenerativeModel(model_name="gemini-2.0-flash")
+    response = model.generate_content([
+        {'mime_type': 'image/jpeg', 'data': image_data},
+        prompt
+    ])
+    return response.text
 
 @app.get("/", response_class=HTMLResponse)
 async def index(request: Request):
     return templates.TemplateResponse("index.html", {"request": request})
 
-
 @app.post("/upload", response_class=JSONResponse)
-async def upload_images(
-    prompt: str = Form(...),
-    files: List[UploadFile] = File(...),
-):
-    # Validate that each uploaded file is an image.
-    for file in files:
-        if not file.content_type.startswith("image/"):
-            raise HTTPException(status_code=400, detail="Only image files are accepted.")
-    
-    temp_dir = tempfile.mkdtemp()
-    saved_paths = []
-    
+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:
-        # Save each uploaded image temporarily.
-        for file in files:
-            file_path = os.path.join(temp_dir, file.filename)
-            with open(file_path, "wb") as buffer:
-                shutil.copyfileobj(file.file, buffer)
-            saved_paths.append(file_path)
-        
-        # Generate the content using the images and prompt.
-        result = generate_content(saved_paths, prompt)
+        image_bytes = await file.read()
+        result = generate_content_from_image(image_bytes)
         return {"result": result}
     except Exception as e:
-        return JSONResponse(status_code=500, content={"detail": f"Error processing the image(s): {str(e)}"})
-    finally:
-        # Clean up temporary files.
-        shutil.rmtree(temp_dir, ignore_errors=True)
-
+        return JSONResponse(status_code=500, content={"detail": f"Error processing the image: {str(e)}"})
 
 if __name__ == "__main__":
     import uvicorn