Update main.py

main.py

@@ -6,6 +6,7 @@ from datetime import datetime
 from gradio_client import Client
 import base64
 import requests
+import os
 
 app = FastAPI()
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
@@ -65,16 +66,8 @@ def generate_image():
         453666937,
         fn_index=0
     )
-    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}
+    image_url = os.path.abspath(os.path.dirname(__file__)) + result
+    return {"response": image_url}
 
 @app.get("/Test")
 def generate_image():