main.py

import os
import shutil
from fastapi import FastAPI, File, UploadFile, HTTPException, Request
from fastapi.responses import JSONResponse, HTMLResponse
from fastapi.templating import Jinja2Templates

# Import the required modules from google.genai
from google import genai
from google.genai import types

app = FastAPI()
templates = Jinja2Templates(directory="templates")

def generate_question(image_path: str) -> str:
    # Initialize the client using your GEMINI_API_KEY environment variable.
    client = genai.Client(api_key=os.environ.get("GEMINI_API_KEY"))

    # Upload the image file
    files = [client.files.upload(file=image_path)]
    model = "gemini-2.0-flash"

    # Prepare the content using the image URI and provided sample text
    contents = [
        types.Content(
            role="user",
            parts=[
                types.Part.from_uri(
                    file_uri=files[0].uri,
                    mime_type=files[0].mime_type,
                ),
            ],
        ),
        types.Content(
            role="model",
            parts=[
                types.Part.from_text(text="""You are absolutely correct! My apologies for the repeated errors. The correct answer is 24 degrees.

```json
{
  \"question\": \"This horizontal bar chart displays the average daily temperatures recorded for each day of the week. If the temperature on Thursday was increased by 1/4th times of the second highest temperature of the week, what temperature would be experienced on Thursday?\",
  \"options\": [
    \"a. 35°\",
    \"b. 24°\",
    \"c. 18°\",
    \"d. 30°\"
  ],
  \"answer\": \"b\"
}
```"""),
            ],
        ),
        types.Content(
            role="user",
            parts=[
                types.Part.from_text(text="INSERT_INPUT_HERE"),
            ],
        ),
    ]
    tools = [types.Tool(function_declarations=[])]
    generate_content_config = types.GenerateContentConfig(
        temperature=1,
        top_p=0.95,
        top_k=40,
        max_output_tokens=8192,
        tools=tools,
        response_mime_type="text/plain",
        system_instruction=[
            types.Part.from_text(text="""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
"""),
        ],
    )

    # Generate content stream and collect the response
    response_text = ""
    for chunk in client.models.generate_content_stream(
        model=model,
        contents=contents,
        config=generate_content_config,
    ):
        if chunk.text:
            response_text += chunk.text
        elif chunk.function_calls:
            response_text += chunk.function_calls[0]

    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_image(file: UploadFile = File(...)):
    if not file.content_type.startswith("image/"):
        raise HTTPException(status_code=400, detail="Only image files are accepted.")
    
    # Save uploaded file temporarily
    temp_dir = "temp"
    os.makedirs(temp_dir, exist_ok=True)
    file_path = os.path.join(temp_dir, file.filename)
    with open(file_path, "wb") as buffer:
        shutil.copyfileobj(file.file, buffer)
    
    try:
        result = generate_question(file_path)
    finally:
        os.remove(file_path)
    
    return {"result": result}

if __name__ == "__main__":
    import uvicorn
    uvicorn.run(app, host="0.0.0.0", port=7860)