Update app.py

app.py

@@ -425,6 +425,52 @@ def make_text():
         # Clean up the temporary file
         os.unlink(temp_file_path)
 
+
+@app.route('/upload-image-dataset', methods=['POST'])
+def upload_image_dataset():
+    data = request.json
+    url = data.get('url')
+    file_name = data.get('fileName')
+    repo_id = data.get('repoId')
+    repo_type = data.get('repoType', 'dataset')
+
+    if not url or not file_name or not repo_id:
+        return jsonify({"error": "Parameters 'url', 'fileName', and 'repoId' are required"}), 400
+
+    if repo_type not in ['space', 'dataset', 'model']:
+        return jsonify({"error": "Invalid 'repoType'. Must be 'space', 'dataset', or 'model'"}), 400
+
+    try:
+        # Download the image
+        response = requests.get(url)
+        response.raise_for_status()
+
+        # Create a temporary file
+        with tempfile.NamedTemporaryFile(delete=False) as temp_file:
+            temp_file.write(response.content)
+            temp_file_path = temp_file.name
+
+        # Upload the file to Hugging Face
+        api.upload_file(
+            path_or_fileobj=temp_file_path,
+            path_in_repo=file_name,
+            repo_id=repo_id,
+            repo_type=repo_type,
+        )
+
+        # Clean up the temporary file
+        os.unlink(temp_file_path)
+
+        return jsonify({
+            "message": f"File '{file_name}' uploaded successfully to {repo_id} ({repo_type})",
+            "size": len(response.content)
+        }), 200
+
+    except requests.RequestException as e:
+        return jsonify({"error": f"Error downloading image: {str(e)}"}), 500
+    except Exception as e:
+        return jsonify({"error": f"Error uploading file: {str(e)}"}), 500
+
     
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860, debug=True)