Update main.py

main.py

@@ -3,6 +3,8 @@ from fastapi.middleware.cors import CORSMiddleware  # Importa il middleware CORS
 from pydantic import BaseModel
 from huggingface_hub import InferenceClient
 from datetime import datetime
+from fastapi.responses import JSONResponse
+import base64
 
 app = FastAPI()
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
@@ -49,6 +51,31 @@ def read_root(request: Request, input_data: InputData):
     generated_response = generate(input_text, history, temperature, max_new_tokens, top_p, repetition_penalty)
     return {"response": generated_response}
 
+@app.get("/Immagine")
+def generate_image():
+    # Chiamata al servizio gradio_client per ottenere l'immagine
+    client = Client("https://openskyml-fast-sdxl-stable-diffusion-xl.hf.space/--replicas/545b5tw7n/")
+    result = client.predict(
+        "a giant monster hybrid of dragon and spider, in dark dense foggy forest ",
+        "",
+        25,
+        5,
+        1024,
+        1024,
+        453666937,
+        fn_index=0
+    )
+    
+    # Assume che 'result' sia l'immagine restituita come BASE64
+    image_base64 = result.get("data")  # Assumi che 'data' contenga l'immagine come BASE64
+    
+    if image_base64:
+        # Ritorna la risposta come JSON contenente il BASE64 dell'immagine
+        return JSONResponse(content={"image_base64": image_base64})
+    else:
+        # Nel caso in cui non si riesca a ottenere l'immagine
+        return JSONResponse(content={"message": "Impossibile ottenere l'immagine"}, status_code=500)
+
 @app.get("/")
 def read_general():
     return {"response": "Benvenuto. Per maggiori info vai a /docs"}  # Restituisci la risposta generata come JSON