update file

api.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, File, UploadFile
+from fastapi import FastAPI, File, UploadFile, HTTPException
 import numpy as np
 from io import BytesIO
 from PIL import Image, UnidentifiedImageError
@@ -30,16 +30,27 @@ async def ping():
     return {"message": "Hello, I am alive"}
 
 def read_file_as_image(data) -> np.ndarray:
-    image = np.array(Image.open(BytesIO(data)))
-    return image
-@app.post("/predict")
+    try:
+        image = Image.open(BytesIO(data))
+        return np.array(image)
+    except UnidentifiedImageError:
+        raise HTTPException(status_code=400, detail="Invalid image format")
+    except Exception as e:
+        raise HTTPException(status_code=400, detail=f"Image processing error: {str(e)}")
+
+
+@app.post("/predict/")
 async def predict(file: UploadFile = File(...)):
     image_data = await file.read()
 
+    # Dosya tipini kontrol et
+    if file.content_type not in ["image/jpeg", "image/png", "image/jpg"]:
+        raise HTTPException(status_code=400, detail="Invalid file type")
+
     try:
         image = read_file_as_image(image_data)
-    except ValueError as e:
-        return {"error": str(e)}
+    except HTTPException as e:
+        return e
 
     # Görüntüyü modelin beklediği boyuta getirme
     image = tf.image.resize(image, (299, 299))  # InceptionV3 için 299x299 olmalı