Update app.py

app.py

@@ -46,9 +46,12 @@ def analyze_human(clothing_color: str, estimated_action: str):
 # Save and upload image to HF
 # -----------------------------------------------------
 def save_and_upload_image(image_b64: str, hf_token: str):
-    """Decode base64 image, save locally, and upload to HF dataset repo."""
+    """Decodes a base64 image, saves it locally, and uploads to Hugging Face Hub."""
     try:
         image_bytes = base64.b64decode(image_b64)
+        size_bytes = len(image_bytes)
+
+        # Ensure the /tmp directory exists
         os.makedirs("/tmp", exist_ok=True)
         timestamp = datetime.now().strftime("%Y%m%d_%H%M%S_%f")
         local_path = f"/tmp/robot_img_{timestamp}.jpg"
@@ -57,12 +60,23 @@ def save_and_upload_image(image_b64: str, hf_token: str):
             f.write(image_bytes)
 
         filename = f"robot_{timestamp}.jpg"
-        upload_file(local_path, path_in_repo=filename, repo_id=HF_DATASET_REPO, token=hf_token, repo_type="dataset")
+
+        # Corrected Hugging Face Hub upload
+        from huggingface_hub import HfApi
+        api = HfApi()
+        api.upload_file(
+            path_or_fileobj=local_path,
+            path_in_repo=filename,
+            repo_id=HF_DATASET_REPO,
+            repo_type="dataset",
+            token=hf_token
+        )
+
         url = f"https://huggingface.co/datasets/{HF_DATASET_REPO}/resolve/main/{filename}"
-        return local_path, url, filename, len(image_bytes)
+        return local_path, url, filename, size_bytes
 
     except Exception as e:
-        print(f"[Error] Image upload failed: {e}")
+        print(f"[Error] during image upload: {e}")
         traceback.print_exc()
         return None, None, None, 0