have to test live

app.py

@@ -226,61 +226,55 @@ def create_gradio_interface():
             if not profile or "username" not in profile:
                 raise gr.Error("Please log in to use this application")
             username = profile["username"]
-            repo_id = f"{username}/handwriting-ocr-private"
-            
+
+            # Common processing: strip metadata, get timestamp, create features, and setup temp directory.
+            stripped_image = strip_metadata(image)
+            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+            features = datasets.Features({
+                'text': datasets.Value('string'),
+                'image': datasets.Image(),
+                'timestamp': datasets.Value('string')
+            })
+            temp_dir = "temp"
+            os.makedirs(temp_dir, exist_ok=True)
+
+            # Define targets based on checkboxes: each entry is (dataset_type, repo_id, suffix, privacy_flag)
+            targets = []
+            if public_checkbox:
+                targets.append(("public", "rawwerks/handwriting-ocr-all", "_public", False))
             if private_checkbox:
-                # Remove all metadata for privacy
-                stripped_image = strip_metadata(image)
-                
-                # Check if the dataset exists; if not, create it as private
+                targets.append(("private", f"{username}/handwriting-ocr-private", "_private", True))
+
+            # Loop over each target, pushing the dataset with shared logic.
+            for ds_type, repo_id, suffix, is_private in targets:
                 try:
                     collector.hf_api.dataset_info(repo_id)
                 except Exception as e:
-                    collector.hf_api.create_repo(repo_id, repo_type="dataset", private=True)
-                
-                # Save the stripped image to a temporary file
-                timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-                filename = f"{timestamp}.png"
-                temp_dir = "temp"
-                os.makedirs(temp_dir, exist_ok=True)
+                    collector.hf_api.create_repo(repo_id, repo_type="dataset", private=is_private)
+
+                filename = f"{timestamp}{suffix}.png"
                 temp_path = os.path.join(temp_dir, filename)
                 stripped_image.save(temp_path)
-                
-                # Create a dataset dictionary with the image-text pair
-                features = datasets.Features({
-                    'text': datasets.Value('string'),
-                    'image': datasets.Image(),
-                    'timestamp': datasets.Value('string')
-                })
-                
+
                 dataset_dict = {
                     'text': [text],
                     'image': [temp_path],
                     'timestamp': [timestamp]
                 }
-                
-                # Create the dataset and push to hub
                 dataset = datasets.Dataset.from_dict(dataset_dict, features=features)
                 dataset.push_to_hub(repo_id)
-                
-                # Remove the temporary file
                 os.remove(temp_path)
-                
-                # Log the submission locally
+
                 collector.collected_pairs.append({
                     "text": text,
                     "image": image,
                     "timestamp": timestamp,
                     "username": username,
-                    "dataset": "private"
+                    "dataset": ds_type
                 })
-                
-                new_text = collector.get_random_text_block(max_words)
-                return None, new_text
-            elif public_checkbox:
-                # Fallback to public submission
-                new_text = collector.get_random_text_block(max_words)
-                return None, new_text
+
+            new_text = collector.get_random_text_block(max_words)
+            return None, new_text
 
         def handle_regenerate(profile, text, max_words):
             # Remove the login check - allow anyone to regenerate text