Oktoberfest, Ghibli and All

app.py

@@ -16,7 +16,8 @@ async def partynight(
 @app.post("/babydoll/")
 async def babydoll(
     image: UploadFile = File(...),
-    outfit_color: Literal["red", "black", "blue", "golden", "pink", "green", "yellow", "purple", "silver", "white", "blue sky", "light blue", "royal blue", "navy blue", "orange", "random", ""] = Form(None)):    
+    outfit_color: Literal["red", "black", "blue", "golden", "pink", "green", "yellow", "purple", "silver", "white", "blue sky", "light blue", "royal blue", "navy blue", "orange", "random", ""] = Form(None)
+    ):    
     if not image.content_type.startswith("image/"):
         return {"error": "The file is not an image."}    
     return await funciones.crea_babydoll(image, outfit_color)
@@ -24,10 +25,11 @@ async def babydoll(
 @app.post("/leather/")
 async def leather(
     image: UploadFile = File(...),
-    outfit_color: Literal["red", "black", "blue", "golden", "pink", "green", "yellow", "purple", "silver", "white", "blue sky", "light blue", "royal blue", "navy blue", "orange", "random", ""] = Form(None)):    
+    #outfit_color: Literal["red", "black", "blue", "golden", "pink", "green", "yellow", "purple", "silver", "white", "blue sky", "light blue", "royal blue", "navy blue", "orange", "random", ""] = Form(None)
+    ):    
     if not image.content_type.startswith("image/"):
         return {"error": "The file is not an image."}    
-    return await funciones.crea_leather(image, outfit_color)  
+    return await funciones.crea_leather(image)  
 
 @app.post("/stewardess/")
 async def stewardess(
@@ -35,4 +37,22 @@ async def stewardess(
     color: Literal["red", "black", "blue", "golden", "pink", "green", "yellow", "purple", "silver", "white", "blue sky", "light blue", "royal blue", "navy blue", "orange", "random", ""] = Form(None)):    
     if not image.content_type.startswith("image/"):
         return {"error": "The file is not an image."}    
-    return await funciones.crea_stewardess(image, color) 
+    return await funciones.crea_stewardess(image, color)
+
+@app.post("/oktoberfest/")
+async def oktoberfest(
+    image: UploadFile = File(...),
+    #color: Literal["red", "black", "blue", "golden", "pink", "green", "yellow", "purple", "silver", "white", "blue sky", "light blue", "royal blue", "navy blue", "orange", "random", ""] = Form(None)
+    ):    
+    if not image.content_type.startswith("image/"):
+        return {"error": "The file is not an image."}    
+    return await funciones.crea_oktoberfest(image)
+
+@app.post("/ghibli/")
+async def ghibli(
+    image: UploadFile = File(...),
+    #color: Literal["red", "black", "blue", "golden", "pink", "green", "yellow", "purple", "silver", "white", "blue sky", "light blue", "royal blue", "navy blue", "orange", "random", ""] = Form(None)
+    ):    
+    if not image.content_type.startswith("image/"):
+        return {"error": "The file is not an image."}    
+    return await funciones.crea_ghibli(image)  

funciones.py

@@ -32,10 +32,10 @@ async def crea_babydoll(image, outfit_color):
         print(f"Error: {e}")
         return {"error": "Unable to detect a face in the image. Please upload a different photo with a clear face."}
     
-async def crea_leather(image, outfit_color):
+async def crea_leather(image):
     endpoint = 'leather'
     try:                   
-        prompt = databank.obtenPrompt(endpoint, outfit_color=outfit_color)
+        prompt = databank.obtenPrompt(endpoint)
         resultado = await herramientas.exeCute(endpoint, image, prompt)
         if type(resultado) is str:
             return FileResponse(resultado, media_type="image/jpg", filename=f"{endpoint}-{timestamp_segundos}.jpg")
@@ -54,6 +54,32 @@ async def crea_stewardess(image, outfit_color):
             return FileResponse(resultado, media_type="image/jpg", filename=f"{endpoint}-{timestamp_segundos}.jpg")
         elif type(resultado) is dict:
             return resultado     
+    except Exception as e:
+        print(f"Error: {e}")
+        return {"error": "Unable to detect a face in the image. Please upload a different photo with a clear face."}
+    
+async def crea_oktoberfest(image):
+    endpoint = 'oktoberfest'
+    try:
+        prompt = databank.obtenPrompt(endpoint)
+        resultado = await herramientas.exeCute(endpoint, image, prompt) #execute ya debe recibir prompt hecho.
+        if type(resultado) is str:
+            return FileResponse(resultado, media_type="image/jpg", filename=f"{endpoint}-{timestamp_segundos}.jpg")
+        elif type(resultado) is dict:
+            return resultado     
+    except Exception as e:
+        print(f"Error: {e}")
+        return {"error": "Unable to detect a face in the image. Please upload a different photo with a clear face."}
+    
+async def crea_ghibli(image):
+    endpoint = 'ghibli'
+    try:
+        prompt = databank.obtenPrompt(endpoint)
+        resultado = await herramientas.exeCute(endpoint, image, prompt) #execute ya debe recibir prompt hecho.
+        if type(resultado) is str:
+            return FileResponse(resultado, media_type="image/jpg", filename=f"{endpoint}-{timestamp_segundos}.jpg")
+        elif type(resultado) is dict:
+            return resultado     
     except Exception as e:
         print(f"Error: {e}")
         return {"error": "Unable to detect a face in the image. Please upload a different photo with a clear face."}

herramientas.py

@@ -67,8 +67,10 @@ def getPosition(carpeta_positions):
     Returns:
     bool: True si se guardó el archivo correctamente.
     """
-
+    #Con carpeta específica.
     ruta_carpeta = os.path.join("images", "positions", carpeta_positions)
+    #O una carpeta específica para todas.
+    ruta_carpeta = os.path.join("images", "positions", "all")
         
     try: 
         lista_archivos = os.listdir(ruta_carpeta)