Return State

app.py

@@ -53,7 +53,7 @@ def perform(input1, input2):
     else:
         info_window = "Out of credits..."
         path = 'no-result.png'
-        return path, info_window, btn_buy
+        return path, info_window, btn_buy, gr.State.tokens
 
     print(f"El path final fue {path}, si es no-result, no debites y controla la info window.")
     print(f"El type de path es: ", type(path))
@@ -81,7 +81,7 @@ def perform(input1, input2):
         #No se hizo un proceso, por lo tanto no debitaremos.
         #En el futuro, como regla de negocio, podría cambiar y que si debitemos.  
     
-    return path, info_window, btn_buy
+    return path, info_window, btn_buy, gr.State.tokens
 
 def click_buy():
 
@@ -135,7 +135,7 @@ with gr.Blocks(theme=gr.themes.Base(), css="footer {visibility: hidden}") as mai
             fn=perform,
             title="", 
             inputs=[source_image, destination_image], 
-            outputs=[result_image, lbl_console, btn_buy], 
+            outputs=[result_image, lbl_console, btn_buy, lbl_credits], 
             )
 
 main.launch(auth=authenticate)