Update app.py

app.py

@@ -45,19 +45,18 @@ token = os.getenv("HF_TOKEN")
 # Nome do dataset no Hugging Face
 DATASET_NAME = "vcollos/inferencias"
 
-# Função para salvar inferências no dataset Hugging Face
-def salvar_inferencia_no_dataset(prompt, image, seed, lora_repo1_selected):
+def salvar_inferencia_no_dataset(prompt, image, seed, selected_lora):
     try:
-        # Carregar o dataset existente
+        # Carregar o dataset do Hugging Face
         dataset = load_dataset(DATASET_NAME)['train']
     except FileNotFoundError:
-        # Criar o dataset vazio caso ele não exista
+        # Criar dataset vazio caso ele não exista
         dataset = Dataset.from_dict({
-            "prompt": [],
-            "image": [],
-            "seed": [],
-            "datetime": [],
-            "lora_repo1": []
+            "prompt": [],       
+            "image": [],        
+            "seed": [],         
+            "datetime": [],     
+            "lora_repo1": []    
         })
 
     # Converter a imagem para bytes
@@ -65,19 +64,26 @@ def salvar_inferencia_no_dataset(prompt, image, seed, lora_repo1_selected):
     image.save(buffered, format="PNG")
     image_bytes = buffered.getvalue()
 
-    # Dados da nova inferência
+    # Adicionar o novo item como dicionário
     new_data = {
-        "prompt": prompt,
-        "image": [image_bytes],  # Armazenar a imagem como bytes
-        "seed": seed,
-        "datetime": [datetime.datetime.now().isoformat()],
-        "lora_repo1": [lora_repo1_selected]
+        "prompt": [prompt],  
+        "image": [image_bytes],  
+        "seed": [seed],  
+        "datetime": [datetime.datetime.now().isoformat()],  
+        "lora_repo1": [selected_lora["repo"]]  
     }
 
-    # Adicionar os novos dados ao dataset existente
-    updated_dataset = dataset.add_item(new_data)
-
-    # Subir para o Hugging Face Hub
+    # Criar um novo dataset com os dados existentes e os novos dados
+    new_dataset = Dataset.from_dict(new_data)
+    updated_dataset = Dataset.from_dict({
+        "prompt": dataset["prompt"] + new_data["prompt"],
+        "image": dataset["image"] + new_data["image"],
+        "seed": dataset["seed"] + new_data["seed"],
+        "datetime": dataset["datetime"] + new_data["datetime"],
+        "lora_repo1": dataset["lora_repo1"] + new_data["lora_repo1"]
+    })
+
+    # Subir o dataset atualizado para o Hugging Face Hub
     updated_dataset.push_to_hub(DATASET_NAME, commit_message="Adicionando nova inferência.")
 
 def calculate_shift(