Update app.py with multimodal OpenRouter

app.py

@@ -89,15 +89,12 @@ class ModelName(str, Enum):
     llama_3_3 = "meta-llama/llama-3.3-70b-instruct:free"
     mistral_small_3 ="mistralai/mistral-small-24b-instruct-2501:free"
     
-@app.post("/open-router")
-async def open_router(
+@app.post("/open-router/text")
+async def open_router_text(
     token: Annotated[str, Depends(verify_token)],
     model: ModelName = Query(..., description="Select a model"), 
     prompt: str = Query(..., description="Enter your prompt")
 ):  
-    print(str(open_router_key))
-    print(model)
-    print(prompt)
     async with httpx.AsyncClient() as client:
         response = await client.post(
             url="https://openrouter.ai/api/v1/chat/completions",
@@ -118,5 +115,53 @@ async def open_router(
             }
         )
 
+        response.raise_for_status()  # Raise HTTPError for bad responses (4xx or 5xx)
+        return response.json()
+
+class MultiModelName(str, Enum):
+    qwen_vl_plus = "qwen/qwen-vl-plus:free"
+    qwen_vl_72b = "qwen/qwen2.5-vl-72b-instruct:free"
+    gemini_2_flash_lite = "google/gemini-2.0-flash-lite-preview-02-05:free"
+    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")
+async def open_router_multimodal(
+    token: Annotated[str, Depends(verify_token)],
+    model: MultiModelName = Query(..., description="Select a model"), 
+    prompt: str = Query(..., description="Enter your prompt (ex: What is in this image?"),
+    image_url: str = Query(..., description="Enter the image URL"),
+): 
+    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": image_url,
+                        }
+                    }
+                    ]
+                }
+                ],
+            }
+        )
+
         response.raise_for_status()  # Raise HTTPError for bad responses (4xx or 5xx)
         return response.json()