Finished winner prompts!

app.py

@@ -1,17 +1,20 @@
+import funciones
 from fastapi import FastAPI, Form
 from fastapi import FastAPI, File, UploadFile
-import funciones
 from typing import Literal
 
 app = FastAPI()
 
+#Future Importante, los errores como no detección de rostro, deben ser devueltos por ésta API.
+
 @app.post("/partynight/")
 async def partynight(
     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_partynight(image, outfit_color)    
+    return await funciones.crea_partynight(image, outfit_color) 
 
 @app.post("/babydoll/")
 async def babydoll(
@@ -73,4 +76,50 @@ async def bride(
     ):    
     if not image.content_type.startswith("image/"):
         return {"error": "The file is not an image."}    
-    return await funciones.crea_bride(image) 
+    return await funciones.crea_bride(image) 
+
+@app.post("/swimsuit/")
+async def swimsuit(
+    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_swimsuit(image, color) 
+
+@app.post("/lingerie/")
+async def lingerie(
+    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_lingerie(image)
+
+@app.post("/schooluniform/")
+async def schooluniform(
+    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_schooluniform(image)
+
+@app.post("/frenchmaid/")
+async def frenchmaid(
+    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_frenchmaid(image)
+
+@app.post("/cheerleader/")
+async def cheerleader(
+    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),
+    hairstyle: Literal["ponytail", "pigtails", ""] = Form(None) 
+    ):    
+    if not image.content_type.startswith("image/"):
+        return {"error": "The file is not an image."}    
+    return await funciones.crea_cheerleader(image, color, hairstyle)

databank.py

@@ -7,31 +7,31 @@ def obtenPrompt(endpoint, **kwargs): #Son opcionales todos los demás argumentos
     hair_style = kwargs["hair_style"] if "hair_style" in kwargs else ""
 
     if endpoint == "partynight":
-        return f"A manga picture of a beautiful girl with a sexy {outfit_color} sequins cocktail dress."
+        return f"A manga picture of a beautiful girl with a sexy {outfit_color} sequins cocktail dress"
     elif endpoint == "babydoll":
-        return f"A manga picture of a beautiful girl with a sexy {outfit_color} babydoll dress."
+        return f"A manga picture of a beautiful girl with a sexy {outfit_color} babydoll dress"
     elif endpoint == "leather":
-        return f"A {defineEstilo('leather')} picture of a beautiful girl with a neck chocker, a black leather skirt and black deep-cut long sleeved blouse."
+        return f"A {defineEstilo('leather')} picture of a beautiful girl with a neck chocker, a black leather skirt and black deep-cut long sleeved blouse"
     elif endpoint == "oktoberfest":
-        return f"A Ghibli manga of a beautiful Oktoberfest waitress with big boobs serving beers."
+        return f"A Ghibli manga of a beautiful Oktoberfest waitress with big boobs serving beers"
     elif endpoint == "ghibli":
-        return f"A Ghibli manga of a beautiful girl."
+        return f"A Ghibli manga of a beautiful girl"
     elif endpoint == "wildwest":
-        return f"A Ghibli manga of a beautiful wild west girl with a poncho, a revolver and bullets bandolier."
+        return f"A Ghibli manga of a beautiful wild west girl with a poncho, a revolver and bullets bandolier"
     elif endpoint == "stewardess":  
-        return f"A Ghibli manga of a beautiful flight attendant in a {outfit_color} uniform skirt with a {obtenAccesorio('stewardess')}."
+        return f"A Ghibli manga of a beautiful flight attendant in a {outfit_color} uniform skirt with a {obtenAccesorio('stewardess')}"
     elif endpoint == "bride":
-        return f"A {defineEstilo('bride')} art nouveau of a beautiful bride in a spectacular wedding dress."
+        return f"A {defineEstilo('bride')} art nouveau of a beautiful bride in a spectacular wedding dress"
     elif endpoint == "swimsuit":
-        return f"A manga picture of a beautiful girl in {outfit_color} swimsuit."
+        return f"A manga picture of a beautiful girl in {outfit_color} swimsuit"
     elif endpoint == "lingerie":
-        return f"A manga picture of a girl in corset and hosiery."
+        return f"A manga picture of a girl in corset and hosiery"
     elif endpoint == "schooluniform":
-        return f"A manga picture of a girl in school uniform."
+        return f"A manga picture of a girl in school uniform"
     elif endpoint == "frenchmaid":
-        return f"A manga picture of a girl in french maid uniform with maid's tiara."
+        return f"A manga picture of a girl in french maid uniform with maid's tiara"
     elif endpoint == "cheerleader":
-        return f"A manga picture of a cheerleader with {outfit_color} uniform and {hair_style}."
+        return f"A manga picture of a cheerleader with {outfit_color} uniform and {hair_style}"
     
 def obtenAccesorio(endpoint):
 

funciones.py

@@ -3,20 +3,17 @@ import time
 import databank
 from fastapi.responses import FileResponse
 
-timestamp_segundos = int(time.time())
-print(f"Printed at: {timestamp_segundos}.") 
-
 async def crea_partynight(image, outfit_color):    
     endpoint = 'partynight'
     try:
         prompt = databank.obtenPrompt(endpoint, outfit_color=outfit_color) #Van a tener que ser kwargs porque variará de lo recibido de ep a ep.                                      
         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")
-        elif type(resultado) is dict:
-            return resultado            
+        if type(resultado) is str:            
+            return FileResponse(resultado, media_type="image/jpg", filename=herramientas.formaFilename(prompt))
+        elif type(resultado) is dict:            
+            return resultado         
     except Exception as e:
-        print(f"Error: {e}")
+        print(f"Error 182: {e}")
         return {"error": e}    
 
 async def crea_babydoll(image, outfit_color):
@@ -25,7 +22,7 @@ async def crea_babydoll(image, outfit_color):
         prompt = databank.obtenPrompt(endpoint, outfit_color=outfit_color)
         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")
+            return FileResponse(resultado, media_type="image/jpg", filename=herramientas.formaFilename(prompt))
         elif type(resultado) is dict:
             return resultado     
     except Exception as e:
@@ -39,7 +36,7 @@ async def crea_leather(image):
         print("Prompt final: ", prompt)
         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")
+            return FileResponse(resultado, media_type="image/jpg", filename=herramientas.formaFilename(prompt))
         elif type(resultado) is dict:
             return resultado     
     except Exception as e:
@@ -52,7 +49,7 @@ async def crea_stewardess(image, outfit_color):
         prompt = databank.obtenPrompt(endpoint, outfit_color=outfit_color)
         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")
+            return FileResponse(resultado, media_type="image/jpg", filename=herramientas.formaFilename(prompt))
         elif type(resultado) is dict:
             return resultado     
     except Exception as e:
@@ -65,7 +62,7 @@ async def crea_oktoberfest(image):
         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")
+            return FileResponse(resultado, media_type="image/jpg", filename=herramientas.formaFilename(prompt))
         elif type(resultado) is dict:
             return resultado     
     except Exception as e:
@@ -78,7 +75,7 @@ async def crea_ghibli(image):
         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")
+            return FileResponse(resultado, media_type="image/jpg", filename=herramientas.formaFilename(prompt))
         elif type(resultado) is dict:
             return resultado     
     except Exception as e:
@@ -91,7 +88,7 @@ async def crea_wildwest(image):
         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")
+            return FileResponse(resultado, media_type="image/jpg", filename=herramientas.formaFilename(prompt))
         elif type(resultado) is dict:
             return resultado     
     except Exception as e:
@@ -104,10 +101,76 @@ async def crea_bride(image):
         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")
+            return FileResponse(resultado, media_type="image/jpg", filename=herramientas.formaFilename(prompt))
+        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_swimsuit(image, outfit_color):
+    endpoint = 'swimsuit'
+    try:
+        prompt = databank.obtenPrompt(endpoint, outfit_color=outfit_color)
+        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=herramientas.formaFilename(prompt))
         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."}    
+        return {"error": "Unable to detect a face in the image. Please upload a different photo with a clear face."} 
+
+async def crea_lingerie(image):
+    endpoint = 'lingerie'
+    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=herramientas.formaFilename(prompt))
+        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_schooluniform(image):
+    endpoint = 'schooluniform'
+    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=herramientas.formaFilename(prompt))
+        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_frenchmaid(image):
+    endpoint = 'frenchmaid'
+    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=herramientas.formaFilename(prompt))
+        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_cheerleader(image, color, hairstyle):
+    endpoint = 'cheerleader'
+    try:
+        prompt = databank.obtenPrompt(endpoint, outfit_color=color, hair_style = hairstyle)
+        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=herramientas.formaFilename(prompt))
+        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

@@ -2,7 +2,7 @@ import os
 import socket
 import random
 import globales
-import databank
+import time
 import tempfile
 import gradio_client
 
@@ -84,7 +84,6 @@ def getPosition(carpeta_positions):
     
 async def exeCute(endpoint, image, prompt):
 
-    print("Prompt to exeCute: ")
     print(prompt)
 
     temp_image = await imageToTemp(image)
@@ -117,11 +116,24 @@ async def exeCute(endpoint, image, prompt):
         return result[0] 
 
     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."} 
+        print(f"Error en exeCute: {e}")        
+        print("E es del tipo : ", type(e))
+        if "PAUSED" in str(e): 
+            return {"error": "Space Paused"}
+        elif "timed out" in str(e):
+            return {"error": "Network error try again."}
+        else: 
+            return {"error": "Face not detected, please try again or with other pic."} 
     finally: 
         borrado = borraTemp(temp_image) 
         print("El resultado de borrarTemp fue: ", borrado)
+
+def formaFilename(prompt):
+
+    timestamp_segundos = int(time.time())
+    print(f"Printed at: {timestamp_segundos}.") 
+    
+    return f"{prompt}-{timestamp_segundos}.jpg"