Update main.py

main.py

@@ -4,6 +4,8 @@ from pydantic import BaseModel
 from huggingface_hub import InferenceClient
 from datetime import datetime
 from gradio_client import Client
+import base64
+import requests
 
 app = FastAPI()
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
@@ -63,7 +65,16 @@ def generate_image():
         453666937,
         fn_index=0
     )
-    return {"response": result}
+    image_url = result
+    if image_url:
+        image_response = requests.get(image_url)
+        if image_response.status_code == 200:
+            image_base64 = base64.b64encode(image_response.content).decode('utf-8')
+            return {"image_base64": image_base64}
+        else:
+            return {"message": "Impossibile ottenere l'immagine dall'URL", "status_code": image_response.status_code}
+    else:
+        return {"message": "Impossibile ottenere l'URL dell'immagine dalla risposta", "response": result}
 
 @app.get("/Test")
 def generate_image():