Create app.py

app.py

@@ -0,0 +1,39 @@
+from fastapi import FastAPI, File, UploadFile, HTTPException
+from roboflow import Roboflow
+import shutil
+from pathlib import Path
+
+# Inicializar la aplicación FastAPI
+app = FastAPI()
+
+# Inicializar Roboflow
+rf = Roboflow(api_key="z15djNx8oHjsud3dWL4A")
+project = rf.workspace().project("stine")
+model = project.version(3).model
+
+# Carpeta temporal para guardar imágenes cargadas
+UPLOAD_DIR = Path("uploads")
+UPLOAD_DIR.mkdir(exist_ok=True)
+
+@app.post("/predict/")
+async def predict_image(file: UploadFile = File(...)):
+    """
+    Endpoint para predecir una imagen usando el modelo de Roboflow.
+    """
+    # Verificar el tipo de archivo
+    if file.content_type not in ["image/jpeg", "image/png"]:
+        raise HTTPException(status_code=400, detail="El archivo debe ser una imagen (JPEG o PNG)")
+
+    # Guardar el archivo temporalmente
+    temp_file = UPLOAD_DIR / file.filename
+    with temp_file.open("wb") as buffer:
+        shutil.copyfileobj(file.file, buffer)
+
+    # Realizar la predicción
+    try:
+        prediction = model.predict(str(temp_file), confidence=40, overlap=30).json()
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Error al realizar la predicción: {e}")
+
+    # Devolver la predicción como respuesta
+    return prediction