Raise 500

funciones.py

@@ -1,12 +1,14 @@
 import io
 import globales
 import espacios
+import random
 import herramientas
 import gradio_client
-from huggingface_hub import InferenceClient
 import conexion_firebase
-import random
-import time
+from fastapi import HTTPException, status
+from huggingface_hub import InferenceClient
+
+servidor = globales.servidor
 
 def genera_platillo_gpu(platillo):
 
@@ -50,21 +52,22 @@ def genera_platillo_gpu(platillo):
         print("Excepción: ", e)        
         # Opción para regresar imagen genérica. (ya no porque se envía desde backend.)
         # return "default.png"
-        error = f"Error 500: {e}"
-        #return '{"Error 500": e}'
-        return error
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail=e
+        )
 
 
 def genera_platillo_inference(platillo):
 
     
-    modelo_actual = conexion_firebase.obtenDato('nowme', 'huggingface', 'modelo_actual')
+    modelo_actual = conexion_firebase.obtenDato('nowme', servidor, 'modelo_actual')
     modelo = modelo_actual
 
     print("Modelo:", modelo)
     
     #Elegir proveedor de inferencia
-    creditos_restantes_inference = conexion_firebase.obtenDato('nowme', 'huggingface', 'hfInference')
+    creditos_restantes_inference = conexion_firebase.obtenDato('nowme', servidor, 'inferencias')
     #creditos_restantes_inference = 5000
 
     #print("Los créditos restantes de hf-inference que tienes son: ", creditos_restantes_inference)
@@ -101,11 +104,14 @@ def genera_platillo_inference(platillo):
             modelo=globales.inferencia_backup
             #Escribe en txt el nuevo modelo.
             herramientas.modificaModeloActual(modelo)        
-        return f"Error: {e}"
+        print(f"Error: {e}")
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail=e
+        )
     
     img_io = io.BytesIO()
     image.save(img_io, "PNG")
     img_io.seek(0)
     print("Platillo generado:", platillo)
-    return img_io
-    
+    return img_io    

herramientas.py

@@ -33,7 +33,7 @@ def obtenSegundosDisponiblesInference():
     #     renuevaModeloPrincipal()
 
     #Finalmente obten los segundos disponibles después de las operaciones.
-    return conexion_firebase.obtenDato('nowme', servidor, 'hfInference')
+    return conexion_firebase.obtenDato('nowme', servidor, 'inferencias')
     
     
 def renuevaSegundosDisponibles():
@@ -74,7 +74,7 @@ def restaSegundosInference(cuantos_segundos):
     # Restar los segundos
     nuevos_segundos_disponibles = segundos_disponibles - cuantos_segundos
     print("Segundos disponibles ahora en restaSegundosInference: ", nuevos_segundos_disponibles)
-    conexion_firebase.editaDato('nowme', servidor, 'hfInference', nuevos_segundos_disponibles)    
+    conexion_firebase.editaDato('nowme', servidor, 'inferencias', nuevos_segundos_disponibles)    
 
 def modificaModeloActual(nuevo_modelo):
     """