Update app.py

app.py

@@ -45,13 +45,19 @@ async def chat_with_text(request: TextRequest):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-@app.post("/chat/image")
-async def chat_with_image(file: UploadFile = File(...)):
-    # Read and encode the image file to base64
-    image_data = await file.read()
-    base64_image = base64.b64encode(image_data).decode("utf-8")
 
-    # Prepare the payload for the NVIDIA API
+
+@app.post("/chat/image-url")
+async def chat_with_image_url(request: ImageURLRequest):
+    try:
+        img_response = requests.get(request.url)
+        if img_response.status_code != 200:
+            raise HTTPException(status_code=400, detail="Failed to fetch image from URL")
+        image_data = img_response.content
+        base64_image = base64.b64encode(image_data).decode("utf-8")
+    except Exception as e:
+        raise HTTPException(status_code=400, detail=f"Image download failed: {e}")
+
     payload = {
         "model": "meta/llama-3.2-90b-vision-instruct",
         "messages": [