:bug: incorrect obj method - removed status printout

gradio_neutral_input_func.py

@@ -6,7 +6,7 @@ import json
 import uuid
 import os
 from stable_diffusion_demo import StableDiffusion
-from datasets import Dataset, Features, Value, Image as HFImage, load_dataset
+from datasets import Dataset, Features, Value, Image as HFImage, load_dataset, concatenate_datasets
 import tempfile
 
 # Setup directories
@@ -47,51 +47,51 @@ def load_dataset_from_hf():
 
 def save_to_hf_dataset(image, description):
     """Save new image and description to HuggingFace dataset"""
+    # try:
+    # Generate UUID for the new entry
+    image_id = str(uuid.uuid4())
+    
+    # Load existing dataset
     try:
-        # Generate UUID for the new entry
-        image_id = str(uuid.uuid4())
+        existing_dataset = load_dataset(DATASET_REPO, split="train")
+    except:
+        # Create empty dataset if it doesn't exist
+        existing_dataset = Dataset.from_dict({
+            "image": [],
+            "description": [],
+            "uuid": []
+        }).cast_column("image", HFImage())
+    
+    # Create temporary file for the image
+    with tempfile.NamedTemporaryFile(suffix='.png', delete=False) as tmp_file:
+        image.save(tmp_file.name, format='PNG')
         
-        # Load existing dataset
-        try:
-            existing_dataset = load_dataset(DATASET_REPO, split="train")
-        except:
-            # Create empty dataset if it doesn't exist
-            existing_dataset = Dataset.from_dict({
-                "image": [],
-                "description": [],
-                "uuid": []
-            }).cast_column("image", HFImage())
+        # Create new entry
+        new_entry = {
+            "image": [tmp_file.name],
+            "description": [description],
+            "uuid": [image_id]
+        }
         
-        # Create temporary file for the image
-        with tempfile.NamedTemporaryFile(suffix='.png', delete=False) as tmp_file:
-            image.save(tmp_file.name, format='PNG')
-            
-            # Create new entry
-            new_entry = {
-                "image": [tmp_file.name],
-                "description": [description],
-                "uuid": [image_id]
-            }
-            
-            # Create new dataset from the entry
-            new_dataset = Dataset.from_dict(new_entry).cast_column("image", HFImage())
-            
-            # Concatenate with existing dataset
-            if len(existing_dataset) > 0:
-                combined_dataset = existing_dataset.concatenate(new_dataset)
-            else:
-                combined_dataset = new_dataset
-            
-            # Push to HuggingFace Hub
-            combined_dataset.push_to_hub(DATASET_REPO, private=False, token=HF_TOKEN)
-            
-            # Clean up temporary file
-            os.unlink(tmp_file.name)
-            
-        return True, "Successfully saved to HuggingFace dataset!"
+        # Create new dataset from the entry
+        new_dataset = Dataset.from_dict(new_entry).cast_column("image", HFImage())
         
-    except Exception as e:
-        return False, f"Error saving to HuggingFace: {str(e)}"
+        # Concatenate with existing dataset
+        if len(existing_dataset) > 0:
+            combined_dataset = concatenate_datasets([existing_dataset, new_dataset])
+        else:
+            combined_dataset = new_dataset
+        
+        # Push to HuggingFace Hub
+        combined_dataset.push_to_hub(DATASET_REPO, private=False, token=HF_TOKEN)
+        
+        # Clean up temporary file
+        os.unlink(tmp_file.name)
+        
+    return True, "Successfully saved to HuggingFace dataset!"
+        
+    # except Exception as e:
+    #     return False, f"Error saving to HuggingFace: {str(e)}"
 
 def save_image_and_description(image, description):
     """Save the generated image and its description to HuggingFace dataset"""
@@ -118,9 +118,9 @@ def save_image_and_description(image, description):
         except:
             pass  # Local save is just backup, don't fail if it doesn't work
         
-        return message, None, load_previous_examples()
+        return None, load_previous_examples()
     else:
-        return message, None, None
+        return None, None
 
 def load_previous_examples():
     """Load examples from HuggingFace dataset"""
@@ -221,7 +221,7 @@ with gr.Blocks(title="Neutral Image App") as demo:
             image_output = gr.Image(type="pil", label="Generated Image", interactive=False)
             description_input = gr.Textbox(label="Describe the image", lines=3)
             save_btn = gr.Button("Save Image and Description")
-            status_output = gr.Textbox(label="Status")
+            # status_output = gr.Textbox(label="Status")
     
     with gr.Accordion("Previous Examples", open=False):
         gallery = gr.Gallery(
@@ -240,7 +240,7 @@ with gr.Blocks(title="Neutral Image App") as demo:
     save_btn.click(
         fn=save_image_and_description,
         inputs=[image_output, description_input],
-        outputs=[status_output, image_output, gallery]
+        outputs=[image_output, gallery]
     )
     
     refresh_btn.click(