Update main.py

main.py

@@ -10,6 +10,7 @@ import os
 import socket
 import time
 from enum import Enum
+import random
 
 #--------------------------------------------------- Definizione Server FAST API ------------------------------------------------------
 app = FastAPI()
@@ -176,7 +177,12 @@ def generate_image(request: Request, input_data: InputImage):
     
     if input_data.style:    
         print(input_data.style)
-        if input_data.style in style_image:
+        if input_data.style == 'RANDOM':
+            random_style = random.choice(list(style_image.keys()))
+            style_info = style_image[random_style]
+            input_data.input = style_info["descrizione"].format(prompt=input_data.input)
+            input_data.negativePrompt = style_info["negativePrompt"]           
+        elif input_data.style in style_image:
             style_info = style_image[input_data.style]
             input_data.input = style_info["descrizione"].format(prompt=input_data.input)
             input_data.negativePrompt = style_info["negativePrompt"]