Update app.py with Open Router multimodal with base64 support

app.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, Depends, Query
+from fastapi import FastAPI, Depends, Query, File, UploadFile
 from fastapi.responses import StreamingResponse
 from pydantic import BaseModel
 from typing import Annotated
@@ -125,7 +125,7 @@ class MultiModelName(str, Enum):
     gemini_2_pro = "google/gemini-2.0-pro-exp-02-05:free"
     llama_3_2_vision = "meta-llama/llama-3.2-11b-vision-instruct:free"
 
-@app.post("/open-router/multimodal")
+@app.post("/open-router/multimodal-url")
 async def open_router_multimodal(
     token: Annotated[str, Depends(verify_token)],
     model: MultiModelName = Query(..., description="Select a model"), 
@@ -164,4 +164,49 @@ async def open_router_multimodal(
         )
 
         response.raise_for_status()  # Raise HTTPError for bad responses (4xx or 5xx)
-        return response.json()
+        return response.json()
+
+@app.post("/open-router/multimodal-b64")
+async def open_router_multimodal_upload(
+    token: Annotated[str, Depends(verify_token)], 
+    image: UploadFile = File(...),
+    prompt: str = Query(..., description="Enter your prompt (ex: What is in this image?")
+):
+    image_bytes = await image.read()
+    encoded_string = base64.b64encode(image_bytes).decode('utf-8')
+    img = f"data:{image.content_type};base64,{encoded_string}"
+
+    async with httpx.AsyncClient() as client:
+    response = await client.post(
+        url="https://openrouter.ai/api/v1/chat/completions",
+        headers={
+            "Authorization": f"Bearer {str(open_router_key)}",
+            "Content-Type": "application/json",
+            "HTTP-Referer": "<YOUR_SITE_URL>",  # Optional
+            "X-Title": "<YOUR_SITE_NAME>",  # Optional
+        },
+        json={
+            "model": model,
+            "messages": [
+            {
+                "role": "user",
+                "content": [
+                {
+                    "type": "text",
+                    "text": prompt,
+                },
+                {
+                    "type": "image_url",
+                    "image_url": {
+                      "url": img,
+                    }
+                }
+                ]
+            }
+            ],
+        }
+    )
+
+    response.raise_for_status()  # Raise HTTPError for bad responses (4xx or 5xx)
+    return response.json()
+