Update app.py

app.py

@@ -2,7 +2,8 @@ import os
 os.environ['HF_HOME'] = '/tmp/cache'
 os.environ['TORCH_HOME'] = '/tmp/cache'
 
-from fastapi import FastAPI, File, UploadFile
+
+from fastapi import FastAPI, File, UploadFile, Response
 from fastapi.middleware.cors import CORSMiddleware
 from PIL import Image
 import torch
@@ -28,6 +29,7 @@ app.add_middleware(
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
+    expose_headers=["*"]
 )
 
 # Liste de catégories possibles en Anglais. Le modèle comprend mieux l'Anglais.
@@ -80,10 +82,18 @@ async def analyze_image(file: UploadFile = File(...)):
         hex_color = "#000000"  # Couleur noire par défault en cas d'erreur
 
     # 4. Renvoie le résultat à Lovable
-    return {
-        "category": category_name,
-        "color_hex": hex_color,
-        "confidence": round(confidence_score, 4)  # Arrondit le score de confiance à 4 décimales
+    return Response(
+        content=json.dumps({
+            "category": category_name,
+            "color_hex": hex_color,
+            "confidence": round(confidence_score, 4)
+        }),
+        media_type="application/json",
+        headers={
+            "Access-Control-Allow-Origin": "*",
+            "Access-Control-Allow-Credentials": "true"
+        }
+    )  # Arrondit le score de confiance à 4 décimales
     }
 
 # Cette partie est importante pour Hugging Face Spaces